NKOJ(5617)黑白球

$NKOJ(5617)$黑白球

设置状态$f[i][j]$表示在前$i+j$个位置已经放好了$i$个白球和$j$个黑球所需的最小步数
方程：
f [ i ] [ j ] = min ⁡ { f [ i − 1 ] [ j ] + c o s t w ( i , j ) , f [ i ] [ j − 1 ] + c o s t b ( i , j ) } f[i][j]=\min\{f[i-1][j]+costw(i,j),f[i][j-1]+costb(i,j)\} f[i][j]=min{f[i−1][j]+costw(i,j),f[i][j−1]+costb(i,j)}
那么$costw$与$costb$怎么计算呢
对于$costw(i,j)$对于$\forall k<i$,每一个在$i$右边的白球都要移到左边去。对于$\forall k⩽j$每一个在$i$右边的黑球也都要移到左边去。
所以可以先处理。
上代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=2000+5;
int n;
int f[maxn][maxn];
int pw[maxn],pb[maxn],cntw[maxn][maxn],cntb[maxn][maxn];
int main()
{
	memset(f,0x7f,sizeof(f));
	cin>>n;
	char c;
	int x;
	for(int i=1;i<=2*n;i++)
	{
		cin>>c>>x;
		if(c=='B')pb[x]=i;
		else pw[x]=i;
	}
	for(int i=0;i<=n;i++)
	{
		for(int j=1;j<i;j++)if(pw[i]<pw[j])cntw[i][0]++;
		for(int j=1;j<=n;j++)cntw[i][j]=cntw[i][j-1]+(pw[i]<pb[j]);
	}
	for(int i=0;i<=n;i++)
	{
		for(int j=1;j<i;j++)if(pb[i]<pb[j])cntb[0][i]++;
		for(int j=1;j<=n;j++)cntb[j][i]=cntb[j-1][i]+(pb[i]<pw[j]);
	}
	f[0][0]=0;
	for(int i=0;i<=n;i++)
	for(int j=0;j<=n;j++)
	{
		if(i)f[i][j]=min(f[i-1][j]+cntw[i][j],f[i][j]);
		if(j)f[i][j]=min(f[i][j-1]+cntb[i][j],f[i][j]);
	}
	cout<<f[n][n];
	return 0;
}