Anton and Permutation(分块)

Posted 2021-11-07 thusloop

Anton and Permutation
题意:对于一个长度为 nn 的序列进行 kk 次操作，每次操作都是交换序列中的某两个数。对于每一个操作，回答当前序列中有多少个逆序对。

//#pragma GCC optimize(2)
//#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
#define int long long
#define fi first
#define se second
#define pb push_back
#define pii pair<int,int>
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const int inf=(int)8e18+100;
const int maxn=2e5+100;
vector<int>g[maxn];
int pos[maxn],a[maxn],sz;
int ask(int l,int r,int d)
{
	if(l>r)return 0;
	int sum=0;
	for(int i=l; i<=min(r,pos[l]*sz); i++)
	{
		if(a[i]<d)sum++;
	}
	if(pos[l]!=pos[r])
	{
		for(int i=pos[l]+1; i<=pos[r]-1; i++)
		{
			sum+=upper_bound(g[i].begin(),g[i].end(),d)-g[i].begin();
		}
		for(int i=(pos[r]-1)*sz+1; i<=r; i++)
		{
			if(a[i]<d)sum++;
		}
	}
	return sum;
}
signed main()
{
	IOS
	int n,q;
	cin>>n>>q;
	sz=sqrt(n);
	for(int i=1; i<=n; i++)
	{
		pos[i]=(i-1)/sz+1;
		a[i]=i;
		g[pos[i]].pb(i);
	}
	int ans=0;
	while(q--)
	{
		int l,r;
		cin>>l>>r;
		if(l>r)swap(l,r);
		if(l==r)
		{
			cout<<ans<<"\\n";
			continue;
		}
		if(a[l]<a[r])ans++;
		else ans--;
		ans+=(ask(l+1,r-1,a[r])-ask(l+1,r-1,a[l]))*2;
		int pl=pos[l],pr=pos[r];
		if(pl!=pr)
		{
			g[pl].erase(lower_bound(g[pl].begin(),g[pl].end(),a[l]));
			g[pr].erase(lower_bound(g[pr].begin(),g[pr].end(),a[r]));
			g[pl].insert(upper_bound(g[pl].begin(),g[pl].end(),a[r]),a[r]);
			g[pr].insert(upper_bound(g[pr].begin(),g[pr].end(),a[l]),a[l]);
		}
		swap(a[l],a[r]);
		cout<<ans<<"\\n";
	}
}