题解 | 「力扣」第 233 题:数字 1 的个数(困难、动态规划)

参考资料 1:https://leetcode-cn.com/problems/number-of-digit-one/solution/zhu-wei-ji-suan-so-easy-by-faith_previal-7lep/

参考资料 2:https://leetcode-cn.com/problems/number-of-digit-one/solution/shu-wei-dp-liang-ge-yi-wei-dp-tablejie-j-7bai/

所有的讨论,认为有前导 0,例如 10 看成 010。

整的区间段(例如 099、0999)可以预先求出

以求解 666 为例,可以将 666 拆分为分别求解:

  • 600~666
  • 500~599(预先求出,这些值是确定的)
  • 400~499(预先求出,这些值是确定的)
  • 300~399(预先求出,这些值是确定的)
  • 200~299(预先求出,这些值是确定的)
  • 100~199(预先求出,这些值是确定的)
  • 0~99 的数字 1 的个数(预先求出)

最后求和得到结果。

  • 第 1 类(整数区间段):对于 500599、400499、300399、200299、0~99,它们的最高位没有 1 ,因此它们的 1 个个数取决于 0~99 中 1 的个数(找到了大问题和小问题之间的联系)。
  • 第 2 类(最高位是 1,每一个数都一定有 1):对于 100~199 ,因为百位数是1,所以 100~199 每个数中都有 1,因此等于 100 加上 0~99 中 1 的个数
  • 第 3 类(不满足一个整的区间段):对于 600~666,1 的个数取决于后两位中 1 的个数,就是 0~66 中 1 的个数,这件事情有点麻烦,后面再说。

对于一个数字,假设从个位到最高位标号为 1~n, 那么需要求出

也就是说,09、099、0999、09999、...、0 ~ 10^{N-1}-1 包含 1 的个数,这里 N 为数字 n 的长度。


分别记为 S1S2S3、... 、S_{n-1} 中包含 1 个数。

以 666 为例,那么 S1 就是 0~6 中包含 1 的个数、S2 就是 0~66 中包含 1 的个数

首先定义 dp 数组 dp1[i] 表示 0~10^{i+1}-1 中包含 1 的个数,例如

  • dp1[0] 表示求 0~9 中包含 1 的个数,
  • dp1[1] 表示求 0~99 中包含 1 的个数

假设已知 dp1[0],那么根据上述分析,可以求出
dp1[1] = 10 * dp[0] + 10^1

这是因为 0~99 可以分为:

  • 十位是 dp[0] ,个位都是 1,有 10 * dp[0] 这么多 1。

    (高位是 dp[0] ,最低位都是 1)

01
11
21
31
41
51
61
71
81
91

  • 10^1:数出出现在十位的 1 有 10 个
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19

dp1[2] 表示: 0~999 里出现的 1 的个数
dp1[2] = 10 * dp1[1] + 100

dp1[1] * 10 指的是:最高位有 0、1、2、3、4、5、6、7、8、9 一共 10 种情况,后两位里 1 的个数由 dp1[1] 决定。

先选最高位,再选低两位,所以是乘法。

特别地,最高位是 1 的每一个数都有 1 所以是 100 个,这里的 100 是 100、101、102、……、199,高位是 1 数出 100 个。因此,
dp1[i] = 10 * dp1[i-1] + 10^i
最基本情况 : 因为 0~9 中只有一个 1,因此 dp1[0] = 1

然后再定义 dp 数组 dp2[i]:表示「从右往左」开始第 i + 1 个数到最后一个数组成的数字包含 1 的个数。

dp2[0] 表示 0 到 6 中包含 1 的个数
例如 dp2[1] 在上例中表示 0~66 包含 1 的个数

对于 i,假设此位为 t,那么需要考虑如下几种情况:

  • (这种情况很简单)如果 t = 0,那么该位的结果只能依赖之前的位置,此时有 dp2[i] = dp2[i - 1]
  • (这种情况也不难)如果 t = 1,例如 166,此时应该求出 0~66 中元素 1 的个数,也就是 dp2[i - 1],然后加上由 100~166 中百位数 1 的个数,有 67 个。再加上 0~991 的个数 dp1[1]。此时有:

dp2[i] = dp2[i - 1] + subnums(i + 1, n) + 1 + dp1[i - 1]

  • 如果 t != 0 并且 t != 1,此时由一开始的分析可以得到

dp2[i] = dp2[i-1] + t * dp1[i-1] + 10^i

  • base case: 如果最后一个数字为 0 那么 dp2[0] = 0,否则 dp2[0] = 1

输出:dp2[len - 1]

参考代码

import java.util.Arrays;

public class Solution {

    public int countDigitOne(int n) {
        // 转换成为字符串
        String s = String.valueOf(n);
        int len = s.length();
        if (len == 1) {
            return n == 0 ? 0 : 1;
        }

        // A[i] 表示:0~10^{i+1} - 1 里包含 1 的个数
        // i = 0 时,10^{i+1} - 1 = 10 - 1 = 9
        // i = 1 时,10^{i+1} - 1 = 100 - 1 = 99
        int[] A = new int[len - 1];
        A[0] = 1;
        for (int i = 1; i < len - 1; i++) {
            A[i] = 10 * A[i - 1] + (int) Math.pow(10, i);
        }

        int[] dp = new int[len];
        if (s.charAt(len - 1) == '0') {
            dp[0] = 0;
        } else {
            dp[0] = 1;
        }
        for (int i = 1; i < len; i++) {
            char currChar = s.charAt(len - i - 1);
            if (currChar == '0') {
                // 高位是 0,没有 1,就取决于低位中 1 的个数
                dp[i] = dp[i - 1];
            } else if (currChar == '1') {
                // 最高位是 1,高位是 1 的个数取决于后面有多少个数,要记得加 1,因为有 10000 这种情况
                int count = Integer.parseInt(s.substring(len - i, len));
                // dp[i - 1] 和情况 1 一样理解
                // A[i - 1] 比如 199,A[i - 1] 表示 0 到 99 的里 1 的个数
                dp[i] = (count + 1) + dp[i - 1] + A[i - 1];
            } else {
                // 最高位是 2、3、4、5、6、7、8、9、10
                // dp[i - 1] 表示余数部分
                // (Character.getNumericValue(currChar)) * A[i - 1] 表示有几个区间段
                // (int) Math.pow(10, i) 最高位是 1 每一位都是 1 所以是 10 的方幂
                dp[i] = dp[i - 1] + (currChar - '0') * A[i - 1] + (int) Math.pow(10, i);
            }
        }
        return dp[len - 1];
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 198,030评论 5 464
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,198评论 2 375
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 144,995评论 0 327
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,973评论 1 268
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,869评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,766评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,967评论 3 388
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,599评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,886评论 1 293
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,901评论 2 314
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,728评论 1 328
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,504评论 3 316
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,967评论 3 302
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,128评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,445评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,018评论 2 343
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,224评论 2 339

推荐阅读更多精彩内容