POJ 3373 Changing Digits 好蛋疼的DP

来源：互联网发布：谷歌浏览器 mac版编辑：程序博客网时间：2024/04/30 00:27

一开始写的高位往低位递推，发现这样有些时候保证不了第四条要求。于是又开始写高位往低位的记忆化搜索，又发现传参什么的蛋疼的要死。然后又发现高位开始的记忆化搜索就是从低位往高位的递推呀，遂过之。

dp[i][j]记录在i位且余数为j时的最优解情况。

dp[i][j].next表示当前的最优解是由哪一种状态转移过来的。

代码又写锉了。。

#include <algorithm>#include <iostream>#include <cstring>#include <cstdlib>#include <cstdio>#include <queue>#include <cmath>#include <stack>#include <map>#pragma comment(linker, "/STACK:1024000000");#define EPS (1e-8)#define LL long long#define ULL unsigned long long#define _LL __int64#define _INF 0x3f3f3f3f#define Mod 9999991using namespace std;struct N{    bool mark;    int ans,dig,next;}dp[111][10000];char s[110];int mod1[110],mod2[110];void Out(int site,int mod){    if(mod == -1)        return ;    printf("%d",dp[site][mod].dig);    Out(site+1,dp[site][mod].next);}int main(){    int k,i,j,l;    while(scanf("%s %d",s+1,&k) != EOF)    {        for(i = 1; s[i] != '\0'; ++i)        {            for(j = 0; j < k; ++j)                dp[i][j].mark = false;        }        mod1[0] = 0,mod2[0] = 1%k;        for(i = 1;s[i] != '\0'; ++i)        {            mod1[i] = (mod1[i-1]*10 + s[i]-'0')%k;            mod2[i] = (mod2[i-1]*10)%k;        }        int site = strlen(s+1);        for(i = 0 ;i <= 9 ; ++i)        {            if(dp[site][i%k].mark == false)            {                dp[site][i%k].mark = true;                dp[site][i%k].ans = (i == s[site]-'0' ?  0 : 1);                dp[site][i%k].dig = i;                dp[site][i%k].next = -1;            }            else            {                if((i == s[site]-'0' ?  0 : 1) < dp[site][i%k].ans)                {                    dp[site][i%k].ans = (i == s[site]-'0' ?  0 : 1);                    dp[site][i%k].dig = i;                }                else if((i == s[site]-'0' ?  0 : 1) == dp[site][i%k].ans && i <= dp[site][i%k].dig)                {                    dp[site][i%k].dig = i;                }            }        }        int mod;        int len = strlen(s+1);        for(--site ; site >= 1 ; --site)        {            mod = 0;            for(i = (site == 1 ? 1 : 0);i <= 9; ++i)            {                for(j = 0;j < k; ++j)                {                    if(dp[site+1][j].mark == true)                    {                        if(dp[site][(mod + i*mod2[len-site] + j)%k].mark == false)                        {                            dp[site][(mod + i*mod2[len-site] + j)%k].mark = true;                            dp[site][(mod + i*mod2[len-site] + j)%k].ans = dp[site+1][j].ans+(s[site]-'0' == i ? 0 :1);                            dp[site][(mod + i*mod2[len-site] + j)%k].dig = i;                            dp[site][(mod + i*mod2[len-site] + j)%k].next = j;                        }                        else                        {                            if(dp[site+1][j].ans+(s[site]-'0' == i ? 0 :1) < dp[site][(mod + i*mod2[len-site] + j)%k].ans)                            {                                dp[site][(mod + i*mod2[len-site] + j)%k].ans = dp[site+1][j].ans+(s[site]-'0' == i ? 0 :1);                                dp[site][(mod + i*mod2[len-site] + j)%k].dig = i;                                dp[site][(mod + i*mod2[len-site] + j)%k].next = j;                            }                            else if(dp[site+1][j].ans+(s[site]-'0' == i ? 0 :1) == dp[site][(mod + i*mod2[len-site] + j)%k].ans && i < dp[site][(mod + i*mod2[len-site] + j)%k].dig)                            {                                dp[site][(mod + i*mod2[len-site] + j)%k].dig = i;                                dp[site][(mod + i*mod2[len-site] + j)%k].next = j;                            }                        }                    }                }            }        }        Out(1,0);        puts("");//        cout<<len<<endl;//        for(i = 1;i <= len; ++i)//        {//            for(j = 0;j < k; ++j)//            {//                if(dp[i][j].mark)//                    printf("%2d ",dp[i][j].next);//                else//                    printf("   ");//            }//            puts(" * ");//        }////        puts("");////        for(i = 1;i <= len; ++i)//        {//            for(j = 0;j < k; ++j)//            {//                if(dp[i][j].mark)//                    printf("%2d ",dp[i][j].dig);//                else//                    printf("   ");//            }//            puts(" * ");//        }////        puts("");//////        for(i = 1;i <= len; ++i)//        {//            for(j = 0;j < k; ++j)//            {//                if(dp[i][j].mark)//                    printf("%2d ",dp[i][j].ans);//                else//                    printf("   ");//            }//            puts(" * ");//        }    }    return 0;}

0 0