1.解析
题目大意,求解0和1的个数相等的最长连续子序列。
2.分析
刚开始我题目理解错啦,参考Grandyang的思路。一般求解连续子序列的题目,都要涉及到累加和,主要利用到的一个知识:a + b = c(a、b和c均为连续子序列的和); 如果a和c序列的和相等,则b子序列和为0。所以还涉及到一个trick,即如果当前元素是0,则标记当前为-1,这样将查找0和1个数相同的子序列转换成查找和为0的子序列。利用hashtable记录检测过的子序列的和sum和对应的起始坐标i,若当前累加和已经出现过,意味着存在一段0和1相等的子序列,更新当前的最大值。设计hashtable初始值为{-1, 0},是为了防止忽略前面的{0, 1}或者{1, 0}。
class Solution {
public:
int findMaxLength(vector<int>& nums) {
unordered_map<int, int> sum_to_index{{0, -1}};
int sum = 0, res = 0;
for (int i = 0; i < nums.size(); ++i){
sum += nums[i] == 0 ? -1 : 1;
if (sum_to_index.count(sum)) //当前和已经存在,则说明存在0和1个数相等的子序列
res = max(res, i - sum_to_index[sum]);
else
sum_to_index[sum] = i;
}
return res;
}
};