洛谷 P1583 魔法照片

P1583 魔法照片

题目描述

一共有n（n≤20000）个人（以1--n编号）向佳佳要照片，而佳佳只能把照片给其中的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]相同，编号小的优先。

输入输出格式

输入格式：

 

第一行输入用空格隔开的两个整数，分别是n和k。

第二行给出了10个正整数，分别是E[1]到E[10]。

第三行给出了n个正整数，第i个数表示编号为i的人的权值W[i]。

 

输出格式：

 

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

 

输入输出样例

输入样例#1： 复制

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

输出样例#1： 复制

10 9 8 7 6 5 4 3 2 1
思路：模拟

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n,k;
int e[11];
struct nond{
    int num,id;
}v[20010];
int cmp(nond a,nond b){
    if(a.num==b.num)    return a.id<b.id;
    return a.num>b.num;
}
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",&v[i].num);
        v[i].id=i;
    }
    sort(v+1,v+1+n,cmp);
    for(int i=1;i<=n;i++)    v[i].num+=e[(i-1)%10+1];
    sort(v+1,v+1+n,cmp);
    for(int i=1;i<=k;i++)    cout<<v[i].id<<" ";
}

 昨天，学妹扔我帮她调一下这个题的代码，当时时间太晚了，加上那个题目也忘了，所以看了一段时间，但是没看出来。
　　今天又看了一遍题目。
　　一眼就发现了问题所在。
　　

#include<algorithm>
#include<iostream> 
#include<cstdio>
using namespace std;
int n,k;
int e[10];
struct s
{
    int w;
    int c;
    int d;
}m[20001];
int cmp(s x,s y)
{
    if(x.w==y.w) return x.d<y.d;
    return x.w>y.w;
}
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",&m[i].w);
        m[i].d=i+1;
        m[i].c=i%10+1;
    }
    sort(m,m+n,cmp);
    for(int i=0;i<n;i++)
        for(int j=0;j<10;j++)
            if(m[i].c==j+1) m[i].w+=e[j];
    sort(m,m+n,cmp);
    for(int i=0;i<k;i++)
        printf("%d ",m[i].d);
    return 0;    
}
/*
14 3
9 2 5 4 0 0 0 0 0 0
1 1 3 4 9 2 8 2 8 8 8 7 1 9

5 7 9
*/

然后，她的错误就十分显然了。

题目中有这么一句话：

这显然是她没有注意到的。

然后，这个题目改完后的代码应该是长得这个样子：

#include<algorithm>
#include<iostream> 
#include<cstdio>
using namespace std;
int n,k;
int e[10];
struct s
{
    int w;
    int d;
}m[20001];
int cmp(s x,s y)
{
    if(x.w==y.w) return x.d<y.d;
    return x.w>y.w;
}
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",&m[i].w);
        m[i].d=i+1;
    }
    sort(m,m+n,cmp);
    for(int i=0;i<n;i++)    m[i].w+=e[i%10];
    sort(m,m+n,cmp);
    for(int i=0;i<k;i++)
        printf("%d ",m[i].d);
    return 0;    
}
/*
14 3
9 2 5 4 0 0 0 0 0 0
1 1 3 4 9 2 8 2 8 8 8 7 1 9

5 7 9
*/