BZOJ1037 DP

2013-11-15 21:51

原题传送门http://www.lydsy.com/JudgeOnline/problem.php?id=1037

拿到这道题想到了DP，后来发现三维无法确定的表示状态（起码我是表示不出来）

然后就用四维表示状态，w[i,j,k1,k2]表示一共i个人，j个男生，前i-1个人中任意一个人到第i个人的区间里，男生比女生最多多k1个人且存在k1的情况，

女生比男生最多多k2个人且存在k2的情况

那么w[i,j,k1,k2]这个状态可以更新w[i+1,j+1,k1+1,max(k2-1,0)]表示第i+1个位置放的男生，w[i+1,j,max(k1-1)，k2+1]表示放的女生，那么就可以转移了。

//By BLADEVIL
const
d39     =12345678;

var
n, m, k     :longint;
i, j, k1, k2     :longint;
w     :array[0..310,0..160,0..20,0..20] of longint;
ans     :longint;
function max(a,b:longint):longint;
begin
if a>b then max:=a else max:=b;
end;
begin
read(n,m,k);
w[1,1,1,0]:=1;
w[1,0,0,1]:=1;
for i:=1 to n+m-1 do
for j:=0 to i do
for k1:=0 to k do
for k2:=0 to k do
begin
if w[i,j,k1,k2]=0 then continue;
if (k1<k) and (j<n) then
w[i+1,j+1,k1+1,max(k2-1,0)]:=(w[i+1,j+1,k1+1,max(k2-1,0)]+w[i,j,k1,k2]) mod d39;
if (k2<k) and (i-j<m) then
w[i+1,j,max(k1-1,0),k2+1]:=(w[i+1,j,max(k1-1,0),k2+1]+w[i,j,k1,k2]) mod d39;
end;
ans:=0;
for i:=0 to k do
for j:=0 to k do ans:=(ans+w[n+m,n,i,j]) mod d39;
writeln(ans);
end.