【寒假每日一题】洛谷 P5594 【XR-4】模拟赛

题目链接：P5594 【XR-4】模拟赛 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题目描述

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

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

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

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

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

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

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

输入格式

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

接下来 n 行，每行 m 个整数，第 i 行第 j 列的整数 ai,j 表示第 i 个人在接下来的 k 天中第 j 个有空的日子为第 ai,j 天。

输出格式

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

样例 #1

样例输入 #1

1 3 5
2 3 5

样例输出 #1

0 1 1 0 1

样例 #2

样例输入 #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

样例输入 #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

提示

本题采用捆绑测试。

- Subtask 1（13 points）：n = m = k = 1。
- Subtask 2（24 points）：n = 1。
- Subtask 3（24 points）：m = 1。
- Subtask 4（39 points）：无特殊限制。

对于 100% 的数据，1 <= n,m,k <= 10^3，m <= k，1 <= ai,1 < ai,2 < ... < ai,m <= k。

AC code 1:（效率较高，耗时较短）

#include<iostream>
#include<algorithm>
#include<vector>
#include<unordered_set>
#include<unordered_map>

using namespace std;

int main()
{
	int n,m,k;
	cin>>n>>m>>k;
	
	vector<vector<int>> a(n,vector<int>(m));
	for(int i = 0 ; i < n ; i ++)
		for(int j = 0 ; j < m ; j ++)
			cin>>a[i][j];
	
	unordered_set<int> s;
	unordered_map<int,int> mp;
	
	for(int j = 0 ; j < m ; j ++)
	{
		for(int i = 0 ; i < n ; i ++)
		{
			s.insert(a[i][j]);
		}
		for(auto i : s)
			mp[i] ++;
		s.clear();
	}
	
	vector<int> res(k + 1);
	for(auto it = mp.begin() ; it != mp.end() ; it ++)
	{
		res[it -> first] = it -> second;
	}
	
	for(int i = 1 ; i <= k ; i ++)
		cout<<res[i]<<" ";
	
	return 0;
}

AC code 2:（效率较低，耗时较长）

#include<iostream>
#include<algorithm>
#include<vector>
#include<set>

using namespace std;

int main()
{
	int n,m,k;
	cin>>n>>m>>k;
	
	vector<vector<int>> a(n,vector<int>(m));
	set<pair<int,int>> s; // 注意：若定义为 set<vector<int>> s; 则 TLE 
	vector<int> res(k + 1);
	
	for(int i = 0 ; i < n ; i ++)
	{
		for(int j = 0 ; j < m ; j ++)
		{
			cin>>a[i][j];
			if(s.find({a[i][j],j}) == s.end())
			{
				s.insert({a[i][j],j});
				res[a[i][j]] ++;
			}
		}
	}
	
	for(int i = 1 ; i <= k ; i ++)
		cout<<res[i]<<" ";
	
	return 0;
}