对于前缀和的应用见的已经很多了,但是它的作用仍然让我意想不到。
基础:
力扣题:和为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;
}