(算法)连续数组————<前缀和数组>

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;
	}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

课堂随笔

感谢支持~~~

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值