资质|算术能力2 |问题6

介绍

这个问题考察的是算术能力和编程实现能力。给定两个正整数N和K，编写一个程序来求N中包含K的数字的个数。

实现思路

1. 将N转换为字符串。
2. 遍历字符串的每一个字符，判断是否等于K，是的话计数器加一。
3. 输出计数器的值即可。

代码实现

def count_k(n: int, k: int) -> int:
    """
    计算n中包含k的数字的个数
    :param n: 正整数N
    :param k: 正整数K
    :return: 包含k的数字的个数
    """
    count = 0
    str_n = str(n)
    for c in str_n:
        if int(c) == k:
            count += 1
    return count

示例

assert count_k(12345, 5) == 1
assert count_k(54321, 5) == 1
assert count_k(55555, 5) == 5
assert count_k(101010, 0) == 2