leetcode 974 整除K的子数组

public static int subarraysDivByK3(int[] A, int K) {
        int lenA = A.length;
        int[] pre = new int[lenA+1];
        //for(int i=0;i<lenA;i++){
        //    pre[i+1]=pre[i]+A[i];
        // }
        HashMap<Integer,Integer> record = new HashMap<>();
        record.put(0,1);
        int mod ;
        int ans = 0;
        int sum=0;
        for(int i=0;i<lenA;i++) {
            sum += A[i];
            mod = (sum % K + K) % K;
            if (record.containsKey(mod)) {
                ans += record.get(mod);
            }
            record.put(mod, record.getOrDefault(mod, 0) + 1);
        }
        return ans;
    }