HDOJ 3555 Bomb (按位DP)
题目大意:求1到n中有多少个数含“49”
第一次接触按位DP,参考了别人的题解。
第一个代码运行速度快些(31ms),但状态的转移的分析比较麻烦,第二个代码要慢些(125ms),但code比较方便。
View Code
#include <stdio.h> #include <string.h> #define LEN 22 typedef __int64 LL; char s[LEN]; LL dp[LEN][3][2]; int next(int cur,int in) { if(cur==2 || cur==1&&in==9) return 2; return in==4; } LL DP(char s[],int len) { memset(dp,0,sizeof(dp)); dp[0][0][1]=1; for(int i=1;i<=len;i++) { dp[i][0][0]=9*dp[i-1][0][0]+8*dp[i-1][1][0]; dp[i][1][0]=dp[i-1][0][0]+dp[i-1][1][0]; dp[i][2][0]=10*dp[i-1][2][0]+dp[i-1][1][0]; for(int j=0;j<s[i]-'0';j++) { dp[i][next(0,j)][0]+=dp[i-1][0][1]; dp[i][next(1,j)][0]+=dp[i-1][1][1]; dp[i][next(2,j)][0]+=dp[i-1][2][1]; } dp[i][next(0,s[i]-'0')][1]+=dp[i-1][0][1]; dp[i][next(1,s[i]-'0')][1]+=dp[i-1][1][1]; dp[i][next(2,s[i]-'0')][1]+=dp[i-1][2][1]; } return dp[len][2][0]+dp[len][2][1]; } int main() { int t; scanf("%d",&t); while(t--) { scanf("%s",s+1); printf("%I64d\n",DP(s,strlen(s+1))); } return 0; }
View Code
#include <stdio.h> #include <string.h> #define LEN 22 typedef __int64 LL; char s[LEN]; LL dp[LEN][3][2]; int nextstate(int cur,int in) { if(cur==2 || cur==1&&in==9) return 2; return in==4; } int nextflag(int cur,int in,int max) { if(cur==1 && in==max) return 1; if(cur==1 && in>max) return -1; return 0; } LL DP(char s[],int len) { memset(dp,0,sizeof(dp)); dp[0][0][1]=1; for(int i=1;i<=len;i++) { for(int state=0;state<3;state++) { for(int flag=0;flag<2;flag++) { for(int j=0;j<10;j++) { if(nextflag(flag,j,s[i]-'0')==-1) continue; dp[i][nextstate(state,j)][nextflag(flag,j,s[i]-'0')]+=dp[i-1][state][flag]; } } } } return dp[len][2][0]+dp[len][2][1]; } int main() { freopen("in.txt","r",stdin); int t; scanf("%d",&t); while(t--) { scanf("%s",s+1); printf("%I64d\n",DP(s,strlen(s+1))); } return 0; }