SZTUOJ 1020.钱钱的steam游戏

Description

一共有n（n≤32768）SZTUer（以1–n编号）向钱钱索要steam的游戏，虽然钱钱十分富裕且帅气，但他最近在谈恋爱很耗钱，所以钱钱只能送给其中的k个人。钱钱按照与他们的关系好坏的程度给每个人赋予了一个初始权值W[i]。然后将初始权值从大到小进行排序，每人就有了一个序号D[i]（取值同样是1–n）。按照这个序号对10取模的值将这些人分为10类。（也就是说定义每个人的类别序号C[i]的值为(D[i]-1) mod 10 +1）显然类别序号的取值为1–10。第i类的人将会额外得到E[i]的权值。你需要做的就是求出加上额外权值以后，最终的权值最大的k个人，并输出他们的编号。在排序中，如果两人的W[i]相同，编号小的优先。

Input

第一行输入用空格隔开的两个整数，分别是n和k。 第二行给出了10个正整数，分别是E[1]到E[10]。 第三行给出了n个正整数，第i个数表示编号为i的人的权值W[i]。

Output

只需输出一行用空格隔开的k个整数，分别表示最终的W[i]从高到低的人的编号。

Sample Input

10 10
1 2 3 4 5 6 7 8 9 10
2 4 6 8 10 12 14 16 18 20

Sample Output

10 9 8 7 6 5 4 3 2 1

Hint

1 ≤ n ≤ 32768，1 ≤W[i],E[i]≤ 1000；

题目解析

题目看上去绕来绕去很复杂，实际上只需要按照题目要求，逐步模拟即可首先建一个结构体，按照初始权值排序，根据排序后的结果对每个人按类型加上一个值，再拍一次序就行了。

#include<bits/stdc++.h>
using namespace std;
int e[12],n,k;
struct person{
    int w,num,d;
}p[20010];
bool cmp(const person &a,const person &b){
    if(a.w!=b.w)return a.w>b.w;
    return a.num<b.num;
}
int main(){
    scanf("%d%d",&n,&k);
    for(int i=0;i<10;i++)scanf("%d",&e[i]);
    for(int i=0;i<n;i++){
        scanf("%d",&p[i].w);
        p[i].num=i+1;
    }
    sort(p,p+n,cmp);
    for(int i=0;i<n;i++)
        p[i].d=i%10,p[i].w+=e[p[i].d];
    sort(p,p+n,cmp);
    for(int i=0;i<k;i++)printf("%d ",p[i].num);
}