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

#include 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的个数为%dn",n,m,getCount(n,m)); }