剑指offer-整数中1出现的次数

题目描述

求1~N之间, 1总共出现了多少次.

题目分析

这个题一共有三种做法.

暴力做法1 - 字符串遍历

这个是最简单的方法, 没什么可说的.

// 暴力求解法1
// T: O(n^2) S: O(1)
public int findNumberBruteForce1(int n) {
    int cnt = 0;
    for (int i = 1; i <= n; i++) {
        char[] c = String.valueOf(i).toCharArray();
        for (int j = 0; j < c.length; j++) {
            if (c[j] == '1') {
                cnt++;
            }
        }
    }
    return cnt;
}

暴力做法2 - 取模判断

取模判断法, 每次只判断最后一位数是否为1, 本质上并没有带来什么改变.

// 暴力求解法2
// T: O(n*logn) S: O(1)
public int findNumberBruteForce2(int n) {
    int cnt = 0, j = 0;
    for (int i = 1; i <= n; i++) {
        j = i;
        while (j != 0) {
            if (j % 10 == 1) cnt++;
            j = j / 10;
        }
    }
    return cnt;
}

最优做法

最优做法就是固定每一位的数, 进行分类讨论.

也就是说: 我只求当前位为1的时候, 总共能有多少种方式.

无非就是以下三种情况:(求百位为例)

• 当前位为0, 如: 1 2 0 4 5, 该位贡献了12*100个带1的数字
• 当前位为1, 如: 1 2 1 4 5, 该位贡献了12*100 + 46个带1的数字
• 当前位大于1, 如: 1 2 3 4 5, 该位贡献了13*100个带1的数字

// 数学分类讨论法 (最优解)
// T: O(logn) S: O(1)
public int findNumberMath(int n) {
    if (n < 1) return 0;
    int factor = 1;
    int cnt = 0;
    while (n / factor != 0) {
        int high = n / (factor * 10);
        int digit = (n / factor) % 10;
        if (digit == 1) {
            cnt += high * factor;
            cnt += (n % factor + 1);
        } else if (digit == 0) {
            cnt += high * factor;
        } else {
            cnt += (high + 1) * factor;
        }
        factor *= 10;
    }
    return cnt;
}