Problem - 1398C - Codeforces

解析：

        可以写出 sum[ j ] - sum[ i ] = j - i +1，对其移项得 sum[ j ] - j = sum[ i ] - （i - 1），问题转换为有多少个前缀和相等，则用map遍历统计一遍即可。 

        注意，当sum[ j ] - j = 0时，说明 1 到 j 区间也符合题意，但是却计算不进去，所以开始令mp[ 0 ] = 1

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=1e5+5;
int t,n,x,sum[N],res;
signed main(){
	scanf("%lld",&t);
	while(t--){
		scanf("%lld",&n);
		map<int,int>mp;
		mp[0]=1,res=0;
		for(int i=1;i<=n;i++){
			scanf("%1lld",&x);
			sum[i]=sum[i-1]+x;
			res+=mp[sum[i]-i]++;
		}
		printf("%lld\n",res);
	}
	return 0;
}