1. 题⽬链接:525.连续数组
2. 题⽬描述:
(暴⼒解法就是枚举所有的⼦数组,然后判断⼦数组是否满⾜要求,这⾥不再赘述。)
3. 解法(前缀和在哈希表中):
算法思路:
稍微转化⼀下题⽬,就会变成我们熟悉的题:
• 本题让我们找出⼀段连续的区间, 0 和1 出现的次数相同。
• 如果将0 记为-1 , 1 记为1 ,问题就变成了找出⼀段区间,这段区间的和等于0 。
• 于是,就和560.和为K的⼦数组这道题的思路⼀样
设i 为数组中的任意位置,⽤sum[i] 表⽰[0, i] 区间内所有元素的和。 想知道最⼤的「以i 为结尾的和为0 的⼦数组」,就要找到从左往右第⼀个x1 使得[x1, i] 区间内的所有元素的和为0 。
那么[0, x1 - 1] 区间内的和是不是就是sum[i] 了。于是问题 就变成:
• 找到在[0, i - 1] 区间内,第⼀次出现sum[i] 的位置即可。 我们不⽤真的初始化⼀个前缀和数组,因为我们只关⼼在i 位置之前,第⼀个前缀和等于sum[i] 的位置。因此,我们仅需⽤⼀个哈希表,⼀边求当前位置的前缀和,⼀边记录第⼀次出现该前缀和的 位置。
C++算法代码:
class Solution
{
public:
int findMaxLength(vector<int>& nums)
{
//将0转化为-1,将问题变化为和为0的最长子序列
for(int i=0;i<nums.size();i++)
{
if(nums[i]==0)
{
nums[i]=-1;
}
}
unordered_map<int,int>hash; //前缀和,下标
hash[0]=-1;
int sum=0;
int max_len=0;
for(int i=0;i<nums.size();i++)
{
//计算前缀和
sum+=nums[i];
//如果有符合条件的则添加
if(hash.count(sum))
{
max_len=max(max_len,i-hash[sum]);
}
else
{
hash[sum]=i;
}
}
return max_len;
}
};
Java算法代码:
class Solution {
public int findMaxLength(int[] nums) {
Map<Integer, Integer> hash = new HashMap<Integer, Integer>();
hash.put(0, -1); // 默认存在⼀个前缀和为 0 的情况
int sum = 0, ret = 0;
for (int i = 0; i < nums.length; i++)
{
sum += (nums[i] == 0 ? -1 : 1); // 计算当前位置的前缀和
if (hash.containsKey(sum)) ret = Math.max(ret, i - hash.get(sum));
else hash.put(sum, i);
}
return ret;
}
}