华为机试：输入法查找

#include <iostream>
#include <map>
#include <vector>
#include <string>
#include <algorithm>

using namespace std;

/* 把输入的每一行的编码解析为拼音和汉字 */
void parse_raw(string &str, string &hanzi, string &pinyin)
{
const char *p = str.c_str();
const char *q = p;

while (*q != '_')
q++;

// note [) 范围是前开后闭
string _hanzi(p, q);
string _pinyin(q + 1, p + str.size());

hanzi = _hanzi;
pinyin = _pinyin;
}

/* 把输入的拼音解析为拼音和数字 */
void parse_input(string &str, string &pinyin, int &nums)
{
const char *p = str.c_str();
const char *q = p;

while (*q != '\0' && !('0' <= *q && *q <= '9'))
q++;

string _pinyin(p, q);
pinyin = _pinyin;

if ('0' <= *q && *q <= '9')
nums = *q - '0';
else
nums = 0;
}

int main()
{
map<string, vector<string> > ssmap;
string str_raw;

/* 定义常量 */
const string tag_start = "Code";
const string tag_search = "Search";
const string tag_end = "End";
string hanzi, pinyin;
int num;

cin >> str_raw;

/* 等于开始信号 */
if (str_raw == tag_start)
{
/* 读取编码表 */
cin >> str_raw;
while (str_raw != tag_search)
{
parse_raw(str_raw, hanzi, pinyin);
ssmap[pinyin].push_back(hanzi);
cin >> str_raw;
}

/* 输入的search */
cin >> str_raw;

/* 解析输入的拼音 */
parse_input(str_raw, pinyin, num);

cin >> str_raw;

/* 结束标志位 */
if (str_raw == tag_end)
{
/* 判断 这个拼音是否存在*/
if (ssmap.find(pinyin) == ssmap.end())
{
cout << "Erro" << endl;
}
else if (num > 0)
{
/* 判断num是否存在 */
if(num < ssmap[pinyin].size())
{
cout << ssmap[pinyin][num - 1] << endl;
}
else
{
cout << "Erro" << endl;
}

}
else
{
/* 输出前5个 */
size_t i = 0;

for ( i = 0; i < ssmap[pinyin].size() - 1 && i != 4; i++)
cout << ssmap[pinyin][i] << ";";
cout << ssmap[pinyin][i] << endl;
}
}
}

return 0;
}