Educational Codeforces Round 93 (Rated for Div. 2) K. Lonely Numbers

Problem - C - Codeforces

翻译：

给定一个数组𝑎1，𝑎2，…，𝑎𝑛，由0到9的整数组成。一子数组𝑎𝑙,𝑎𝑙+ 1,𝑎𝑙+ 2,…,𝑎𝑟−1,𝑎𝑟很好如果这个子数组的元素之和等于这个子数组的长度(∑𝑖=𝑙𝑟𝑎𝑖=𝑟−𝑙+ 1)。

例如,如果𝑎=(1 2 0),然后有3好子阵:𝑎1…1 =[1],𝑎2…3 =(2,0)和𝑎1…3 =(1 2 0)。

计算数组𝑎的好子数组的数量。

输入
第一行包含一个整数𝑡(1≤𝑡≤1000)——测试用例的数量。

每个测试用例的第一行包含一个整数𝑛(1≤𝑛≤105)——数组𝑎的长度。

每个测试用例的第二行包含一个由𝑛十进制数字组成的字符串，其中𝑖-th数字等于𝑎𝑖的值。

可以保证所有测试用例中𝑛的总和不超过105。

输出
对于每个测试用例，打印一个整数——数组𝑎的好子数组的数量。

例子
inputCopy
3.
3.
120
5
11011
6
600005
outputCopy
3.
6
1
请注意
语句中考虑第一个测试用例。

在第二个测试用例,有6好子阵:𝑎1…1,𝑎2…2,𝑎1…2,𝑎4…4,𝑎5…5和𝑎4…5。

在第三个测试用例中，只有一个好的子数组:𝑎2…

思路：

区间和等于区间长度，我们将其转化成为前缀和，那就是s[r]-s[l]=r-l+1，将其每次减1，那么就是s[r]-r=s[l]-l,所以就是相同的前缀然后其中取两个就好了。

代码：

/*Looking! The blitz loop this planet to search way
 
 Only my RAILGUN can shoot it 今すぐ
 
 身体中を  光の速さで
 
 駆け巡った確かな予感
 
 掴め！ 望むものなら残さず
 
 輝ける自分らしさで
 
 信じてるよ  あの日の誓いを
 
 この瞳に光る涙それさえも  強さになるから
 
 */
#include <iostream>
#include <algorithm>
#include <string.h>
#include <string>
#include <math.h>
#include <stdio.h>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<tuple>
#include<numeric>
#include<stack>
using namespace::std;
typedef long long  ll;
int n,t;
inline __int128 read(){
    __int128 x = 0, f = 1;
    char ch = getchar();
    while(ch < '0' || ch > '9'){
        if(ch == '-')
            f = -1;
        ch = getchar();
    }
    while(ch >= '0' && ch <= '9'){
        x = x * 10 + ch - '0';
        ch = getchar();
    }
    return x * f;
}
inline void print(__int128 x){
    if(x < 0){
        putchar('-');
        x = -x;
    }
    if(x > 9)
        print(x / 10);
    putchar(x % 10 + '0');
}

int a[100005];
ll s[100005];
char ss;
void wanyurukong(){
    cin>>n;
    ll an=0;
    map<ll,ll>q;
    for (int i=1; i<=n; i++) {
        cin>>ss;
        a[i]=ss-'0';
    }
    q[0]=1;
    for (int i =1; i<=n; i++) {
        s[i]=s[i-1]+a[i]-1;
        q[s[i]]++;
    }

    for(auto x:q){
        an+=x.second*(x.second-1)/2;
    }
    printf("%lld\n",an);
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(); cout.tie();
    cin>>t;
    while (t--) {
        wanyurukong();
    }
    //wanyurukong
    return 0;
}