题目链接:https://vjudge.net/contest/398847#problem/I
题意:给出一个不含0的数组,求最少插入几个数,使其所有子数组的和都不为0。
思路:利用前缀和和set集合;注意要先在set中插入0;从前往后往set中插入前缀和,若两处前缀和相等,说明中间的和为0;此时要清空set,插入当前这个数和0,继续往下找。
如图:第1个位置和第5个位置的前缀和相等,因为3+1+(-2)+(-2)=0
#include<iostream>
#include<set>
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
set<ll>st;
int main(){
int n; cin>>n;
ll presum=0;
st.insert(0);
ll ans=0;
for(int i=0;i<n;i++){
int x; cin>>x;
presum+=x;
if(st.count(presum)){
st.clear();
st.insert(0);
presum=x;
ans++;
}
st.insert(presum);
}
cout<<ans;
}