题目描述
一共有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<<" "; }
昨天,学妹扔我帮她调一下这个题的代码,当时时间太晚了,加上那个题目也忘了,所以看了一段时间,但是没看出来。
今天又看了一遍题目。
一眼就发现了问题所在。
![技术分享图片](/img/jia.gif)
#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 */