洛谷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.