Educational Codeforces Round 78 (Rated for Div. 2) C - Berry Jam(前缀和)


🤔 🤔 🤔

题意,可以从中间往左连续吃几罐,从中间往右连续吃几罐,最少一共吃多少可以使剩下的红蓝罐数目相等

遇到这种题,一般不是将按贡献换成01序列就是换成1-1序列,这个题显然是需要换成1-1序列之后,求出一个前缀和与后缀和相加等于0的位置,后缀和可以由前缀和相减得到,所以我们这里只计算一个前缀和;枚举右边吃了多少罐,对于每一个后缀和查找是否有和他互为相反数的前缀和(这里就不得不提一下map,内部自动有序查找太方便了),最后就是注意边界问题,因为可能往左往右一个不吃;

int sum[MAXN];
int32_t main()
{
    int t;cin>>t;sum[0]=0;
    while(t--)
    {
        int n;cin>>n;
        map<int,int>pos;
        pos[0]=0;//往左全部吃完
        for(int i=1;i<=n*2;++i)
        {
            int a;cin>>a; if(a==2) a=-1;
            sum[i]=sum[i-1]+a;
            if(i<=n) pos[sum[i]]=i;//i=n,往左一个也不吃
        }
        int ans=n*2;
        for(int i=n;i<=n*2;++i)//i=n,往右一个也不吃,i=2*n,往右全部吃完
        {
            auto it=pos.find(sum[i]-sum[n*2]);//
            if(it!=pos.end()) ans=min(ans,i-it->second);//
        }
        cout<<ans<<endl;
    }
    return 0;
}


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM