560. 和為 K 的子數組

解題思路

• 創建一個前綴和數組 preSum，其長度比原數組 nums 多 1。preSum[i] 表示 nums 中前 i 個元素的和。
• 通過遍歷 nums 數組，計算前綴和數組 preSum。
  • 在嵌套的兩個循環中，對所有可能的子數組進行窮舉：
  • 外層循環從 1 到 n（n 是數組長度），代表子數組的結束位置。
  • 內層循環從 0 到 i-1，代表子數組的起始位置。
• 在每一對起始位置和結束位置上，通過計算前綴和數組中的差值，判斷子數組的和是否等于 k。
• 如果等于 k，則將結果計數器 ans 加一。

class Solution {public int subarraySum(int[] nums, int k) {// 前綴和數組int n = nums.length;int[] preSum = new int[n + 1]; preSum[0] = 0;for(int i = 0; i < n; i++){preSum[i + 1] = preSum[i] + nums[i];}// preSum[i] 代表nums[0...i - 1]的和int ans = 0;// 窮舉所有子數組for(int i = 1; i <= n; i++){for(int j = 0; j < i; j++){if(preSum[i] - preSum[j] == k){ans++;}}}return ans;}
}