火柴棒的题目在ACM里面我觉得是很常见的了,这一次居然在DP里面遇见了。
题目大意是用少于等于n根的火柴,拼出能够被m整除的最大的数。
第一次见这样的状态定义,学习学习.....
定义状态dp[i][j]表示在已经用了i根火柴的情况下拼出来了剩余部分(是剩余部分,不是已经拼出来了的)为j(需要%m)的最大长度,一个辅助数组p[i][j]表示状态[i][j]的最高位
注:此处的最高以右边为高。因此,一个合法的状态转移就是:对于一个数字d
dp[i][j]=dp[i-need[d]][(j*10+d)%m]+1,p[i][j]=d(要求状态dp[i-need[d]][(j*10+d)%m]合法且+1之后大于dp[i][j])
need[k]表示数字k需要的火柴数
在实际的程序实现当中还要注意表示状态不存在。
这样表示的话重构解的时候也方便一些。
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int kcase=1,n,m,dp[105][3005],p[105][3005],te;
const int need[]{6,2,5,5,4,5,6,3,7,6};
int main(){
while(scanf("%d%d",&n,&m)==2){
printf("Case %d: ",kcase++);
for(int i=0;i<=n;++i)
for(int j=0;j<m;++j){
dp[i][j]=p[i][j]=-1;
if(!j)
dp[i][j]=0;
for(int d=9;d>=0;--d){
if(i>=need[d]){
te=dp[i-need[d]][(j*10+d)%m];
if(te>=0&&te+1>dp[i][j])
dp[i][j]=te+1,p[i][j]=d;
}
}
}
if(p[n][0]<0)
printf("-1\n");
else{
int i=n,j=0;
while(dp[i][j]!=0)
printf("%d",p[te=i][j]),i-=need[p[i][j]],j=(j*10+p[te][j])%m;
printf("\n");
}
}
return 0;
}
在枚举数字d的时候可以发现,循环是从高向低的,这样写的原因就是保证是最大的数,假设是从低到高的话,0和9用的火柴一样多,按照我们的转移的写法,把0更新掉之后,9就没办法更新了,(假如状态的前驱的值一样的话),否则我们还需要特判这一种情况来保证是最优解,从高到低的话就可以避免这一种情况的发生了。
还是一道挺不错的题目的。