函数调用，统计数中'2’的个数

Description
定义并调用函数countdigit(number, digit),它的功能是统计整数number中数字digit的个数，如countdigit(10090, 0)的返回值是3. 在主函数中定义并调用该函数，统计任意一个输入整数中’2’的个数。
Input
多组测试数据，每组输入一个整数。
Output
输出该数中’2’的个数。
Sample Input
1002
2323
4567
Sample Output
1
2
0

#include<stdio.h>
int countdigit(int n,int digit)
{
    int i=0;
	while(n)
	{
	    if((n%10)==digit)
		i++;
		n/=10;	
	}
	return i;	
}
int main()
{
	int a,b=2;
	while(scanf("%d",&a)!=EOF)
	printf("%d\n",countdigit(a,b));
	return 0;
}