【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
,ls
,rs
,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=
10a37
=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;
}