PAT乙级-1021

题目

给定一个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

代码

#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <math.h>
#include <string.h>
using namespace std;

int main()
{
    char a[1000]; 
    gets(a);
    int c[10]={0};
    for(int i=0; i<strlen(a); i++){
        switch(a[i]){
            case '0':c[0]++;break;
            case '1':c[1]++;break;
            case '2':c[2]++;break;
            case '3':c[3]++;break;
            case '4':c[4]++;break;
            case '5':c[5]++;break;
            case '6':c[6]++;break;
            case '7':c[7]++;break;
            case '8':c[8]++;break;
            case '9':c[9]++;break;
        }
    }
    int flag = 1;
    for(int i=0; i<10; i++){
        if(c[i] != 0){
            if(flag){
                cout<<i<<":"<<c[i];
                flag = 0;
            }else
                cout<<"\n"<<i<<":"<<c[i];
        }
    }

    return 0;
}