个位数统计 (15)

时间限制 1000 ms 内存限制 32768 KB 代码长度限制 100 KB 判断程序 Standard (来自 小小)
题目描述
给定一个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

本题的思路是将其每个数字和每个数字的个数以键值对的方式存在map里，难点是对map进行构造比较器时通过vector的方式进行定义比较器，然后进行迭代输出，以下时我的代码：

#include <iostream>
#include <vector>
#include <map>
#include <algorithm>
using namespace std;

bool compByKey(const pair<char, int>& a,const pair<char, int>& b)
{
    return a.first < b.first;
}

map<char, int> count(string s)
{
    map<char, int> mymap;
    map<char, int>::iterator it;
    int i = 0;
    while(s[i] != '\0')
    {
        it = mymap.find(s[i]);
        if(it != mymap.end())
            mymap[s[i]] = mymap[s[i]] + 1;
        else
            mymap.insert( pair<char,int>(s[i], 1) );
        i++;
    }
    return mymap;
}

int main(int argc, char **argv)
{
    string N;
    cin >> N;
    map<char, int> mymap = count(N);
    vector< pair<char, int> >::iterator it;
    vector< pair<char, int> > vec(mymap.begin(), mymap.end());
    sort(vec.begin(), vec.end(), compByKey);
    for(it = vec.begin(); it != vec.end(); it++)
        cout << it->first << ":" << it->second << endl;
    return 0;
}