B3784 [语言月赛202306] 演唱会

一.题目

Description

一个特殊的节日，zyl 举办了一场演唱会，他想要规划一下演唱会的曲目。

根据同学的要求，他要从 nn 首歌（编号 1\sim n1∼n）中选出 mm 首加入演唱会歌单。

选择的规则很简单，班上有 aa 名学生（不包括 zyl），学号为 1\sim a1∼a。zyl 会按照学号统计每位同学从每首歌中得到的快乐值，记为 h_{i,j}hi,j​。一首歌的「欢乐度」是它所给予其他所有同学的「快乐值」总和。

巧妙的是，由于歌曲风格迥异，同一个学生一定会从不同的歌中会收获不同的快乐值，且不会有任何两首歌的欢乐度相同。

zyl 选出欢乐度最高的 mm 首歌，并将它们按欢乐度从大到小排好序。

但 zyl 注意到了她的快乐值…

无论如何，zyl 一定会把她最喜欢的歌放进歌单（她获得快乐值最高的歌即为她最喜欢的歌）！
如果她最喜欢的歌之前已经在歌单中，那么 zyl 会把这首歌提到歌单的第一个位置（即，仅将该歌曲的位置变为第一位，其余歌曲的相对位置不变）；但如果不在，zyl 则会把已经选择好的歌单里的最后一首歌给删掉。然后把她最喜欢的曲目放在歌单的最后。

现在 zyl 将统计的快乐值数据和她的学号交给你，请你帮他列出演唱会的歌单。

Input

输入共 a + 1a+1 行。

第一行共四个整数 n, m, a, bn,m,a,b，分别代表待选的歌的数量，需要选出的歌的数量，班上的人数，她的学号。
接下来 aa 行，每行 nn 个整数，第 i+1i+1 行第 jj 个数代表学号为 ii 的学生从第 jj 首歌收获的快乐值 h_{i,j}hi,j​。

Output

输出共一行用空格隔开的 mm 个整数，依次代表演唱会的歌单上的歌的编号。

Sample 1

Inputcopy	Outputcopy
4 2 3 2 2 4 3 1 1 2 3 4 2 3 4 1	3 4

Hint

样例 1 解释

按照欢乐度排序后靠前的两道是 3, 23,2，但由于她最喜欢的歌是 44，不在歌单里，于是将 22 从歌单里删除，之后将 44 放在最后。最后的歌单为 3, 43,4。

数据规模与约定

对于 100\%100% 的数据，1\leq m \leq n \leq 10^51≤m≤n≤105，1\leq b \leq a\leq 1001≤b≤a≤100。0\leq h_{i,j}\leq 10^60≤hi,j​≤106。

测试点编号	nn	mm	aa	h_{i,j}hi,j​
1 \sim 21∼2	\leq5≤5	=1=1	\leq5≤5	\leq 10000≤10000
3 \sim 43∼4	\leq1000≤1000	=n=n	\leq100≤100	\leq 10000≤10000
5 \sim 65∼6	\leq 1000≤1000	\leq 1000≤1000	=1=1	\leq 10000≤10000
7 \sim 97∼9	\leq1000≤1000	\leq1000≤1000	\leq100≤100	\leq 10000≤10000
1010	\leq 10^5≤105	\leq 10^5≤105	\leq 100≤100	\leq 10^6≤106

二.分析

先定义一个结构体,存储每首歌的编号和总喜爱度,使用结构体进行操作在排序后可以保留原序号,方便后续操作.

将总喜爱度输入累加之后使用sort进行降序排序

注意不能使用冒泡排序,测试提交会超时

最后单独处理b同学最喜欢的歌,进行输出

#include<bits/stdc++.h>
using namespace std;
const int N=100005;
struct son{
	long long k;
	long long sum;
};
struct son s[N];
bool cmp(son x,son y){
    return x.sum > y.sum;//降序排序
}
int main()
{
	long long n,m,a,b;
	scanf("%lld%lld%lld%lld",&n,&m,&a,&b);
	memset(s,0,sizeof(s));
	for(long long i=1;i<=n;i++)
	s[i].k=i;
	long long favoritevalue=0,favoriteid=0;//定义学号为b的同学最喜欢的歌
	for(long long i=1;i<=a;i++)
	{
		for(long long j=1;j<=n;j++)
		{
			long long t;
			scanf("%lld",&t);
			s[j].sum+=t;//累加歌曲的欢喜度
			if(i==b&&t>favoritevalue){//学号为b的同学单独处理
				favoritevalue=t;
				favoriteid=j;
			}
		}
	}
	long long i,j; 
	sort(s + 1, s + n + 1, cmp);//降序排序
	int flag=0;
	for(i=1;i<=m;i++)
	{
		if(s[i].k==favoriteid) {
			flag=1;//flag=1表示b同学喜欢的歌出现在了榜单上,需要提前输出.
			printf("%lld ",favoriteid);
			s[i].k=-1;
			break;
		}
	}
	if(flag==1)
	{
		for(i=1;i<=m;i++)
		{
			if(s[i].k>0) printf("%lld ",s[i].k);
		}
	}
	else if(flag==0)//b同学喜欢的歌未出现在榜单,替换榜单最后一名的歌曲
	{
		for(i=1;i<m;i++)
		printf("%lld ",s[i].k);
		printf("%lld ",favoriteid);
	}
	return 0; 
}