Description
排排坐,吃果果,生果甜嗦嗦,大家笑呵呵。你一个,我一个,大的分给你,小的留给我,吃完果果唱支歌,大家乐和和。红星幼儿园的小朋友们排起了长长地队伍,准备吃果果。不过因为小朋友们的身高有所区别,排成的队伍高低错乱,极不美观。设第i个小朋友的身高为hi,我们定义一个序列的杂乱程度为:满足ihj的(i,j)数量。幼儿园阿姨每次会选出两个小朋友,交换他们的位置,请你帮忙计算出每次交换后,序列的杂乱程度。为方便幼儿园阿姨统计,在未进行任何交换操作时,你也应该输出该序列的杂乱程度。
Input
第一行为一个正整数n,表示小朋友的数量;第二行包含n个由空格分隔的正整数h1,h2,…,hn,依次表示初始队列中小朋友的身高;第三行为一个正整数m,表示交换操作的次数;以下m行每行包含两个正整数ai和bi¬,表示交换位置ai与位置bi的小朋友。
Output
输出文件共m行,第i行一个正整数表示交换操作i结束后,序列的杂乱程度。
Sample Input
【样例输入】
3
130 150 140
2
2 3
1 3
Sample Output
1
0
3
【样例说明】
未进行任何操作时,(2,3)满足条件;
操作1结束后,序列为130 140 150,不存在满足ihj的(i,j)对;
操作2结束后,序列为150 140 130,(1,2),(1,3),(2,3)共3对满足条件的(i,j)。
【数据规模和约定】
对于100%的数据,1≤m≤2*103,1≤n≤2*104,1≤hi≤109,ai≠bi,1≤ai,bi≤n。
题解 树套树暴力。
代码
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define N 15000005 int sum[N],ls[N],rs[N],sz,root[80005]; struct node1{int v,id;}a[20005]; int cnt,n,m,hash[20005],ans; using namespace std; inline int read() { int x=0;char ch=getchar(); while (ch<'0'||ch>'9') ch=getchar(); while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar(); return x; } bool cmpls(node1 a,node1 b){return a.v<b.v;} int query(int k,int l,int r,int x,int y) { if (!k) return 0; if (l==x&&r==y) return sum[k]; int mid=(l+r)>>1; if (y<=mid) return query(ls[k],l,mid,x,y); else if (x>mid) return query(rs[k],mid+1,r,x,y); return query(ls[k],l,mid,x,mid)+query(rs[k],mid+1,r,mid+1,y); } int queryhigher(int k,int l,int r,int x,int y,int rk) { if (x>y) return 0; if (l>r) return 0; if (rk==cnt) return 0; if (l==x&&r==y) return query(root[k],1,cnt,rk+1,cnt); int mid=(l+r)>>1; if (y<=mid) return queryhigher(k<<1,l,mid,x,y,rk); else if (x>mid) return queryhigher(k<<1|1,mid+1,r,x,y,rk); return queryhigher(k<<1,l,mid,x,mid,rk)+queryhigher(k<<1|1,mid+1,r,mid+1,y,rk); } int querylower(int k,int l,int r,int x,int y,int rk) { if (x>y) return 0; if (l>r) return 0; if (rk==1) return 0; if (l==x&&r==y) return query(root[k],1,cnt,1,rk-1); int mid=(l+r)>>1; if (y<=mid) return querylower(k<<1,l,mid,x,y,rk); else if (x>mid) return querylower(k<<1|1,mid+1,r,x,y,rk); return querylower(k<<1,l,mid,x,mid,rk)+querylower(k<<1|1,mid+1,r,mid+1,y,rk); } void update(int &k,int l,int r,int x,int y) { if (!k) k=++sz; if (l==r) { sum[k]+=y; return; } int mid=(l+r)>>1; if (x<=mid) update(ls[k],l,mid,x,y); else update(rs[k],mid+1,r,x,y); sum[k]=sum[ls[k]]+sum[rs[k]]; } void change(int k,int l,int r,int x,int y,int z) { update(root[k],1,cnt,y,z); if (l==r)return; int mid=(l+r)>>1; if (x<=mid) change(k<<1,l,mid,x,y,z); else change(k<<1|1,mid+1,r,x,y,z); } int main() { n=read(); for (int i=1;i<=n;i++) { a[i].v=read(); a[i].id=i; } sort(a+1,a+n+1,cmpls); int pre=-1; for (int i=1;i<=n;i++) { if (a[i].v!=pre) pre=a[i].v,cnt++; hash[a[i].id]=cnt; } for (int i=1;i<=n;i++) { ans+=queryhigher(1,1,n,1,i-1,hash[i]); change(1,1,n,i,hash[i],1); } printf("%d\n",ans); m=read(); while (m--) { int l=read(),r=read(); if (l>r) swap(l,r); ans+=queryhigher(1,1,n,l+1,r-1,hash[l])+querylower(1,1,n,l+1,r-1,hash[r]); ans-=queryhigher(1,1,n,l+1,r-1,hash[r])+querylower(1,1,n,l+1,r-1,hash[l]); change(1,1,n,l,hash[l],-1);change(1,1,n,r,hash[r],-1); change(1,1,n,l,hash[r],1);change(1,1,n,r,hash[l],1); if (hash[l]>hash[r]) ans--;else if (hash[l]!=hash[r]) ans++; swap(hash[l],hash[r]); printf("%d\n",ans); } return 0; }