2141: 排队

分块模板题。

在块内的暴力统计，每一块开一个树状数组。

我SB逆序对求法写错。

code：

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm> 
using namespace std;
int n,pre[20010],tr[150][20010],num=0;
int ri[150];
struct node{
	int b,a,num;
}a[20010];int ans=0;
int lowbit(int x){return x&(-x);}
void change(int root,int k,int c)
{
	for(int i=k;i<=num;i+=lowbit(i))
		tr[root][i]+=c;
}
int findans(int root,int k)
{
	int ANS=0;
	for(int i=k;i>=1;i-=lowbit(i))
		ANS+=tr[root][i];
	return ANS;
}
int get(int root,int l,int r){return findans(root,r)-findans(root,l-1);}
bool cmp(node a,node b){return a.b<b.b;}
void turn_ri(int x,int y)
{
	while(x<y)
	{
		if(a[x].a>a[x+1].a) ans--;
		if(a[x].a<a[x+1].a) ans++;
		swap(a[x].a,a[x+1].a);
		x++;
	}
}
void turn_le(int x,int y)
{
	while(x>y)
	{
		if(a[x].a>a[x-1].a) ans++;
		if(a[x].a<a[x-1].a) ans--;
		swap(a[x].a,a[x-1].a);
		x--;
	}
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i].b),a[i].num=i;
	int tmp;sort(a+1,a+n+1,cmp);
	for(int i=1;i<=n;i++)
	{
		if(i==1||a[i].b!=a[i-1].b) num++;
		a[a[i].num].a=num;
	}
	tmp=sqrt(n);
	for(int i=1;i<=n;i++)
	{
		pre[i]=i/tmp+1;
		if(pre[i]!=pre[i-1]) ri[pre[i-1]]=i;
		if(i==n) ri[pre[i]]=n+1;
	}
	for(int i=1;i<=n;i++) ans+=get(1,a[i].a+1,num),change(1,a[i].a,1);
	printf("%d\n",ans);
	memset(tr,0,sizeof(tr));
	for(int i=1;i<=n;i++) change(pre[i],a[i].a,1);
	int m;scanf("%d",&m);
	while(m--)
	{
		int x,y;scanf("%d %d",&x,&y);
		if(x>y) swap(x,y);
		if(pre[x]==pre[y]) turn_ri(x,y-1),turn_le(y,x);
		else
		{
			turn_ri(x,ri[pre[x]]-1);turn_le(y,ri[pre[y]-1]);
			int l=ri[pre[x]]-1,r=ri[pre[y]-1];
			for(int i=pre[x]+1;i<=pre[y]-1;i++)
			{
				ans-=findans(i,a[l].a-1);
				ans+=get(i,a[l].a+1,num);
				ans+=findans(i,a[r].a-1);
				ans-=get(i,a[r].a+1,num);
			}
			change(pre[x],a[l].a,-1);change(pre[y],a[r].a,-1);
			change(pre[x],a[r].a,1);change(pre[y],a[l].a,1);
			if(a[l].a<a[r].a)ans++;
			if(a[l].a>a[r].a)ans--;
			swap(a[l].a,a[r].a);
			turn_ri(r,y);turn_le(l,x);
		}
		printf("%d\n",ans);
	}
}