FilesExpand file tree

 master

/

SubarraySumEqualsK.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

26 lines (23 loc) · 883 Bytes

 master

/

SubarraySumEqualsK.java

Copy path

Top

File metadata and controls

• Code
• Blame

26 lines (23 loc) · 883 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

import java.util.HashMap;

/**

* https://leetcode.com/articles/subarray-sum-equals-k/

*/

public class SubarraySumEqualsK {

/**

* 注意map.put(0,1)，即当前数自己也算是一种

* TestCase [1], 0，即k=0的情况

*/

public int subarraySum(int[] nums, int k) {

HashMap<Integer, Integer> map = new HashMap<>();

map.put(0, 1);

int count = 0;

for (int i = 0, sum = 0; i < nums.length; i++) {

sum += nums[i];

// 以下两句顺序非常重要，因为两个差要至少相差一个数以上，因此不能刚给当前sum放到Map就参与运算

// 比如k=0，如果刚给sum放到map就参与运算，那么sum-sum=0的情况是非法的

count += map.getOrDefault(sum - k, 0);

map.put(sum, map.getOrDefault(sum, 0) + 1);

}

return count;

}

}