类似题:蓝桥杯 k倍区间 前缀和
#include <bits/stdc++.h>
using namespace std;
#define ll long long
ll s[100010];
int main()
{
std::ios::sync_with_stdio(false);
int t;
cin>>t;
while(t--)
{
memset(s,0,sizeof s);
map<int,int> c;
int n;
cin>>n;
string str;
cin>>str;
ll sum=0;
int a;
for(int i=0;i<str.size();i++)
{
a=str[i]-'0'-1;
s[i]=s[i-1]+a;
sum+=c[s[i]];
c[s[i]]++;
}
sum+=c[0];
cout<<sum<<endl;
}
return 0;
}