求调
查看原帖
求调
701387
xiaoliebao1115楼主2024/7/18 21:56

前面是树状数组求逆序对,后面借鉴了一下第一篇题解的写法,RE+WA

#include<bits/stdc++.h>
using namespace std;
const int nn=2e4+5;
int n;
struct node
{
	int h,id,nh;
};
node x[nn];
bool cmp(node l1,node l2)
{
	return l1.h<l2.h;
}
bool cmp2(node l1,node l2)
{
	return l1.id<l2.id;
}
int tree[nn],sum,m;
int query(int x)
{
	int ans=0;
	for(int i=x;i>=1;i-=i&(-i)){
		ans+=tree[i];
	}
	return ans;
}
void add(int x)
{
	for(int i=x;i<=n;i+=i&(-i)){
		tree[i]++;
	}
}
int main()
{
	std::ios::sync_with_stdio(false),std::cin.tie(nullptr);
	cin>>n;
	for(int i=1;i<=n;i++) cin>>x[i].h,x[i].id=i;
	cin>>m;
	sort(x+1,x+n+1,cmp);
	for(int i=1;i<=n;i++){
		if(x[i].h==x[i-1].h){
			x[i].nh=x[i-1].nh;
		}
		else x[i].nh=x[i-1].nh+1;
	}
	sort(x+1,x+n+1,cmp2);
	for(int i=1;i<=n;i++) cout<<x[i].nh<<" ";
	cout<<endl;
	for(int i=n;i>=1;i--){
		sum+=query(x[i].nh-1);
		add(x[i].nh);
	}
	cout<<sum<<endl;
	int ans=sum;
	int j;
	 for(int i=1;i<=m;i++){
	 	int a,b;
        scanf("%d%d",&a,&b);
        if(a>b){
            j=a;
            a=b;
            b=j;
        }
        if(x[b].nh>x[a].nh)
            ans++;
        else if(x[b].nh<x[a].nh)
            ans--;
        for(j=a+1;j<=b-1;j++){
            if(x[j].nh>x[a].nh)
                ans++;
            else if(x[j].nh<x[a].nh)
                ans--;
            if(x[j].nh<x[b].nh)
                ans++;
            else if(x[j].nh>x[b].nh)
                ans--;
        }
        j=x[a].nh;
        x[a].nh=x[b].nh;
        x[b].nh=j;
        printf("%d\n",ans);
    }
	return 0;
}
2024/7/18 21:56
加载中...