NKOJ(5617)NKOJ(5617)NKOJ(5617)黑白球
设置状态f[i][j]f[i][j]f[i][j]表示在前i+ji+ji+j个位置已经放好了iii个白球和jjj个黑球所需的最小步数
方程:
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)\}f[i][j]=min{f[i−1][j]+costw(i,j),f[i][j−1]+costb(i,j)}
那么costwcostwcostw与costbcostbcostb怎么计算呢
对于costw(i,j)costw(i,j)costw(i,j)对于∀k<i\forall k<i∀k<i,每一个在iii右边的白球都要移到左边去。对于∀k⩽j\forall k\leqslant j∀k⩽j每一个在iii右边的黑球也都要移到左边去。
所以可以先处理。
上代码
#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;
}