比赛 EYOI暨SBOI暑假快乐赛6th 评测结果 AAATTTTTTT
题目名称 Count 1s 最终得分 30
用户昵称 ┭┮﹏┭┮ 运行时间 7.057 s
代码语言 C++ 内存使用 5.81 MiB
提交时间 2022-06-30 09:56:51
显示代码纯文本
#include <bits/stdc++.h>
using namespace std;
int n,s,ans,ss;
int a[200010],aa[200010];
int main(){
    freopen("count1s.in","r",stdin);
    freopen("count1s.out","w",stdout);
    cin>>n;
    for(int i = 1;i <= n;i++){
        cin>>a[i];
        if(a[i] == 1)s++;
    }
    aa[s] = 1;ans++;
    for(int l = 1;l <= n;l++){
        for(int r = l;r <= n;r++){
            ss = 0;
            for(int i = l;i <= r;i++){
                if(a[i] == 1){
                    ss++;
                }
            }
            ss = s + ss - (r - l + 1 - ss);
            if(aa[ss] != 1){
                ans++;
                aa[ss] = 1;
            }
        }
    }
    cout<<ans<<endl;
        
    return 0;
}