F - Sorting Color Balls ( 树状数组 )

题目链接

这个题感觉比E简单的多…

分析

不考虑颜色的影响,问题就是:
给一个数组,每次只能交换相邻的元素,问对数组排序的价值;
很经典的就是求原数组中逆序对的数量,树状数组搞一下就行了;

考虑颜色的影响:
很显然的,最终答案就是 = 总的逆序对数量 - 相同颜色逆序对的数量;
所以还是一样的问题, 求逆序对, 直接树状数组搞就行了;

每次求完一种颜色的后, 我们直接在树状数组上-1消除影响就行,不用重开,不然很慢;

代码

#include<iostream>
#include<queue>
#include<cstring>
#include<vector>
#include<stdio.h>
#include<map>
#include<algorithm>
#include<deque>
#include<stack>
#include<set>
#include <random>
#include<bitset>
// #include <unordered_map>
#include<math.h>
#include<string.h>
#define IOS ios::sync_with_stdio(false),cin.tie(0);
using namespace std;
 
#define pb push_back
#define coutl cout<<"------------"<<endl;
#define fi first
#define se second

#define ire(x) scanf("%d",&x)
#define iire(a,b) scanf("%d %d",&a,&b)
#define lre(x) scanf("%lld",&x)
#define llre(a,b) scanf("%lld %lld",&a,&b)
#define dre(x) scanf("%lf",&x)
#define ddre(a,b) scanf("%lf %lf",&a,&b)
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define endl "\n"
#define PI acos(-1.0)
// #define int long long
// #define double long double
// typedef __int128 ll;
typedef long long ll;
typedef unsigned long long ull;

typedef pair<int, int> PII;
typedef pair<double, int> PDI;
typedef pair<ll, ll> PLL;
typedef pair<double, double> PDD;
typedef pair<double, pair<int, double> > PDID;
typedef pair<char, char> PCC;
typedef pair<char, pair<int, int> > PCII;
typedef pair<int, pair<int, int> > PIII;
typedef pair<int, pair<int, pair<int, int> > > PIIII;
typedef pair<ll, pair<int, int> > PLII;

const int maxn = 3e5 + 7;
const int N = 405;
const int M = 4e6 + 7;
const int mod = 998244353;
const int inv = mod - mod/2;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const double pi = acos(-1);
const double eps = 1e-8;
  
ll gcd(ll a,ll b) {return b==0 ? a : gcd(b,a%b);}
ll lcm(ll a,ll b) {return a*b / gcd(a,b);}
ll qmi(ll a,ll b,ll p) {ll ans = 1; while(b) { if(b & 1) ans = ans * a % p; a = a * a % p; b >>= 1; } return ans;}
int lowbit(int x) {return x & (-x);}

int c[maxn];
int a[maxn];
int n;

vector<int> g[maxn];	//每种颜色的值

int tr[maxn];

void update(int x,int v)
{
	while(x <= n)
	{
		tr[x] += v;
		x += lowbit(x);
	}
}

int query(int x)
{
	int ans = 0;
	while(x)
	{
		ans += tr[x];
		x -= lowbit(x);	
	}
	return ans;
}

void solve()
{
	cin>>n;
	for(int i=1;i<=n;i++) cin>>c[i];
	for(int i=1;i<=n;i++) cin>>a[i];
	
	for(int i=1;i<=n;i++) g[c[i]].push_back(a[i]);
	
	//总的逆序对数量
	ll ans = 0;
	for(int i=1;i<=n;i++)
	{
		ans += query(n) - query(a[i]);
		update(a[i],1);
	}
	
	for(int i=1;i<=n;i++) update(a[i],-1);	//消除影响
	
	ll cnt = 0;
	for(int i=1;i<=n;i++)	//每种颜色
	{
		for(auto x : g[i])
		{
			cnt += query(n) - query(x);
			update(x,1);
		}
		
		for(auto x : g[i]) update(x,-1);	//消除影响(清空树状数组)
	}
	
	cout<<ans-cnt<<'\n';
}

int main()
{
	IOS;
	
	int t;
//	ire(t);
	t = 1;
	while(t--)
	{
		solve();
	}
	
	return 0;
}