洛谷P5594-【XR-4】模拟赛

题目描述

X 校正在进行 CSP 前的校内集训。

一共有 nn 名 OIer 参与这次集训，教练为他们精心准备了 mm 套模拟赛题。

然而，每名 OIer 都有各自的时间安排，巧合的是，他们在接下来的 kk 天中都恰好有 mm 天有空打模拟赛。

为了方便管理，教练规定一个人必须按顺序打完 mm 套模拟赛题。

比如，小 X 在接下来的第 2,3,52,3,5 天有空打模拟赛，那么他就必须在第 22 天打第 11 套模拟赛题，第 33 天打第 22 套模拟赛题，第 55 天打第 33 套模拟赛题。

教练需要为每一个人的每一次模拟赛做准备，为了减小工作量，如果在某一天有多个人打同一套模拟赛题，那么教练只需要在这一天准备一场使用这一套题的模拟赛即可。

你作为机房大佬，教练想请你帮他计算一下，他每天需要准备多少场模拟赛。

输入格式

第一行三个整数 n,m,kn,m,k。

接下来 nn 行，每行 mm 个整数，第 ii 行第 jj 列的整数 a_{i,j}ai,j​ 表示第 ii 个人在接下来的 kk 天中第 jj 个有空的日子为第 a_{i,j}ai,j​ 天。

输出格式

一行 kk 个整数，第 ii 个整数表示接下来的第 ii 天教练需要准备的模拟赛场数。

样例输入 1

1 3 5
2 3 5

样例输出 1

0 1 1 0 1

样例输入 2

6 3 7
2 3 4
2 5 7
3 5 7
1 3 5
5 6 7
1 2 3

样例输出 2

1 2 3 1 3 1 1

样例输入 3

10 10 20
2 3 4 8 9 11 12 16 17 18
2 3 6 10 12 13 14 15 19 20
1 3 7 10 11 13 14 15 17 19
1 2 4 6 7 9 15 17 19 20
2 3 5 6 9 11 14 16 19 20
1 2 3 8 9 10 11 12 15 19
1 4 6 7 9 12 13 17 18 19
1 7 8 9 10 11 13 15 18 20
1 5 6 7 8 9 13 16 18 19
4 5 7 10 11 13 14 17 18 20

样例输出 3

1 2 2 3 2 2 4 3 3 3 3 4 2 1 3 1 2 2 2 1

解题思路：

这道题目的核心就在这句话：  教练需要为每一个人的每一次模拟赛做准备，为了减小工作量，如果在某一天有多个人打同一套模拟赛题，那么教练只需要在这一天准备一场使用这一套题的模拟赛即可。

程序代码：   

#include <iostream>
#include <numeric>
#include <vector>
using namespace std;

int main() {
    int n, m, k;
    cin>>n>>m>>k;
    vector<vector<int>> A;
    vector<int> a;
    vector<vector<int>> B(k + 1, vector<int>(m, 0));//记录每天需要比赛的场次
    for(int i=0; i<n; i++) {
        for(int j=0; j<m; j++) {
            int t;
            cin>>t;
            a.push_back(t);
        }
        A.push_back(a);
        a.clear();
    }
    for(int i=0; i<m; i++) {
        for(int j=0; j<n; j++) {
            B[A[j][i]][i] = 1;//标记已出现过
        }
    }
    for(int i=1; i<=k; i++) {
        if(i == 1)
            cout<<accumulate(B[i].begin(), B[i].end(), 0);//统计每天准备比赛场次的和
        else
            cout<<" "<<accumulate(B[i].begin(), B[i].end(), 0);
    }
}