60pts,B3784 [语言月赛202306] 演唱会

#include<bits/stdc++.h>
using namespace std;
struct node{
	int id,sum;
}a[10000001];
bool cmp(node a,node b){
	return a.sum>b.sum;
}
int n,m,x,y,z,maxn=INT_MIN,max_num=0;
bool flag=false;                                          
int main(){
	cin>>n>>m>>x>>y;
	for(int i=1;i<=n;i++) a[i].id=i;
	for(int i=1;i<=x;i++){
		for(int j=1;j<=n;j++){
			cin>>z;
			if(i==y){
				if(maxn<z){
					maxn=z;
					max_num=j;
				}
			}
			a[j].sum+=z;
		}
	}
	sort(a+1,a+n+1,cmp);
	for(int i=1;i<=m;i++){
		if(a[i].id==max_num){
			flag=true;
			swap(a[1],a[i]);
			break;
		}
	}
	if(!flag)
		a[m].id=max_num;
	for(int i=1;i<=m;i++) cout<<a[i].id<<" ";
    return 0;
}