題意,可以從中間往左連續吃幾罐,從中間往右連續吃幾罐,最少一共吃多少可以使剩下的紅藍罐數目相等
遇到這種題,一般不是將按貢獻換成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;
}