CF371E Subway Innovation

题目描述

Berland 正在经历困难时期——泥土的价格下跌,这对国家经济是沉重打击。众所周知,Berland 是全球头号泥土出口国! Berland 的总统只能被迫在现有的 $n$ 个地铁站中关闭一些站点,只保留 $k$ 个。 这些地铁站一个接一个的排在一条直线上,列车会依次经过各个站点。你可以把站点看作位于 $x$ 轴上,第 $i$ 个站点的坐标是 $x_i$。这样你就可以用公式 $|x_i-x_j|$ 简单地计算站点 $i$ 与站点 $j$ 之间的距离。 眼下,交通部正在抉择关闭哪些站,保留哪些站。显然,首都的居民们对于这次革新并不会太热情,所以交通部决定给人们展示最好的一面。他们希望选出 $k$ 个站点使地铁的平均通行时间最小! 假设列车的速度恒定,地铁的平均通行时间定义为:所有站点两两之间距离的总和,除以站点的对数(即 $\frac{n(n-1)}{2}$),再除以列车的速度。 请你帮助交通部解决这个难题。编写一个程序,求出使平均通行时间最小的 $k$ 个站点的坐标。

输入格式

第一行,一个正整数 $n~(3 \le n \le 3 \times 10^5)$,表示革新前地铁站的个数。 第二行,$n$ 个整数 $x_1,x_2,...,x_n~(-10^8 \le x_i \le 10^8)$,表示每个地铁站的坐标。 第三行,一个正整数 $k~(2 \le k \le n - 1)$,表示需要保留的地铁站的个数。

输出格式

一行,$k$ 个正整数 $t_1,t_2,...,t_k$,表示保留的地铁站的编号。假设地铁站以输入的顺序被编号为 $1$ 到 $n$。两个正整数之间用一个空格连接。 如果答案不唯一,输出任意一个即可。

说明/提示

在示例测试用例中,最优答案是摧毁第一个站点 $(x=1)$。这样,平均通行时间将等于 $1$。