O(nsqrt n)求条
查看原帖
O(nsqrt n)求条
752953
sLMxf楼主2025/6/13 19:59

rt

#include<bits/stdc++.h>
#define lowbit(x) (x&(-x))
using namespace std;
int a[1000006],id[1000006],ll[1000006],b[1000006],tr[1000006],xx[1000006];
void update(int w,int c)
{
	int W=id[w];
	a[w]=c;
	for(int i=ll[W];i<ll[W+1];i++) b[i]=a[i];
	sort(b+ll[W],b+ll[W+1]);
}
int query1(int l,int r,int c)
{
	if(l>r) return 0;
	int L=id[l],R=id[r],ans=0;
	if(L==R)
	{
		for(int i=l;i<=r;i++) ans+=(a[i]>c);
		return ans;
	}
	else
	{
		for(int i=l;i<ll[L+1];i++) ans+=(a[i]>c);
		for(int i=ll[R];i<=r;i++) ans+=(a[i]>c);
		for(int i=L+1;i<R;i++) ans+=ll[i+1]-(upper_bound(b+ll[i],b+ll[i+1],c)-b)+1;
		return ans;
	}
}
int query2(int l,int r,int c)
{
	if(l>r) return 0;
	int L=id[l],R=id[r],ans=0;
	if(L==R)
	{
		for(int i=l;i<=r;i++) ans+=(a[i]<c);
		return ans;
	}
	else
	{
		for(int i=l;i<ll[L+1];i++) ans+=(a[i]<c);
		for(int i=ll[R];i<=r;i++) ans+=(a[i]<c);
		for(int i=L+1;i<R;i++) ans+=(lower_bound(b+ll[i],b+ll[i+1],c)-b)-ll[i];
		return ans;
	}
}
int n;
struct node{
	int num,i;
}c[1000006];
bool cmp(node A,node B)
{
	if(A.num!=B.num) return A.num<B.num;
	else return A.i<B.i;
}
void add(int x,int k)
{
	while(x<=n)
	{
		tr[x]+=k;
		x+=lowbit(x);
	}
}
int sum(int x)
{
	int ans=0;
	while(x!=0)
	{
		ans+=tr[x];
		x-=lowbit(x);
	}
	return ans;
}
signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
	int q,cnt=0,ans=0;
	cin>>n;
	int kc=sqrt(n*log2(n));
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];c[i].num=a[i];
		c[i].i=i;
		id[i]=(i-1)/kc+1;
		b[i]=a[i];
		if(id[i]!=id[i-1]) ll[++cnt]=i;
	}
	ll[++cnt]=n+1;
	for(int i=1;i<cnt;i++) sort(b+ll[i],b+ll[i+1]);
	
	sort(c+1,c+n+1,cmp);
	for(int i=1;i<=n;i++) xx[c[i].i]=i;
	for(int i=1;i<=n;i++)
	{
		add(xx[i],1);
		ans+=i-sum(xx[i]);
	}
	
	cout<<ans<<'\n';
	cin>>q;
	while(q--)
	{
		int l,r,x,y;
		cin>>l>>r;
		if(l>r) swap(l,r);
		x=a[l],y=a[r];
		update(l,y);
		update(r,x);
		if(x==y)
		{
			cout<<ans<<'\n';
			continue;
		}
		if(x<y) ans++;
		else ans--;
		ans+=query1(l+1,r-1,x)-query2(l+1,r-1,x)-query1(l+1,r-1,y)+query2(l+1,r-1,y);
		cout<<ans<<'\n';
	}
	return 0;
}
2025/6/13 19:59
加载中...