PAT 1039

题目链接: http://pat.zju.edu.cn/contests/pat-a-practise

题目不难，但是最后一个case数据量大，用map会超时，所以根据学生姓名建立四维数组，方便查询。

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;

struct Node
{
    char name[10];
};

int main()
{
    int N, K;
    while(scanf("%d%d", &N, &K) != -1)
    {
        /*根据学生姓名建的四维数字将学生名字转化为数字存储，
        在windows下声明这个数组会栈溢出，因为超过windows下函数内栈大小，
        另外，换成cin cout最后一个case会超时*/
        vector<int> check[26][26][26][10];
        vector<Node> list(N);
        for(int i=0; i<K; ++i)
        {
            int num, r; scanf("%d%d", &num, &r);
            for(int j=0; j<r; ++j)
            {
                char name[10]; 
                scanf("%s", name);
                int a=name[0]-'A';
                int b=name[1]-'A';
                int c=name[2]-'A';
                int d=name[3]-'0';
                check[a][b][c][d].push_back(num);
            }
        }
        for(int i=0; i<N; ++i)
            scanf("%s",list[i].name);
        for(int i=0; i<list.size(); ++i)
        {
            int a=list[i].name[0]-'A';
            int b=list[i].name[1]-'A';
            int c=list[i].name[2]-'A';
            int d=list[i].name[3]-'0';
            if(check[a][b][c][d].size() == 0)
                printf("%s 0
", list[i].name);
            else
            {
                sort(check[a][b][c][d].begin(), check[a][b][c][d].end());
                int length = check[a][b][c][d].size();
                printf("%s %d", list[i].name, length);
                for(int j=0; j<length; ++j)
                    printf(" %d", check[a][b][c][d][j]);
                printf("
");
            }
        }
    }
    return 0;
}