洛谷P1583 魔法照片【排序】

题目描述 https://www.luogu.org/problemnew/show/P1583

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct node
{
	int w,nm;
}a[20005];
int n,k,e[15];
bool cmp(const node &a,const node &b)
{
	if(a.w!=b.w) return a.w>b.w;
    return a.nm<b.nm;
} 
int main() 
{ 
   scanf("%d%d",&n,&k);
   for(int i=1;i<=10;i++) scanf("%d",&e[i]);
   for(int i=1;i<=n;i++)
   {
      scanf("%d",&a[i].w);
	  a[i].nm=i;	
   } 
   sort(a+1,a+n+1,cmp);
   for(int i=1;i<=n;i++) a[i].w+=e[(i-1)%10+1];
   sort(a+1,a+n+1,cmp);
   for(int i=1;i<=k;i++) printf("%d ",a[i].nm);
	    
  return 0;	
}