Gym - 101889E 记忆化搜索

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/du_lun/article/details/82933288

思维还是太将江华,开始一直想dp,就是这一位余数固定时取最小的一个字符串,但是字符串太大,赋值的时候超时,其实根本没必要存字符串,只要记忆化搜索,看看[pos][res]这个状态是否能构成就行,这样的话1000*1000个状态, 稳稳的,然后贪心从小枚举,第一个可行的一定是最优解。

#include<bits/stdc++.h>
using namespace std;
const int N=1e3+10;
char s[N], ans[N];
int k, n;
int dp[N][N];

int dfs(int pos, int res){
    if(pos==n)    return res==0;
    int& ret=dp[pos][res];
    if(ret!=-1) return ret;

    if(s[pos]=='?'){
        for(int i=0; i<10; i++){
            if(!pos&&!i) continue;
            ans[pos]=i+'0';
            ret=dfs(pos+1, (res*10+i)%k);
            if(ret==1) return ret;
        }
    }
    else{
        ans[pos]=s[pos];
        ret=dfs(pos+1, (res*10+s[pos]-'0')%k);
    }
    return ret;
}

int main(){
    scanf("%s", s);
    scanf("%d", &k);
    n=strlen(s);
    memset(dp, -1, sizeof dp);

    if(dfs(0,0)==1){
        ans[n]=0;
        for(int i=0; i<n; i++)
            putchar(ans[i]);
        puts("");
    }
    else
        puts("*");
    return 0;
}

猜你喜欢

转载自blog.csdn.net/du_lun/article/details/82933288