统计某个数字（C语言）

【问题描述】要求实现一个函数，可统计任一整数中某个数出现的次数。例如-125222中，2出现了4次，则该函数应该返回4。

【输入形式】输入一个整数和0-9之间的某个数
【输出形式】输出这个整数中某个数出现的次数
【样例输入】213 0
【样例输出】0

【样例输入】46226 6
【样例输出】2

【样例说明】输入的整数不超过int的范围，某个数的取值范围在[0,9]的区间内。

 代码如下：

#include <stdio.h>
#include <stdlib.h>
int countdig(int num,int dig);
int main()
{
    int num,dig;

    scanf("%d %d",&num,&dig);

    printf("%d",countdig(num,dig));

    return 0;
}
int countdig(int num,int dig)
{
    int a,m= 0;
    if(num == dig)
        m++;
    while(num != 0)
    {
        a = num % 10;
        if(a == dig || a == -dig)
            m++;
        num = num / 10;
    }
    return m;
}

运行截图：