习题5-5 使用函数统计指定数字的个数 (15 分)

1. 题目摘自

https://pintia.cn/problem-sets/12/problems/305

2. 题目内容

本题要求实现一个统计整数中指定数字的个数的简单函数。

函数接口定义：

int CountDigit( int number, int digit );
其中 number 是不超过长整型的整数，digit 为 [0, 9] 区间内的整数。函数CountDigit 应返回 number 中 digit 出现的次数。

输入样例：

-21252 2

输出样例：

Number of digit 2 in -21252: 3

3. 源码参考

#include<iostream>

using namespace std;

int CountDigit(int number, int digit);

int main()
{
    int number, digit;

    cin >> number >> digit;
    cout << "Number of digit " << digit << " in " << number << ": " << CountDigit(number, digit) << endl;

    return 0;
}

int CountDigit(int number, int digit)
{
    int cnt = 0, n;

    if (number > 0)
    {
        n = number;
    }
    else
    {
        n = -number;
    }

    while (n)
    {
        if (n % 10 == digit)
        {
            cnt++;
        }

        n /= 10;
    }

    return cnt;
}