Given a binary array, find the maximum length of a contiguous subarray with equal number of 0 and 1.
思路:为了找0,1数量相同的最长子列,题目也有tag使用hash表,我们用map充当hash表,希望用和的形式表示出0,1的数量相同,我们将数组里的0全部换为-1,这样的话-1 + 1的最大子列就是和为0的时候,若最大子列无法使得和为0,这个和一定是之前出现过的,就是说一定是-1和1的数目相等前就出现过一次了,这样可以判断子列的长度。hash表中我们存放和sum以及对应此和的数组下标。
class Solution {
public:
int findMaxLength(vector<int>& nums) {
if (nums.size() == 0) {
return 0;
}
int result = 0, sum = 0;
map<int, int>hash;
hash[0] = -1;
for(int i = 0; i < nums.size(); i++) {
if (nums[i] == 0) {
nums[i] = -1;
}
sum += nums[i];
map<int, int>::iterator it = hash.find(sum);
if (it == hash.end()) {
hash[sum] = i;
} else {
result = max(result, i - hash[sum]);
}
}
return result;
}
};