【HDOJ】1251 统计难题

使用STL实现。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <map>
using namespace std;

int main() {
    char ch;
    string str="";
    map<string, int> words;

    while (1) {
        scanf("%c", &ch);
        if (ch == '\n') {
            scanf("%c", &ch);
            str = "";
        }
        // two consecutive '\n' then break;
        if (ch == '\n')
            break;
        str += ch;
        words[str]++;
    }

    while (cin>>str)
        printf("%d\n", words[str]);

    return 0;
}