题意,可以从中间往左连续吃几罐,从中间往右连续吃几罐,最少一共吃多少可以使剩下的红蓝罐数目相等
遇到这种题,一般不是将按贡献换成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;
}