题目不难,但是直接暴力搜索会超时。
然后分析一下题目就知道是dp问题。
n = 0时只有0一个数;
即dp[0] = 1;
n = 1时有0—9一共十个数;
即dp[1] = 1 + 9 = 10;
n>1的时候,
首先有不到n位的所有结果之和,再加上n位数的结果。
继续举例子,n = 2时,有dp[1] + 两位符合要求的数;
考虑两位符合要求的数:
第一位不能为0,即9种选择;
第二位不能和第一位相同,也9种选择;
所以dp[2] = 1 + 9 + 9 * 9 = 91;
同理dp[3] = 91 + 9 * 9 * 8 = 739;
后面的类似于阶乘,发现阶乘是从9开始到(10 - n + 1)结束
在(10 - n + 1)前面的 恰好是n-1位的结果,所以可以表示为dp[n - 1] - dp[n - 2];
最后状态转移方程为:
dp[0] = 1;
dp[1] = 10;
dp[n] = dp[n - 1] + (dp[n - 1] - dp[n - 2]) * (10 - n + 1);
class Solution {
public:
int countNumbersWithUniqueDigits(int n) {
int ans[9];
ans[0] = 1;
ans[1] = 10;
for(int i = 2; i <= n; i ++){
ans[i] = ans[i - 1] + (ans[i - 1] - ans[i - 2]) * (10 - i + 1);
}
return ans[n];
}
};