借鉴了前面一题,采用哈希的思路来处理,但是最后一个case还是无法通过。
另外,不能用scanf(“%s"...)直接读入string,要预先分配空间:https://blog.youkuaiyun.com/vsooda/article/details/7478646
#include <iostream>
#include<algorithm>
#include<string>
#include<vector>
#include <set>
using namespace std;
int getNum(string s){//哈希
int temp = 0;
for(int i = 0; i < 3; i++)
temp = temp * 26 + s[i] - 'A';
temp = temp * 10 + s[3] - '0';
return temp;
}
vector<string> S(400002);
vector<int>course[2501];
vector<int>::iterator it;
int main(){
string tempstu;
int n, k, cnt, temp, tempnum,index,i,j;
scanf("%d %d",&n, &k);
for(i = 0; i < n; i++){
cin>>tempstu>>cnt;
//cout<<tempstu<<endl;
tempnum = getNum(tempstu);
S[tempnum]=tempstu;
//cout<<tempnum<<"*"<<S[tempnum];
//cout<<"----1----"<<cnt<<endl;
for(j = 0; j < cnt; j++){
scanf("%d",&index);
course[index].push_back(tempnum);
}
//cout<<"----2----"<<endl;
}
for(i =1; i <=k; i++){
printf("%d %d\n",i, course[i].size());
sort(course[i].begin(),course[i].end());
for(j=0;j<course[i].size();j++)
cout<<S[course[i][j]]<<endl;
//printf("%s\n",S[course[i][j]]);
}
system("pause");
return 0;
}
下面就参考柳婼大神:https://www.liuchuo.net/archives/2147
大神代码真优雅!对于本题,cmp()函数借助字符数组下标,对全局变量进行排序,真的是简洁又方便。
#include <iostream>
#include <vector>
#include <algorithm>
#include <string.h>
using namespace std;
char name[40010][5];
vector<int> course[2510];
bool cmp1(int a, int b) {
return strcmp(name[a], name[b]) < 0;
}
int main() {
int n, k;
scanf("%d %d", &n, &k);
for(int i = 0; i < n; i++) {
int cnt, temp;
scanf("%s %d", name[i], &cnt);
for(int j = 0; j < cnt; j++) {
scanf("%d", &temp);
course[temp].push_back(i);
//i是name的下标,temp号课程向量中有name[i]同学
}
}
for(int i = 1; i <= k; i++) {
printf("%d %d\n", i, course[i].size());
sort(course[i].begin(), course[i].end(), cmp1);
//cmp函数给定姓名在name中的下标,比较对应字符串
for(int j = 0; j < course[i].size(); j++)
printf("%s\n", name[course[i][j]]);
}
return 0;
}