每日算法系列【LeetCode 233】数字 1 的个数

简介: 每日算法系列【LeetCode 233】数字 1 的个数

题目描述

给定一个整数 n,计算所有小于等于 n 的非负整数中数字 1 出现的个数。

示例1

输入:
13
输出:
6
解释:
数字 1 出现在以下数字中: 1, 10, 11, 12, 13 。

题解

这题是我搜数位 dp 题目搜出来的,于是我直接用数位 dp 方法把它过了,后来发现其实没必要这么麻烦,简单的计算就能算出来了,这里两个方法我都讲一下。

数学方法

我们不妨用 n = 12345 来举个例子。要求小于等于 n 的数字里有多少个 1 ,我们不妨转换个角度,看某一位数字是 1 的话,有多少数字小于 n 。

例如从右向左数第 i = 2 位(数字 3 ),如果这一位取 1 ,那么左边 2 位如果取 0~11 ,那么右边 2 位就没有任何限制,从 0 取到 99 都行。如果左边 2 位如果取 12 ,那么就得考虑 n 中第 i 位是几了,如果大于 1 ,那么右边 2 位还是没有限制;如果等于 1 ,那么右边 2 位只能取 0~45 ;如果等于 0 ,那就没得取了。

下面这张图是我打的草稿,看的更清楚一点:

一般化描述就是,考虑从右往左数第 i 位是 1 的数字数量。那么 n 中第 i 位左边部分的数字是  ,而右边可以取的数量是  ,相乘就是总的数量  。如果左边直接取最大值,那么就要考虑第 i 位数字是几了,计算可以得到第 i 位数字为  ,记为 x 。如果  ,那么右边无限制,有  种取法;如果  ,那么右边有  种取法;如果  ,那么右边无法取,因为第 i 位都没法取 1 。

综上,令  ,那么答案就是:

数位dp

数位 dp 就麻烦许多了,不想看的可以直接跳过了。

首先我们从最高位开始往右递归计算,用 pos, count, limit 来表示计算到第 pos 位(从左往右,和数学方法不一样)时,已经出现了 count 个 1 ,并且之后的数字有无限制(也就是能否取遍 0~9 ),这种状态之下方法数是多少。

那么第 pos 位我们可以取的数字有哪些呢?如果 limit = 1 也就是有限制,那么只能取 0~n中第pos位,如果没有限制那就取 0~9 。

假设第 pos 位取 1 ,那么 pos 就转移到了 pos+1 ,count 转移到了 count+1 ,limit 呢?只有当原来有限制,并且第 pos 位正好取了最大值也就是 n 中第 pos 位数字时,limit 还是 1 ,否则的话限制取消,后面的数字随便取。如果第 pos 位不取 1 ,那么除了 count 不变以外,其他两个状态还是跟上面一样转移。

终止状态的话,如果遍历到了最后一位结束,就返回 count 数量就行了,表示当前数字中有 count 个 1 。

这样的话会有很多重复计算的状态,所以需要用到记忆化搜索,用 dp[pos][count] 来保存 pos, count, limit=0 状态下的答案。为什么只保存 limit=0 的答案呢?因为只有无限制的情况下,后面的数字才能随便取,跟 n 是多少没有关系。否则的话 n 变了后面的值就会受限于 n ,那么就不是一个定值了,没法保存。

那么 limit=1 不保存的话会不会超时呢?不会的,因为每一位只有一种取法会使得后面的数字继续有限制,所以整体上来看,有限制的状态个数是个常数,并不需要担心超时。

代码

数学方法(c++)

class Solution {
public:
    int countDigitOne(int n) {
        int res = 0;
        for (long i = 1; i <= n; i *= 10) {
            res += n / (i * 10) * i;
            int x = (n / i) % 10;
            res += x > 1 ? i : (n % i + 1) * x;
        }
        return res;
    }
};

数位dp(c++)

class Solution {
public:
    int a[14];
    int dp[14][14];
    int dfs(int pos, int count, int limit) {
        if (!pos) return count;
        if (!limit && dp[pos][count] != -1) return dp[pos][count];
        int res = 0, ub = limit ? a[pos] : 9;
        for (int i = 0; i <= ub; ++i) {
            res += dfs(pos-1, count+(i==1), limit&&i==a[pos]);
        }
        return limit ? res : dp[pos][count]=res;
    }
    int countDigitOne(int n) {
        memset(dp, -1, sizeof dp);
        int len = 0;
        while (n) {
            a[++len] = n % 10;
            n /= 10;
        }
        return dfs(len, 0, 1);
    }
};

数学方法(python)

class Solution:
    def countDigitOne(self, n: int) -> int:
        res, i = 0, 1
        while i <= n:
            res += n // (i * 10) * i
            x = (n // i) % 10
            res += i if x > 1 else (n % i + 1) * x
            i *= 10
        return res
相关文章
|
1月前
|
算法
Leetcode 初级算法 --- 数组篇
Leetcode 初级算法 --- 数组篇
38 0
|
1天前
|
存储 算法 Java
leetcode算法题-有效的括号(简单)
【11月更文挑战第5天】本文介绍了 LeetCode 上“有效的括号”这道题的解法。题目要求判断一个只包含括号字符的字符串是否有效。有效字符串需满足左括号必须用相同类型的右括号闭合,并且左括号必须以正确的顺序闭合。解题思路是使用栈数据结构,遍历字符串时将左括号压入栈中,遇到右括号时检查栈顶元素是否匹配。最后根据栈是否为空来判断字符串中的括号是否有效。示例代码包括 Python 和 Java 版本。
|
23天前
|
算法
每日一道算法题(Leetcode 20)
每日一道算法题(Leetcode 20)
21 2
|
3月前
|
算法 Java
LeetCode经典算法题:矩阵中省份数量经典题目+三角形最大周长java多种解法详解
LeetCode经典算法题:矩阵中省份数量经典题目+三角形最大周长java多种解法详解
49 6
|
3月前
|
人工智能 算法 Java
LeetCode经典算法题:井字游戏+优势洗牌+Dota2参议院java解法
LeetCode经典算法题:井字游戏+优势洗牌+Dota2参议院java解法
48 1
|
3月前
|
存储 算法 Java
LeetCode经典算法题:预测赢家+香槟塔java解法
LeetCode经典算法题:预测赢家+香槟塔java解法
53 1
|
3月前
|
存储 算法 Java
LeetCode经典算法题:二叉树遍历(递归遍历+迭代遍历+层序遍历)以及线索二叉树java详解
LeetCode经典算法题:二叉树遍历(递归遍历+迭代遍历+层序遍历)以及线索二叉树java详解
77 0
|
3月前
|
算法 Java
LeetCode初级算法题:子数组最大平均数+二叉树的最小深度+最长连续递增序列+柠檬水找零
LeetCode初级算法题:子数组最大平均数+二叉树的最小深度+最长连续递增序列+柠檬水找零
40 0
|
3月前
|
算法 Java
LeetCode初级算法题:环形链表+排列硬币+合并两个有序数组java解法
LeetCode初级算法题:环形链表+排列硬币+合并两个有序数组java解法
54 0
|
3月前
|
存储 算法 Java
LeetCode初级算法题:两数之和+斐波拉契数列多种java解法
LeetCode初级算法题:两数之和+斐波拉契数列多种java解法
36 0