巧用前缀和

对于前缀和的应用见的已经很多了,但是它的作用仍然让我意想不到。

基础:

力扣题:和为k的子数组
这道题

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int,int> hash;
        hash[0]=1;//如果某一个前缀和的值直接就是所求的k那么此时前缀和sum-k=0 而前缀和map里已经有了0, 那么就判定存在
        int sum=0,cot=0;
        for(int i=0;i<nums.size();i++){
            sum+=nums.at(i);
            if(hash.find(sum-k)!=hash.end()){
               //前面存储的前缀和里面已经有sum-k了,那么此时的sum到当初的sum-k这一段,就是所求的那一段和为k的子数组
                cot+=hash[sum-k];
            }
             hash[sum]++;
        }
        return cot;
    }
};

进一步

CodeForce round918 div4 E题
两个前缀和数组A,B并列放,A和B分别是C和D的前缀和数组,它们之间各自选出两个数,a1,a2,b1,b2众所周知,a2-a1与b2-b1的意义一目了然,那么a1-b1和a2-b2的意义呢?
换句话说,a1-b1=a2-b2时意味着什么?意味着a2-a1=b2-b1,也就意味着C和D这两个数组这一段的子数组和相同。便能解决这个问题

#include <iostream>
#include <bits/stdc++.h>

#define For for(int i=1;i<=n;i++)
#define Whole(x) for(auto item:x)
const int N = 2e5 + 10;
using ll = long long;
using namespace std;

inline void solve() {
   int n;cin>>n;
   vector<int> arr(N);
   For{
       cin>>arr[i];
   };
   ll s1=0,s2=0;//分别记录着奇数前缀和此时的值以及偶数前缀和的值
   map<ll,int> hash;//一开始用了unordered_map结果哈希冲突了,TLE了
   For{
       if(i&1){
           s1+=arr[i];
       }else s2+=arr[i];
       if(s1==s2){//此时奇数偶数和相同,自然而然输出
           cout<<"Yes"<<'\n';
           return;
       }else if(hash.find(s1-s2)!=hash.end()){
           //奇数偶数前缀和的差在两个时间段相同了,那么意味着这一时间段上的所有奇数的和与偶数的和相同
           cout<<"Yes"<<'\n';
           return;
       }
       hash[s1-s2]++;
   };
    cout<<"No"<<'\n';
    return;

}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0), cout.tie(0);
    int num = 1;
    cin >> num;
    while (num--)
        solve();
    return 0;
}
  • 7
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值