数据较大时的输入输出

先来看问题

假设全校有最多40000名学生和最多2500门课程。现给出每个学生的选课清单，要求输出每门课的选课学生名单。

输入格式:

输入的第一行是两个正整数：N（≤40000），为全校学生总数；K（≤2500），为总课程数。此后N行，每行包括一个学生姓名（3个大写英文字母+1位数字）、一个正整数C（≤20）代表该生所选的课程门数、随后是C个课程编号。简单起见，课程从1到K编号。

输出格式:

顺序输出课程1到K的选课学生名单。格式为：对每一门课，首先在一行中输出课程编号和选课学生总数（之间用空格分隔），之后在第二行按字典序输出学生名单，每个学生名字占一行。

输入样例:
10 5
ZOE1 2 4 5
ANN0 3 5 2 1
BOB5 5 3 4 2 1 5
JOE4 1 2
JAY9 4 1 2 5 4
FRA8 3 4 2 5
DON2 2 4 5
AMY7 1 5
KAT3 3 5 4 2
LOR6 4 2 4 1 5
输出样例:
1 4
ANN0
BOB5
JAY9
LOR6
2 7
ANN0
BOB5
FRA8
JAY9
JOE4
KAT3
LOR6
3 1
BOB5
4 7
BOB5
DON2
FRA8
JAY9
KAT3
LOR6
ZOE1
5 9
AMY7
ANN0
BOB5
DON2
FRA8
JAY9
KAT3
LOR6
ZOE1
代码长度限制

16 KB

时间限制

1000 ms

内存限制

64 MB

我们只需要设置一个vector<vector<string>> course(k);表示k门课程，当输入学生选课名单时对应的课程上加一个名字，随后输出所有课程对应的名字就行。所以我就写了下面的代码，但是交上去有个点没过。

#include<iostream>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;
int main()
{
    int n, k;
    cin >> n >> k;
    string name;
    int i, j, m;
    vector<vector<string>> course(k);
    while (n--)
    {
        cin >> name >> m;
        while (m--)
        {
            cin >> i;
            course[i - 1].emplace_back(name);
        }
    }
    
    for (i = 0; i < k; i++)
    {
        sort(course[i].begin(), course[i].end());
        cout << i + 1 << " " << course[i].size() << endl;
        for (j = 0; j < course[i].size(); j++)
        {
            cout << course[i][j] << endl;
        }
    }
}

为什么会这样呢？我查了资料发现，简单来讲就是，c++的cin和cout函数输入输出的时候会把输入输出的数据先放到一个地方，然后再输入输出，而c语言的scanf和printf是直接输入输出，所以cin和cout不如后者。当输入输出数据量很大的时候，cin，cout就会慢好多，要提高cin，cout的效率，就得再使用cin，cout之前加上 ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);这三句话，这样就能将“把数据先放到一个地方”的步骤给去掉，效率大大提高。要注意的是，加上这三句话后，不能cout<<endl;应该用'\n'，要不然速度还是会慢好多。看我改进后的结果。

#include<iostream>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;
int main()
{
   ios::sync_with_stdio(false);
    cin.tie(0);
   cout.tie(0);
    int n, k;
    cin >> n >> k;
    string name;
    int i, j, m;
    vector<vector<string>> course(k);
    while (n--)
    {
        cin >> name >> m;
        while (m--)
        {
            cin >> i;
            course[i - 1].emplace_back(name);
        }
    }
    
    for (i = 0; i < k; i++)
    {
        sort(course[i].begin(), course[i].end());
        cout << i + 1 << " " << course[i].size() << "\n";
        for (j = 0; j < course[i].size(); j++)
        {
            cout << course[i][j] << "\n";
        }
    }
}