PAT (Advanced Level) 1039. Course List for Student (25)

map会超时，二分吧...

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<map>
#include<queue>
#include<string>
#include<vector>
using namespace std;

int n,m;
int tot;

vector<int>v[];
struct X
{
    int id;
    int num;
    char name[][];
}s[];

struct Name
{
    char name[];
}op[];

bool cmp2(const Name&a,const Name&b)
{
    return strcmp(a.name,b.name)<;
}

bool cmp(const X&a,const X&b)
{
    return a.id<b.id;
}

char quary[][];

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=;i<=m;i++)
    {
        scanf("%d",&s[i].id);
        scanf("%d",&s[i].num);
        for(int j=;j<=s[i].num;j++) scanf("%s",s[i].name[j]);
    }
    for(int i=;i<=n;i++)
    {
        scanf("%s",op[i].name);
        strcpy(quary[i],op[i].name);
    }
    sort(op+,op++n,cmp2);

    sort(s+,s++m,cmp);

    for(int i=;i<=m;i++)
    {
        for(int j=;j<=s[i].num;j++)
        {
            int h=;
            int l=,r=n;
            while(l<=r)
            {
                int mid=(l+r)/;
                if(strcmp(op[mid].name,s[i].name[j])<) l=mid+;
                else if(strcmp(op[mid].name,s[i].name[j])>) r=mid-;
                else
                {
                    h=mid;
                    break;
                }
            }
            if(h==) continue;
            v[h].push_back(s[i].id);
        }
    }

    for(int i=;i<=n;i++)
    {
        int q=;
        int l=,r=n;
        while(l<=r)
        {
            int mid=(l+r)/;
            if(strcmp(op[mid].name,quary[i])<) l=mid+;
            else if(strcmp(op[mid].name,quary[i])>) r=mid-;
            else
            {
                q=mid;
                break;
            }
        }
        if(q==) continue;

        printf("%s ",quary[i]);
        printf("%d",v[q].size());
        for(int j=;j<v[q].size();j++) printf(" %d",v[q][j]);
        printf("\n");
    }

    return ;
}