思路:只有题目中出现2 5 6 9才可能是好数,如果出现了3 4 7,那么直接返回false。
数据范围是[1, 10000],所以暴力枚举可以过。
class Solution {
public:
bool ok(int num){
//统计数字中是否出现2 5 6 9,如果可以,那么才可能是好数
int cnt = 0;
while(num){
int x = num % 10;
num /=10;
if(x == 2 || x == 5 || x == 6 || x == 9)
cnt++;
else if(x == 0 || x == 1 || x == 8)
continue;
else
return false;
}
return cnt > 0;
}
int rotatedDigits(int n) {
int res = 0;
for(int i = 1; i <= n; i++){
if(ok(i))
res++;
}
return res;
}
};
做完看了一下别人的题解发现可以用数位dp做。思路就是将数字转化为字符串,然后声明一个dp数组来记忆化搜索过的状态。
设置一个dif数组对10个数字进行分类。
class Solution {
public:
vector<vector<int>> dp;
string str;
int len;
int dif[10] = {0, 0, 1, -1, -1, 1, 1, -1, 0, 1};
/*
@is_limit当前位置是否受到前一位的限制
@has_diffs是否含有2 5 6 9其中的一位
*/
int dps(int index, bool is_limit, int has_diffs){
if(index == len) return has_diffs;
//只有无限制的才是搜索完的,才可以返回
if(!is_limit && dp[index][has_diffs] >= 0) return dp[index][has_diffs];
int res = 0;
int up = is_limit ? str[index] - '0' : 9;
for(int i = 0; i <= up; i++)
if(dif[i] != -1) //该位置不是 3 4 7
res += dps(index+1, is_limit && i == up, has_diffs || dif[i]);
if(!is_limit) dp[index][has_diffs] = res;
return res;
}
int rotatedDigits(int n) {
str = to_string(n);
len = str.size();
dp.resize(len, vector<int>(2,-1));
return dps(0, true, 0);
}
};