8.16 佳佳的魔法照片 1255

题目

【题目背景】
如果你看过《哈利•波特》，你就会知道魔法世界里的照片是很神奇的。也许是因为小魔法师佳佳长的太帅，很多人都找他要那种神奇的魔法照片，而且还都要佳佳和他的MM的合照。那些照片可是非常珍贵的，他到底应该把照片给谁呢？
　　一共有n个人（以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]相同，编号小的优先。

题解

模拟+多关键字快排（也可用其他稳定排序）
注意多关键字快排有一种很神奇的错误

时间复杂度O(n log n)

代码

var
  n,i,j,k:longint;
  e:array[1..10]of longint;
  a,b:array[0..50050]of longint;

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

begin
  assign(input,'photo.in');
  assign(output,'photo.out');
  reset(input);rewrite(output);
  readln(n,k);
  for i:=1 to 10 do
    read(e[i]);
  for i:=1 to n do
    begin
      read(a[i]);
      b[i]:=i;
    end;
  qsort(1,n);
  for i:=1 to i do
    a[i]:=a[i]+e[(i-1) mod 10+1];
  qsort(1,n);
  for i:=1 to k do
    write(b[i],' ');
  close(input);close(output);
end.