C语言中如何定义一个函数，用来求一个数中某些数字的个数如：2092中2的个数是2个

C语言中如何定义一个函数，用来求一个数中某些数字的个数如：2092中2的个数是2个

#include <stdio.h>int getCount(int num, int alb){    int a[] = { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }; //初始化数组0~9分别代表数字0~9的个数    while(num != 0)    {        a[num % 10]++;        num /= 10;    }    return a[alb];} int main(){    int n, m;    printf("请输入一个整数：");    scanf("%d", &n);    printf("需要计算哪个数字的个数：");    scanf("%d", &m);    printf("%d中含有%d的个数为%d\n", n, m, getCount(n, m));}