欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

巧用char stu[maxn][5]保存名字,这样在存储数组可开成vector<int> course[2505],而不用vector<string> courese[2505](这样会超时)

程序员文章站 2022-07-12 16:48:41
...

用string型的vector最后一个测试点会超时:

#include<cstdio>
#include<algorithm>
#include<string>
#include<queue>
#include<iostream>
using namespace std;
//12:35
vector<string> course[2505];
int main(){
    int n,k;
    scanf("%d %d",&n,&k);
    string name;
    for(int i=0;i<n;i++){
        cin>>name;
        int c,x;
        scanf("%d",&c);
        for(int j=0;j<c;j++){
            scanf("%d",&x);
            course[x].push_back(name);
        }
    }
    for(int i=1;i<=k;i++){
        printf("%d %d\n",i,course[i].size());
        sort(course[i].begin(),course[i].end());
        for(int j=0;j<course[i].size();j++)
            cout<<course[i][j]<<endl;
    }
    return 0;
}

改进:用stu[maxn][5]的char型二位数组,按序保存每个同学的名字。

#include<cstdio>
#include<algorithm>
#include<string>
#include<queue>
#include<string.h>
#include<iostream>
using namespace std;
//12:35
const int maxn=40005;
vector<int > course[2505];
char stu[maxn][5];
bool cmp(int x,int y){
    return strcmp(stu[x],stu[y])<0;
}
int main(){
    int n,k;
    scanf("%d %d",&n,&k);
    for(int i=0;i<n;i++){
        scanf("%s",stu[i]);
        int c,x;
        scanf("%d",&c);
        for(int j=0;j<c;j++){
            scanf("%d",&x);
            course[x].push_back(i);
        }
    }
    for(int i=1;i<=k;i++){
        printf("%d %d\n",i,course[i].size());
        sort(course[i].begin(),course[i].end(),cmp);
        for(int j=0;j<course[i].size();j++)
            printf("%s\n",stu[course[i][j]]);
    }
    return 0;
}