题目:给定一个二进制数组 nums
, 找到含有相同数量的 0
和 1
的最长连续子数组,并返回该子数组的长度。
输入: nums = [0,1,0]
输出: 2
说明: [0, 1] (或 [1, 0]) 是具有相同数量 0 和 1 的最长连续子数组。
解题思路:
//运用前缀和+哈希的思想
//相同1和0的数量可以被理解为当遇到1时总和sum+1,遇到0时,sum-1,当1和0的数量相等时,sum=0
//哈希表存储的是sum的每个取值第一次出现的下标,因此当遇到重复的前缀和时,根据当前下标和哈希表中存储的下标计算得到的子数组长度是以当前下标结尾的子数组中满足有相同数量的 0和1 的最长子数组的长度。
源代码如下:
class Solution {
public:
int findMaxLength(vector<int>& nums) {
int maxlen=0;//保存含有相同数量0和1的子数组的最大长度
unordered_map<int,int> mp;//定义一个哈希表,mp[sum]=i表示当前下标i的前缀和
int sum=0;
mp[sum]=-1;//规定空的前缀的结束下标为 −1,由于空的前缀的元素和为 0因此在遍历之前,首先在哈希表中存入键值对 (0,−1)
int n=nums.size();
for(int i=0;i<n;i++)
{
if(nums[i]==1)//遇到1,sum+1
{
sum++;
}
//由于题目是一个二进制数组,只有0和1
else//遇到0,sum-1
{
sum--;
}
//map.count()用来判断键值元素是否存在,若存在,则返回1,反之,不存在,返回0
if(mp.count(sum))//sum的值存在,则说明找到了1和0数量相同的子数组
{
int pre_index=mp[sum];//取出sum在哈希表中对应的下标pre_index
maxlen=max(maxlen,i-pre_index);//当前子数组的长度为i-pre_index,不断更新maxlen
}
else//若不存在,就将i的值存放到mp[sum]
{
mp[sum]=i;
}
}
//返回最大长度
return maxlen;
}
};