力扣 2025. 分割数组的最多方案数前缀和+哈希

最新推荐文章于 2022-09-02 17:32:27 发布

csu_xiji

最新推荐文章于 2022-09-02 17:32:27 发布

阅读量205

点赞数

分类专栏：力扣哈希思维文章标签：数组分割前缀和哈希表算法 LeetCode

本文链接：https://blog.csdn.net/xiji333/article/details/120609242

版权

力扣同时被 3 个专栏收录

282 篇文章 2 订阅

订阅专栏

思维

181 篇文章 1 订阅

订阅专栏

哈希

32 篇文章 1 订阅

订阅专栏

https://leetcode-cn.com/problems/maximum-number-of-ways-to-partition-an-array/
在这里插入图片描述
思路：第一次在比赛中间干掉 $h a r d$ 题，纪念一下。先计算出前缀和数组 $s u m$ ，并设整个数组的总和为 $t o t$ ，那么问题中的条件就等效于在前缀和数组中寻找 $sum_i=tot-sum_{i}$ 的 $i$ 的个数，也即 $sum_i=tot/2$ ，注意这里 $i$ 的取值范围是 $[0, n - 1)$ 。我们仔细思考一下，修改一个数会产生哪些影响？不妨设把 $nums_i$ 改成 $k$ ，那么 $s u m [0 \dots i)$ 无变化，但是 $s u m [i, \dots n)$ 会加上 $k-nums_i$ ，且整个数组的总和 $t o t$ 也需要加上 $k-nums_i$ 。不妨设 $dis=k-nums_i$ ，在经过这次变化之后，分割的方法数就等于 $j<i\&\&sum_j=(tot+dis)/2$ 的个数加上 $j>=i\&\&sum_j=(tot-dis)/2$ 的个数。如果我们维护两个哈希表，分别记录 $i$ 左右两侧的前缀和数组的元素出现次数，那么就可以在 $O (1)$ 复杂度内计算出分割方法数，整个算法的复杂度降低到了 $O (n)$ 。

class Solution {
public:
    int waysToPartition(vector<int>& nums, int k) {
        int n=nums.size();
        using ll=long long;
        vector<ll> sum(n);
        unordered_map<ll,int> leftCnt,rightCnt;
        sum[0]=nums[0];
        for(int i=1;i<n;i++)
        {
            sum[i]=sum[i-1]+nums[i];
            ++rightCnt[sum[i-1]];
        }
        int ans=0;
        ll tot=sum[n-1];
        if(!(tot&1)&&rightCnt.count(tot>>1))
            ans=rightCnt[tot>>1];
        for(int i=0;i<n;i++)
        {
            int dis=k-nums[i];
            if(!((tot+dis)&1))
            {
                ll leftVal=(tot+dis)>>1;
                ll rightVal=(tot-dis)>>1;
                int curAns=0;
                if(leftCnt.count(leftVal))
                    curAns+=leftCnt[leftVal];
                if(rightCnt.count(rightVal))
                    curAns+=rightCnt[rightVal];
                ans=max(ans,curAns);
            }
            --rightCnt[sum[i]];
             ++leftCnt[sum[i]];
        }
        return ans;
    }
};