PAT A1039 Course List for Student

自己写的最笨的方法果然Time Limit Exceeded,根据相关的解答，使用map输出也有相关的问题，而且注意，如果使用cin和cout也会大大的增大时长，所以一定在当数据量过大的时候注意这一点;

#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#include<vector>
#include<string>
using namespace std;
using std::vector;
vector<vector<string> > course;
vector<string>search_mem;
bool search_c(vector<string> vec,string name){
    bool flag=false;
    for(int i=0;i<vec.size();i++){
        if(vec[i]==name){
            flag=true;
            break;
        }
    }
    //true代表选课
    return flag;
}
int main(){
    int sn,cn;
    scanf("%d%d",&sn,&cn);
    course.resize(cn);
    for(int i=0;i<cn;i++){
        int index,num;
        scanf("%d%d",&index,&num);
        course[index-1].resize(num);
        getchar();
        for(int j=0;j<num;j++){
            string name;
            cin>>name;
            course[index-1][j]=name;
        }
    }
    for(int i=0;i<sn;i++){
        string name;
        cin>>name;
        search_mem.push_back(name);
    }
    vector<int>infor;
    infor.resize(0);
    for(int i=0;i<sn;i++){
        int select_num=0;
        cout<<search_mem[i];
        for(int j=0;j<cn;j++){
            if(search_c(course[j],search_mem[i])){
                select_num++;
                infor.push_back(j+1);
            }
        }
        if(select_num!=0){
            printf(" %d",select_num);
            for(int i=0;i<infor.size();i++){
                printf(" %d",infor[i]);
            }
        }else{
            printf(" 0");
        }
        infor.resize(0);
        cout<<endl;
    }
    system("pause");
    return 0;
}

 

#include<stdio.h>
#include<stdlib.h>
#include<vector>
#include<algorithm>
using namespace std;
const int N=40010;
const int M=26*26*26*10+1;
vector<int> selectCourse[M];
int getID(char name[]){
    int id=0;
    for(int i=0;i<3;i++){
        id=id*26+(name[i]-'A');
    }
    id=id*10+(name[3]-'0');
    return id;
}
int main(){
    char name[5];
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=0;i<k;i++){
        int course;
        int x;
        scanf("%d%d",&course,&x);
        for(int j=0;j<x;j++){
            scanf("%s",name);
            int id=getID(name);
            selectCourse[id].push_back(course);
        }
    }
    for(int i=0;i<n;i++){
        scanf("%s",name);
        int id=getID(name);
        sort(selectCourse[id].begin(),selectCourse[id].end());
        printf("%s %d",name,selectCourse[id].size());
        for(int j=0;j<selectCourse[id].size();j++){
            printf(" %d",selectCourse[id][j]);
        }
        printf("\n");
    }
    system("pause");
    return 0;
}

这个题目没有想到的是Hash值，Hash值可以很好很大的降低时间复杂度；

大致的对应方式很值得学习：一个字母位对应的数字*26（26是字母个数），一个数字为对应的数字*10。由于一个人的名字构成为三个字母+一个数字，所以HASH数组的大小为26*26*26*10，所以相应的一个人的名字转换的法则如下：

所以，大致的代码如下所示：

#include<stdio.h>
#include<stdlib.h>
#include<vector>
#include<algorithm>
using namespace std;
const int N=40010;
const int M=26*26*26*10+1;
vector<int> selectCourse[M];
int getID(char name[]){
    int id=0;
    for(int i=0;i<3;i++){
        id=id*26+(name[i]-'A');
    }
    id=id*10+(name[3]-'0');
    return id;
}
int main(){
    char name[5];
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=0;i<k;i++){
        int course;
        int x;
        scanf("%d%d",&course,&x);
        for(int j=0;j<x;j++){
            scanf("%s",name);
            int id=getID(name);
            selectCourse[id].push_back(course);
        }
    }
    for(int i=0;i<n;i++){
        scanf("%s",name);
        int id=getID(name);
        sort(selectCourse[id].begin(),selectCourse[id].end());
        printf("%s %d",name,selectCourse[id].size());
        for(int j=0;j<selectCourse[id].size();j++){
            printf(" %d",selectCourse[id][j]);
        }
        printf("\n");
    }
    system("pause");
    return 0;
}