【LeetCode】233. 数字 1 的个数

题目描述

 
 

题解思路

以下内容摘自官方题解评论区@龅牙叔：

这道题看了K神@Krahets的剑指题解才算弄明白, 我把题解再解释一下:

举个例子： n = 2304 。答案为四个部分之和：

1. 所有小于等于2304的正整数中，个位出现1的次数.
2. 所有小于等于2304的正整数中，十位出现1的次数.
3. 所有小于等于2304的正整数中，百位出现1的次数.
4. 所有小于等于2304的正整数中，千位出现1的次数.

这四部分可以只考虑一部分，另外三部分就异曲同工了。用第二部分来举例，也就是计算所有小于等于2304的正整数中，十位出现1的次数：

为了帮助理解，我们先想象有一个自行车密码锁（这个比喻来自@ryan0414），一共有四位，每一位可单独滚动。为了计算十位出现1的次数，我们考虑三种情况：

1. n中的十位为0. 即 n = 2304。

我们先锁住十位，强行让十位变成1，剩下三位可以随意滚动：XX1X。那么求十位出现一的个数也就是，我可以滚出多少种密码组合，使得该密码小于等于n（注意十位被锁定成了1，转不动）。

不难发现，我们能滚出的最大数是：2219,

我们能滚出的最小数是：0010。

那么0010到2219之间有多少种十位为1的密码呢？我们去掉十位，得到000和229。一共就是229-000+1 = 230种。即n千位和百位构成的数。

2. n中的十位为1. 即 n = 2314。

我们先锁住十位，强行让十位变成1，剩下三位可以随意滚动：XX1X。那么求十位出现一的个数也就是，我可以滚出多少种密码组合，使得该密码小于等于n（注意十位被锁定成了1，转不动）。

不难发现，我们能滚出的最大数是：2314,

我们能滚出的最小数是：0010。

那么0010到2314之间有多少种十位为1的密码呢？我们去掉十位，得到000和234，一共就是23x10+4+1 = 235种。即n千位和百位构成的数*10 + n个位的数字+1.

3. n中的十位为2~9中任意数字. 即 n = 2324（只是一个例子，n = 2394是一样的）。

我们先锁住十位，强行让十位变成1，剩下三位可以随意滚动：XX1X。那么求十位出现一的个数也就是，我可以滚出多少种密码组合，使得该密码小于等于n（注意十位被锁定成了1，转不动）。

不难发现，我们能滚出的最大数是：2319,

我们能滚出的最小数是：0010。

那么0010到2319之间有多少种十位为1的密码呢？我们去掉十位，得到000和239，一共就是(23+1) x 10 = 240种。 即(n万位和千位构成的数+1)*10

如果我们定义十位左边的数为高位，2304的高位为23，十位右边的数为低位，2304的低位为4，那么以上规律就可以写成高位和低位的规律。我们分别对2304的每一位做一次分析，并将四部分结果相加就得到了答案。

 
 

代码详解

代码来自K神：

class Solution {
    public int countDigitOne(int n) {
        int digit = 1, res = 0;
        int high = n / 10, cur = n % 10, low = 0;
        while(high != 0 || cur != 0) {
            if(cur == 0) res += high * digit;
            else if(cur == 1) res += high * digit + low + 1;
            else res += (high + 1) * digit;
            low += cur * digit;
            cur = high % 10;
            high /= 10;
            digit *= 10;
        }
        return res;
    }
}

 
 
以下为本人进行的代码拆解，我们把代码拆解为三部分：

1. 变量准备
2. 更新维护
3. 循环主体

1.变量准备

        int digit = 1, res = 0;
        int high = n / 10, cur = n % 10, low = 0;

以2304为例：
cur指向的数将整个分为两部分high和low
res用于保存结果（1的总个数），digit用于记录cur的位置（在计算中另有他用）

循环过程如下图：

2.更新维护

            low += cur * digit;
            cur = high % 10;
            high /= 10;
            digit *= 10;

每次循环，更新cur、high、low、digit
以low为例

3.循环主体

        while(high != 0 || cur != 0) {
            if(cur == 0) res += high * digit;
            else if(cur == 1) res += high * digit + low + 1;
            else res += (high + 1) * digit;

这部分遵照题解思路：(见上方黄色高光，一一对应)
（1）cur为0时
（2）cur为1时
（3）cur为2-9时

 
 

心得体会

这是一个偏数学的问题，并没有用到编程特有的数据结构知识。
但是，为我们提供了一个解决数字问题的很好思路或模式。
比如变量准备以及更新维护，还有一部分思路参考。