PAT上的一道水题，网上提交总是提示运行超时

Question

以下是我的代码：
#include
int main(void)
{
char ch;
int count[10];

for (int i = 0;  i < 10; i++) {
    count[i] = 0;
}
while ((ch = getchar()) != '\n')
    count[ch - '0']++;
for (int i = 0; i < 10; i++)
    if (count[i] != 0)
        printf("%d:%d\n", i, count[i]);
return 0;

}
我是在xcode中运行的没有错误啊
以下是题目：
题目描述
给定一个k位整数N = dk-1*10k-1 + ... + d1*101 + d0 (0<=di<=9, i=0,...,k-1, dk-1>0)，请编写程序统计每种不同的个位数字出现的次数。例如：给定N = 100311，则有2个0，3个1，和1个3。

输入描述:
每个输入包含1个测试用例，即一个不超过1000位的正整数N。

输出描述:
对N中每一种不同的个位数字，以D:M的格式在一行中输出该位数字D及其在N中出现的次数M。要求按D的升序输出。

输入例子:
100311

输出例子:
0:2

1:3

3:1

Answer 1

他给出的测试用例可能不是以'\n'结尾的，所以你读取输入的while会永远循环下去，这就是超时的原因。
我的建议是开一个1005的字符数组，然后直接scanf输入一个字符串（%s），然后将字符串长度作为循环次数来统计count