[POI2007]立方体大作战tet
题目
做法
很巧妙的题,注意每种颜色只有两个
消除一种颜色,其实就是看中间有多少个没有被消除的块,这种动态距离问题显然能用树状数组解决
洛谷输出方案,暴力往下爬就行
My complete code
#include<bits/stdc++.h>
using namespace std;
typedef int LL;
const LL maxn=1e6+9;
LL n,top,ans,xiaochu;
LL pre[maxn],tree[maxn],sta[maxn];
inline LL Lowbit(LL x){ return x&-x; }
inline void Add(LL x,LL val){ for(;x<=2*n;x+=Lowbit(x)) tree[x]+=val; }
inline LL Query(LL x){ LL ret(0); for(;x;x-=Lowbit(x)) ret+=tree[x]; return ret; }
int main(){
scanf("%d",&n);
for(LL i=1;i<=2*n;++i) Add(i,1);
for(LL i=1;i<=2*n;++i){
LL col; scanf("%d",&col);
if(!pre[col]) pre[col]=i;
else{
ans+=Query(i-1)-Query(pre[col]);
LL dis(Query(i-1)-Query(pre[col])),now(i);
while(dis){
sta[++top]=now-xiaochu-1;
--now;
--dis;
}
Add(i,-1); Add(pre[col],-1);
xiaochu+=2;
}
}
printf("%d\n",ans);
for(LL i=1;i<=top;++i) printf("%d\n",sta[i]);
return 0;
}