【Leetcode】233. Number of Digit One

题目地址：

https://leetcode.com/problems/number-of-digit-one/description/

给定一个非负整数$n$，求从$1\sim n$这些正整数中各个位总共有多少个$1$。

先将$n$的每一位求出来然后从左到右排列。接下来考虑每一位为$1$的小于等于$n$的数有多少个。假设$n=\overline{abcdefg}$，我们考虑$d$这一位，有多少个数能取$1$。如果$d=0$，那么其左边的方案是$0\sim \overline{abc}-1$，右边的方案为$0\sim 999$；如果$d=1$，那么除了上面那个方案之外，左边是可以取$\overline{abc}$的，而此时右边可以取$0\sim \overline{efg}$；如果$d>1$，那么其左边的方案数是$0\sim \overline{abc}$，右边的方案为$0\sim 999$。对每一位求个总和即可。代码如下：

class Solution {
 public:
  int countDigitOne(int n) {
    vector<int> a;
    while (n) a.push_back(n % 10), n /= 10;
    reverse(a.begin(), a.end());
    int res = 0;
    for (int i = 0; i < a.size(); i++) {
      int d = a[i];
      int l = 0, r = 0, p = 1;
      for (int j = 0; j < i; j++) l = l * 10 + a[j];
      for (int j = i + 1; j < a.size(); j++) {
        r = r * 10 + a[j];
        p = p * 10;
      }

      if (!d)
        res += l * p;
      else if (d == 1)
        res += l * p + r + 1;
      else
        res += (l + 1) * p;
    }

    return res;
  }
};

时间复杂度$O({\log }^{2}n)$，空间$O(\log n)$。