974-和可被K整除的子數組

前言

Weekly Contest 119的 和可被K整除的子數組：

給定一個整數數組 A，返回其中元素之和可被 K 整除的（連續、非空）子數組的數目。

示例：

輸入：A = [4,5,0,-2,-3,1], K = 5
輸出：7
解釋：
有 7 個子數組知足其元素之和可被 K = 5 整除：
[4, 5, 0, -2, -3, 1], [5], [5, 0], [5, 0, -2, -3], [0], [0, -2, -3], [-2, -3]

提示：

1. 1 <= A.length <= 30000
2. -10000 <= A[i] <= 10000
3. 2 <= K <= 10000

解題思路

本題描述十分簡單，根據連續、非空子數組這個要求能夠得知，子數組應該是原數組的截取（能夠是原數組自己），即要保證元素的順序。

實現代碼

/**
     * 974. 和可被 K 整除的子數組
     * @param A
     * @param K
     * @return
     */
    public int subarraysDivByK(int[] A, int K) {
        int result=0;
        for(int i=0;i<A.length;i++){
            //使用一個方法變量記錄累加結果，這個方法變量爲當前子數組的元素之和
            int total=0;
            for(int j=i;j<A.length;j++){
                total+=A[j];
                if(total%K==0){//判斷當前子數組元素之和是否可以被K整除
                    ++result;
                }
            }
        }
        return result;
    }