LeetCode--11_974_Subarray_Sums_Divisible_by_K

public static int subarraysDivByK(int[] A, int K) {
        int ans = 0,sum = 0;
        int[] B = new int [K];
        for(int j=0;j<A.length;j++) {
            sum+=A[j];
            B[(sum%K+K)%K]++;
        }
        for(int j=0;j<K;j++) {
            if(B[j]>1) {
                ans+=(B[j]-1)*B[j]/2;
            }
        }
        ans+=B[0];
        return ans ;
    }