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

最新推荐文章于 2020-03-31 11:40:46 发布

小松萘

最新推荐文章于 2020-03-31 11:40:46 发布

阅读量328

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/weixin_44116061/article/details/103626344

本文探讨了一种算法，旨在解决红蓝罐数量平衡问题。通过将问题转化为1-1序列，利用前缀和与后缀和的概念，结合map数据结构进行高效查找，最终找到使红蓝罐数量相等所需的最小操作次数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

🤔 🤔 🤔

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

遇到这种题，一般不是将按贡献换成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;
}