【bzoj2141】排队

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;}