1. 程式人生 > >國家集訓隊 排隊

國家集訓隊 排隊

題目描述

題解:

將交換看作兩個插入+兩個刪除。

然後CDQ。

程式碼:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 30050
#define ll long long
inline int rd()
{
    int f=1,c=0;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='
0'&&ch<='9'){c=10*c+ch-'0';ch=getchar();} return f*c; } int n,m,cnt,stu[N]; struct Pair { int x,id; }tk[N]; bool cmp(Pair a,Pair b) { return a.x<b.x; } struct node { int pos,val,id,wgt; node(){} node(int p,int v,int i,int w):pos(p),val(v),id(i),wgt(w){} }p[N],tmp[N];
void Sort(int l,int r) { int mid = (l+r)>>1; int i=l,j=mid+1,k=l; while(i<=mid&&j<=r) { while(i<=mid&&p[i].pos<=p[j].pos) { tmp[k]=p[i]; i++,k++; } while(j<=r&&p[i].pos>p[j].pos) { tmp[k]
=p[j]; j++,k++; } } while(i<=mid) { tmp[k]=p[i]; i++,k++; } while(j<=r) { tmp[k]=p[j]; j++,k++; } for(i=l;i<=r;i++) p[i]=tmp[i]; } int f[N]; void up(int x,int d) { if(!x)return ; while(x<N)f[x]+=d,x+=(x&-x); } int down(int x) { if(!x)return 0; int ret = 0; while(x)ret+=f[x],x-=(x&-x); return ret; } int k = 0; ll ans[N]; void cdq(int l,int r) { if(l==r)return ; int mid = (l+r)>>1; cdq(l,mid),cdq(mid+1,r); Sort(l,mid),Sort(mid+1,r); int i,j; j=l; for(i=mid+1;i<=r;i++) { while(j<=mid&&p[j].pos<=p[i].pos)up(p[j].val,p[j].wgt),j++; ans[p[i].id]+=p[i].wgt*(down(k)-down(p[i].val)); } for(j=j-1;j>=l;j--)up(p[j].val,-p[j].wgt); j=mid; for(i=r;i>mid;i--) { while(j>=l&&p[j].pos>=p[i].pos)up(p[j].val,p[j].wgt),j--; ans[p[i].id]+=p[i].wgt*down(p[i].val-1); } for(j=j+1;j<=mid;j++)up(p[j].val,-p[j].wgt); } int main() { n=rd(); for(int x,i=1;i<=n;i++) { x=rd(); tk[i].x=x,tk[i].id=i; } sort(tk+1,tk+n+1,cmp); for(int las=-1,i=1;i<=n;i++) { if(las!=tk[i].x) { las=tk[i].x; k++; } int j = tk[i].id; p[j]=node(j,k,0,1); stu[j] = k; } m=rd(); for(int l,r,i=1;i<=m;i++) { l=rd(),r=rd(); p[++n]=node(l,stu[l],i,-1); p[++n]=node(r,stu[r],i,-1); swap(stu[l],stu[r]); p[++n]=node(l,stu[l],i,1); p[++n]=node(r,stu[r],i,1); } cdq(1,n); for(int i=1;i<=m;i++)ans[i]+=ans[i-1]; for(int i=0;i<=m;i++)printf("%lld\n",ans[i]); return 0; }