3295: [Cqoi2011]动态逆序对
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 5987 Solved: 2080
[Submit][Status][Discuss]
Description
对于序列A，它的逆序对数定义为满足i<j，且Ai>Aj的数对(i,j)的个数。给1到n的一个排列，按照某种顺序依次删除m个元素，你的任务是在每次删除一个元素之前统计整个序列的逆序对数。
Input
输入第一行包含两个整数n和m，即初始元素的个数和删除的元素个数。以下n行每行包含一个1到n之间的正整数，即初始排列。以下m行每行一个正整数，依次为每次删除的元素。
Output
输出包含m行，依次为删除每个元素之前，逆序对的个数。
Sample Input
5 4
1
5
3
4
2
5
1
4
2
Sample Output
5
2
2
1
样例解释
(1,5,3,4,2)?(1,3,4,2)?(3,4,2)?(3,2)?(3)。
HINT
N<=100000 M<=50000
每删去一个数,影响答案的就是它之后小于它的数和它之前大于它的数
树套树,外层维护区间,内层维护权值
#include<cstdio> #include<iostream> #include<algorithm> #include<cstring> #define ll long long #define N 100050 #define M N*100 using namespace std; int n,m,cnt,rt[N],pos[N],ls[M],rs[M],sum[M],c[N]; void add(int p){ while(p<=n){ c[p]++; p+=p&-p; } } int getsum(int p){ int t=0; while(p){ t+=c[p]; p-=p&-p; } return t; } void update(int &u,int l,int r,int p,int v){ if(!u)u=++cnt;sum[u]+=v; if(l==r)return; int mid=l+r>>1; if(p<=mid)update(ls[u],l,mid,p,v); else update(rs[u],mid+1,r,p,v); } void insert(int x,int p,int val){ while(x<=n){ update(rt[x],1,n,p,val); x+=x&-x; } } int ask(int u,int l,int r,int p){ if(l==r)return sum[u]; int mid=l+r>>1; if(p<=mid)return ask(ls[u],l,mid,p); return sum[ls[u]]+ask(rs[u],mid+1,r,p); } int query(int p,int x){ int t=0; while(p){ t+=ask(rt[p],1,n,x); p-=p&-p; } return t; } int pre(int a,int b){ return query(b,n)-query(b,a); } int suf(int a,int b){ return query(n,a)-query(b,a); } int main(){ scanf("%d%d",&n,&m);int x; ll ans=0; for(int i=1;i<=n;i++){ scanf("%d",&x);pos[x]=i; ans+=getsum(n)-getsum(x); add(x);insert(i,x,1); } for(int i=1;i<=m;i++){ printf("%lld\n",ans);scanf("%d",&x); ans-=pre(x,pos[x])+suf(x,pos[x]); insert(pos[x],x,-1); } return 0; }
