Berry Jam codeforces 1278C

题目大意：

有两种类型的果酱，一个梯子，从中间开始吃，可以吃左边的，也可以吃右边的，最终要使两种类型的果酱的数量想等

题解：

思路对了，但是没考虑完。

对梯子的左侧的果酱I我们用两个数组记录其从1到i的1的数量和从1到i的2的数目，对梯子右侧的果酱保存从i到n的1的数目和2的数目。

然后两种情况，

第一种是最终答案最侧与右侧均有果酱剩余。我们只需要让左侧的1-2等于右侧2-1。

第二种情况，最终答案出现在一侧， 即左侧都吃光或者右侧都吃光。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=1E6+;
const ll INF=1e9+;
ll arr[N];
ll brr[N];
ll marka1[N];
ll marka2[N];
ll markb1[N];
ll markb2[N];
map<ll,ll >mp;
void solve(){
    mp.clear();
    ll n;
    cin>>n;
    for(ll i=;i<=n;i++)   cin>>arr[i];
    for(ll i=;i<=n;i++)   cin>>brr[i];
    for(ll i=;i<=n;i++){
        if(arr[i]==) {
            marka1[i]=marka1[i-]+;
            marka2[i]=marka2[i-];
        }
        else {
            marka1[i]=marka1[i-];
            marka2[i]=marka2[i-]+;
        }
    }
    markb1[n+]=;
    markb2[n+]=;
    for(ll i=n;i>=;i--){
        if(brr[i]==){
            markb1[i]=markb1[i+]+;
            markb2[i]=markb2[i+];
        }
        else {
            markb1[i]=markb1[i+];
            markb2[i]=markb2[i+]+;
        }
    }
    for(ll i=n;i>=;i--){
        ll k=marka1[i]-marka2[i];
        if(mp[k]==) {
            mp[k]=i;
        }
    }
    ll ans=INF;
    for(ll i=;i<=n;i++){
        ll k=markb2[i]-markb1[i];
        if(mp[k]!=){
            ans=min(ans,n-mp[k]+i-);
        }
    }
    ll ans2=;
    for(int i=;i<=n;i++){
        if(marka1[i]==marka2[i]) ans2=i;
    }
    ans2=n-ans2+n;
    ll ans3=n+;
    for(int i=n;i>=;i--){
        if(markb1[i]==markb2[i])  ans3=i;
    }
    ans3=n+ans3-;
    cout<<min(ans3,min(ans,ans2))<<endl;
}
int main(){
    ios::sync_with_stdio();
    ll t;
    cin>>t;
    while(t--) solve();
    return ;
}