洛谷1583 佳佳的魔法照片

佳佳的魔法照片

题目描述

一共有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]相同，编号小的优先。

解析

按初始权值排序得到一个新的序列，这个序列中每个人有一个与开始不同的编号（1~n），按这个编号对10取模，最后加上额外权值即可

时间复杂度

O（n logn）

代码

var
  w,a,s,c:array[0..50000] of longint;
  n,k,i:longint;


procedure qsort(l,r:longint);
var
  i,j,key,k:longint;
begin
  if l>=r then exit;
  i:=l; j:=r;
  key:=w[(l+r) div 2];
  k:=s[(l+r) div 2];
  repeat
    while (w[i]>key) or ((w[i]=key) and (s[i]<k)) do inc(i);
    while (w[j]<key) or ((w[j]=key) and (s[j]>k)) do dec(j);
    if i<=j then
      begin
        w[0]:=w[i];w[i]:=w[j];w[j]:=w[0];
        s[0]:=s[i];s[i]:=s[j];s[j]:=s[0];
        inc(i); dec(j);
      end;
  until i>j;
  qsort(l,j);
  qsort(i,r);
end;


begin
  readln(n,k);
  for i:=1 to 10 do
    read(a[i]);
  readln;
  for i:=1 to n do
    begin
      read(w[i]);
      s[i]:=i;
    end;
  qsort(1,n);
  for i:=1 to n do
    c[i]:=(i-1) mod 10+1;
  for i:=1 to n do
    w[i]:=w[i]+a[c[i]];
  qsort(1,n);
  for i:=1 to k do
    write(s[i],' ');
end.