帮忙编个用switch 统计数字出现次数的程序？

#include<stdio.h>
void main()
{
char ch;
int i;
int a[10]={0};
while ((ch=getchar())!='\n')
{
switch(ch)
{
case '0':a[0]++;break;
case '1':a[1]++;break;
case '2':a[2]++;break;
case '3':a[3]++;break;
case '4':a[4]++;break;
case '5':a[5]++;break;
case '6':a[6]++;break;
case '7':a[7]++;break;
case '8':a[8]++;break;
case '9':a[9]++;break;
default:break;
}
}
for (i=0; i<10; i++)
{
printf("%d出现%d次\n", i, a[i]);
}
}

#include<stdio.h>
void main()
{
char ch[20];
char *p=ch;
int a[10]={0},i;
printf("请输入数字：");
scanf("%s",ch);
while (*p)
{
switch(*p)
{
case '0':a[0]++;break;
case '1':a