codeforces 182E Wooden Fence 动态规划

http://www.codeforces.com/problemset/problem/182/E

给你n中篱笆，每种篱笆都有一个长和一个宽，设为a[i],b[i]，再给你一个长度L

问用这些篱笆的长可以组成多少长度为 L 的漂亮的篱笆

漂亮的定义： 1、相邻的两个篱笆要属于不同的种类

2、 后一个篱笆的长必须等于前一个篱笆的宽

注意：一个篱笆的两个边长都可以作为长

因为所选的每个篱笆都有两种状态，一种是用a[i]作为长，一种是用b[i]作为长

所以可以设dp[i][j][0]表示组成i长度的篱笆，最后一个为第j个篱笆，且第j个篱笆用a[i]作为长的方案数

同样dp[i][j][1]表示第j个篱笆用b[i]作为长

有了状态，转移过程就不太难了 j的前面必须有一个篱笆的宽等于j的长时才能转移

转移的时候还要注意一个篱笆的两个边长相等的情况

每次按要求 更新dp[i][j][0]和dp[i][j][1]即可

最后的答案是所有dp[L][i][0]和dp[L][i][1]的和，当然如果a[i]==b[i]，就只加一次

View Code

#include<cstdio>
#include<cstring>
const int mod = 1000000007;
int dp[3005][110][2];
int a[110],b[110];
int main()
{
int n,l,i,j,k;
while(scanf("%d%d",&n,&l)!=EOF)
{
for(i=1;i<=n;i++) scanf("%d%d",&a[i],&b[i]);
memset(dp,0,sizeof(dp));
for(i=1;i<=l;i++)
{
for(j=1;j<=n;j++)
{
if(a[j]==i) dp[i][j][0]++;
if(b[j]==i) dp[i][j][1]++;
}
}
for(i=1;i<=l;i++)
{
for(j=1;j<=n;j++)
{
for(k=1;k<=n;k++)
{
if(j==k) continue;
if(a[j]==a[k] && i>a[j] && dp[i-a[j]][k][1] ) dp[i][j][0]=(dp[i][j][0]+dp[i-a[j]][k][1])%mod;
if(a[j]==b[k] && i>a[j] && a[k]!=b[k] && dp[i-a[j]][k][0] ) dp[i][j][0]=(dp[i][j][0]+dp[i-a[j]][k][0])%mod;

if(b[j]==b[k] && i>b[j] && dp[i-b[j]][k][0] ) dp[i][j][1]=(dp[i][j][1]+dp[i-b[j]][k][0])%mod;
if(b[j]==a[k] && i>b[j] && a[k]!=b[k] && dp[i-b[j]][k][1] ) dp[i][j][1]=(dp[i][j][1]+dp[i-b[j]][k][1])%mod;
}
}
}
__int64 ans=0;
for(i=1;i<=n;i++)
{
ans+=dp[l][i][0]; ans%=mod;
if(a[i]!=b[i])
ans+=dp[l][i][1]; ans%=mod;
}
printf("%I64d\n",ans);
}
}