Educational Codeforces Round 93 (Rated for Div. 2) C. Good Subarrays

C. Good Subarrays

原题链接:https://codeforces.ml/contest/1398/problem/C

题目大意:

给定一个序列,序列中的数为$0,1...9$,求有多少个连续子序列的和等于它的长度。

解题思路:

首先将所有的数都减去1,就可以转化为求有多少个区间和为0,可以维护一个前缀和,当前$i$个数的前缀和等于前$j$个数,则说明$(i,j]$这个区间和为0,所以记录每个前缀和的个数,然后遍历序列即可。要注意的是初始化时$dp_0 = 1$。

代码:

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 typedef long long ll;
 4 typedef pair<ll,ll> pi;
 5 typedef complex <double> cp;
 6 #define debug(a) cout<<#a<<":"<<a<<endl;
 7 #define fr freopen("in.txt","r",stdin);
 8 #define Fill(x,a) memset(x,a,sizeof(x))
 9 #define cpy(a,b) memcpy(a,b,sizeof(a))
10 const double PI = acos(-1);
11 const ll INF=0x3f3f3f3f;
12 const ll N=1e6+7;
13 const ll mod=1e9+7;
14 ll maxn,minn;
15 ll T,n,m,q;
16 string s;
17 ll arr[N];
18 map<int,int>mp;
19 
20 
21 int main(){
22     ll ans,res;
23     cin>>T;
24     while(T--){
25         cin>>n>>s;
26         ans=res=0;
27         mp.clear();
28         for(ll i=0;i<n;i++){
29             arr[i+1]=s[i]-'0'-1;    
30         }
31         mp[0]=1;
32         for(ll i=1;i<=n;i++){
33             res+=arr[i];
34             ans=ans+mp[res];
35             mp[res]++;
36         }
37         cout<<ans<<endl;
38         
39     }
40 
41 
42     return 0;
43 }
44  

原文地址:https://www.cnblogs.com/meanttobe/p/13514695.html