AtCoder Regular Contest 097 - E - Sorted and Sorted （逆序对）

点击打开链接

题意：

给你黑白两种1-n编号的球。让你交换相邻的球，让黑i在黑i+1的上面。

百i在白i+1的上面。

POINT：

dp[x][y]代表排好了白1-x和黑1-y 编号的球。

dp[x][y]从dp[x-1][y]或dp[x][y-1]推出来。

用树状数组算出逆序对。

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<stack>
using namespace std;
const int maxn = 100000+55;
const int inf = 0x3f3f3f3f;
#define LL long long

int pos[2][2020];
int num[4020];
int dp[2020][2020];

int w;
void add(int x,int p) { for(int i=x;i<=w;i+=i&-i) num[i]+=p;}

int sum(int x){int ans=0; for(int i=x;i>=1;i-=i&-i) ans+=num[i]; return ans;}


int main()
{
	int n;
	scanf("%d",&n);
	w=2*n;
	for(int i=1;i<=w;i++){
		char c;int x;
		cin>>c>>x;
		pos[c=='W'][x]=i;
		add(i,1);
	}
	pos[0][0]=pos[1][0]=w+1;
	for(int i=0;i<=n;i++){
		add(pos[0][i],-1);
		for(int j=0;j<=n;j++){
			add(pos[1][j],-1);
			if(i||j){
				dp[i][j]=inf;
				if(i)
					dp[i][j]=min(dp[i][j],dp[i-1][j]+sum(pos[0][i]-1));
				if(j)
					dp[i][j]=min(dp[i][j],dp[i][j-1]+sum(pos[1][j]-1));
			}
		}
		for(int j=0;j<=n;j++){
			add(pos[1][j],1);
		}
	}
	printf("%d\n",dp[n][n]);
	return 0;
}