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