bzoj1026: [SCOI2009]windy數 數位dp
阿新 • • 發佈:2017-09-15
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 intdp[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 intans = 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