1. 程式人生 > >bzoj1026: [SCOI2009]windy數 數位dp

bzoj1026: [SCOI2009]windy數 數位dp

windy數 class tar 搜索 abs www main pos 之間

題目:

http://www.lydsy.com/JudgeOnline/problem.php?id=1026

題意:

Description
windy定義了一種windy數。不含前導零且相鄰兩個數字之差至少為2的正整數被稱為windy數。 windy想知道,
在A和B之間,包括A和B,總共有多少個windy數?

Input
包含兩個整數,A B。

Output
一個整數

思路:

數位dp,記憶化搜索。

 1 #include <bits/stdc++.h>
 2 
 3 using namespace std;
 4 
 5 const int N = 30 + 10;
 6 
 7 int
dp[N][10];//定義dp[i][j]為第i位前驅數字為j時的方案數,註意前驅要合法 8 int dig[N]; 9 int tot = 0; 10 int dfs(int pos, int pre, bool f, bool limit) 11 {//f用來標記是否有合法前驅 12 if(pos < 1) return 1; 13 if(!limit && f && dp[pos][pre] != -1) return dp[pos][pre]; 14 int en = limit ? dig[pos] : 9; 15 int
ans = 0; 16 for(int i = 0; i <= en; i++) 17 if(! f) ans += dfs(pos-1, i, f || i != 0, limit && i == en); 18 else if(abs(i - pre) >= 2) ans += dfs(pos-1, i, f || i != 0, limit && i == en); 19 if(! limit && f) dp[pos][pre] = ans; 20 return ans; 21 }
22 int work(int n) 23 { 24 int tn = n; 25 tot = 0; 26 while(tn) dig[++tot] = tn % 10, tn /= 10; 27 memset(dp, -1, sizeof dp); 28 return dfs(tot, 0, 0, 1); 29 } 30 int main() 31 { 32 int n, m; 33 while(~ scanf("%d%d", &n, &m)) 34 printf("%d\n", work(m) - work(n-1)); 35 return 0; 36 }

  

bzoj1026: [SCOI2009]windy數 數位dp