1. 程式人生 > 實用技巧 >LeetCode523. 連續的子陣列和

LeetCode523. 連續的子陣列和

因為子陣列是連續的,所以判斷連續的子陣列的和時,我們往往開一個字首和陣列預處理出所有數的字首和,這樣能夠降低求子陣列的和的時間複雜度。

這題需要單獨處理k為0的情況,由於陣列所有元素都是非負數,所以當k為0時,如果存在兩個相鄰的數的值都為0,則返回true,否則返回false。

使用字首和判斷是否存在滿足條件的子陣列時,只需從小到大列舉子陣列的長度,然後再在內層迴圈內列舉子陣列的長度,然後使用字首和計算區間和的公式計算是否存在子陣列的和滿足條件即可。有點類似區間dp,hh。

程式碼如下:

class Solution {
public:
    bool checkSubarraySum(vector<int>& nums, int k) {
        int n = nums.size();
        if(k == 0) {
            for(int i = 0; i < n; ++i) {
                if(nums[i] == 0 && i + 1 < n && nums[i + 1] == 0) {
                    return true;
                }
            }
            return false;
        }
        vector<int> preSum(n + 1, 0);                  // preSum是字首和, preSum[i]表示nums[0] + nums[1] + ... + nums[i - 1]
        for(int i = 1; i <= n; ++i) {
            preSum[i] = preSum[i - 1] + nums[i - 1];
        }
        for(int len = 2; len <= n; ++len) {             // 從小到大列舉子陣列的長度,注意,題目要求子陣列的長度至少是2
            for(int i = 1; i + len - 1 <= n; ++i) {      // 列舉子陣列的起點,i + len - 1是子陣列的終點
                int j = i + len - 1;
                if((preSum[j] - preSum[i - 1]) % k == 0) {
                    return true;
                }
            }
        }
        return false;
    }
};