洛谷 P1583 魔法照片

#include<bits/stdc++.h>
using namespace std;
const int N=111111;
int e[N];
struct node
{
    int w,id;
}q[N];
bool cmp(node a,node b)
{
    if(a.w==b.w) return a.id<b.id;
    else return a.w>b.w;
}
int main()
{
    int n,k,i,flag;
    scanf("%d%d",&n,&k);
    for(i=1;i<=10;i++) scanf("%d",&e[i]);
    for(i=1;i<=n;i++)
    {
        scanf("%d",&q[i].w);
        q[i].id=i;
    }
    sort(q+1,q+n+1,cmp);
    for(i=1;i<=n;i++)
    {
        flag=(i-1)%10 +1;
        q[i].w+=e[flag];
    }
    sort(q+1,q+n+1,cmp);
    for(i=1;i<=k;i++) printf("%d ",q[i].id);
    return 0;
}