1039 Course List for Student

此题遇到了多次错误
一开始flag使用多维数组vectorflag[91][91][91][10];
但是多维数组初始化实在太慢太慢太慢。不放在oj里只是在线下编译器里运行都等好久反应
于是查找后把多维数组变为一维数组，用一个have函数进行变化，把字符串name可以转换为一个唯一的整数放在flag数组里
以下是vector的两种写法

#include<cstdio>
#include<cstring>
#define maxn 175761
#include<vector>
#include<algorithm>
using namespace std;
struct student{
	char name[5];
	vector<int>course;
	int size;
}stu[40000];
vector<int>flag[maxn];
int have(char name[5])
{
	return (name[0] - 'A') * 26 * 26 * 10\
		+ (name[1] - 'A') * 26 * 10\
		+ (name[2] - 'A') * 10\
		+ (name[3] - '0');

}
int main()
{
	int n,k;
	while(scanf("%d%d",&n,&k)==2)
	{
		int cou;
		for(int i=0;i<k;i++)
		{
			int num;
			scanf("%d%d",&cou,&num);
			for(int j=0;j<num;j++)
			{
				char name[5];
				scanf("%s",name);
				flag[have(name)].push_back(cou);
			}
		}
		for(int i=0;i<n;i++)
		{
			char name[5];
			scanf("%s",name);
			strcpy(stu[i].name,name);
			stu[i].size=flag[have(name)].size();
			for(int j=0;j<stu[i].size;j++)
			{
				stu[i].course.push_back(flag[have(name)][j]);
				//这里因为写成stu[i].course[j]=flag[have(name)][j];而出错，
				因为stu[i].course尚未形成stu[i].course[j]的内存空间，此时这样使用会出现verctor错误溢出
				
			}
			sort(stu[i].course.begin(),stu[i].course.end());
		}

		
		for(int i=0;i<n;i++)
		{
			printf("%s %d",stu[i].name,stu[i].size);
			for(int j=0;j<stu[i].size;j++)
				printf(" %d",stu[i].course[j]);
			printf("\n");
		}
	}
	return 0;
}

使用map和string后

#include<iostream>
#include<set>
#include<map>
#include<string>
using namespace std;
#define maxn 40001
map<string,set<int> >mp;
int main()
{
	int n,k;
	cin>>n>>k;
	for(int i=1;i<=k;i++)
	{
		int course,num;
		string name;
		cin>>course>>num;
		for(int j=0;j<num;j++)
		{
			cin>>name;
			mp[name].insert(course);
		}
	}
	for(int i=0;i<n;i++)
	{
		string name;
		cin>>name;
		cout<<name<<" "<<mp[name].size();
		for(set<int>::iterator j=mp[name].begin();j!=mp[name].end();j++)
			cout<<" "<<*j;
		cout<<endl;
	}
	return 0;
}

虽然看着简洁明了但是最后一个测试点不能通过，运行超时了……

#include<cstdio>
#include<cstring>
#define maxn 175761
#include<vector>
#include<algorithm>
using namespace std;
struct student{
	char name[5];
	vector<int>course;
	int size;
}stu[40000];
vector<int>flag[maxn];
int have(char name[5])
{
	return (name[0] - 'A') * 26 * 26 * 10\
		+ (name[1] - 'A') * 26 * 10\
		+ (name[2] - 'A') * 10\
		+ (name[3] - '0');

}
int main()
{
	int n,k;
	while(scanf("%d%d",&n,&k)==2)
	{
		int cou;
		for(int i=0;i<k;i++)
		{
			int num;
			scanf("%d%d",&cou,&num);
			for(int j=0;j<num;j++)
			{
				char name[5];
				scanf("%s",name);
				flag[have(name)].push_back(cou);
			}
		}
		vector<int>::iterator j;
		for(int i=0;i<n;i++)
		{
			char name[5];
			scanf("%s",name);
			strcpy(stu[i].name,name);
			stu[i].size=flag[have(name)].size();
			for(j=flag[have(name)].begin();j!=flag[have(name)].end();j++)
			{
				stu[i].course.push_back(*j);
				//注意这里不是stu[i].course.push_back(flag[have(name)][j]);
			}
			sort(stu[i].course.begin(),stu[i].course.end());
		}

		
		for(int i=0;i<n;i++)
		{
			printf("%s %d",stu[i].name,stu[i].size);
			for(j=stu[i].course.begin();j!=stu[i].course.end();j++)
				printf(" %d",*j);
			printf("\n");
		}
	}
	return 0;
}