（水题）洛谷 - P1583 - 魔法照片

https://www.luogu.org/problemnew/show/P1583

设计一个strcut cmp用来比较，就可以了。

#include<bits/stdc++.h>
using namespace std;
#define ll long long

struct P{
    int W;
    int idx;
}p[];

struct cmp{
    bool operator()(P p1,P p2){
        if(p1.W!=p2.W)
            return p1.W>p2.W;
        else
            return p1.idx<p2.idx;
    }
};

int E[];

int main(){
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=;i<=;i++)
        scanf("%d",&E[i]);
    for(int i=;i<=n;i++){
        scanf("%d",&p[i].W);
        p[i].idx=i;
    }

    sort(p+,p++n,cmp());
    for(int i=;i<=n;i++){
        p[i].W+=E[(i-)%+];
    }

    sort(p+,p++n,cmp());

    for(int i=;i<=k;i++){
        printf("%d",p[i].idx);
        if(i==k)
            printf("\n");
        else
            printf(" ");
    }

}