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