您的位置:首页 > 其它

数位DP的记忆化搜索形式

2015-03-16 18:51 141 查看
数位DP的记忆化搜索形式比一般递推形式好写多了,而且一般不容易出错。

dfs求[0,n]有多少个符合的,先把n换成字符串形式。

cur:现在处理到哪一位。

s:搜索到目前为止,之前的状态,具体什么状态看情况而定。

e:是否到达边界,如果没到这一位只取到9,否则只能取到bit[cur]。

z:前导0标记,也就是是否前面的都是0,有些情况需要特殊判断。

一般情况下dp数组开二维,dp[cur][s]为e和z都为0的情况,因为e和z都为0的情况比较多。特殊情况开三维什么的都是可能的。

当cur<0时判断当前状态是否符合,符合返回1,不符合返回0。

每次根据s和当前的取值i计算出新的s。

int dfs(int cur,int s,int e,int z){
if(cur<0) return check(s);
if(!e&&!z&&dp[cur][s]!=-1) return dp[cur][s];
int end=e?bit[cur]:9;
int ans=0;
for(int i=0;i<=end;i++){
if(z&&!i) ans+=dfs(cur-1,0,e&&i==end,1);
else ans+=dfs(cur-1,get_news(s,i),e&&i==end,0);
}
if(!e&&!z) dp[cur][s]=ans;
return ans;
}

int solve(int n){
int len=0;
while(n){
bit[len++]=n%10;
n/=10;
}
return dfs(len-1,0,1,1);
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: