D. Non-zero Segments
过程:手玩样例发现,前缀和中出现相同值即为区间和为0
由前缀和公式(S_i = S_{i - 1} + A_i)可以知道,区间([l,r])满足题意即为(S_r = S_{l-1}),对于每个这样的区间插入个巨大的数,让区间([1, r-1])的前缀和都作废,注意对0处理
even such a huge that they can’t be represented in most standard programming languages
维护是否有相同的前缀和可以用(\text{set})
更多细节见代码
ll n, s[N], ans;
set<ll> S;
void solve()
{
cin>>n;
for(int i = 1; i <= n; i++)
cin>>s[i];
S.insert(0);
for(int i = 1; i <= n; i++)
{
s[i] += s[i - 1];
if(S.count(s[i]))
{
ans++;
S.clear();
S.insert(0);
s[i] -= s[i - 1];
}
S.insert(s[i]);
}
cout<<ans<<endl;
return;
}