题目描述
小易将n个棋子摆放在一张无限大的棋盘上。第i个棋子放在第x[i]行y[i]列。同一个格子允许放置多个棋子。每一次操作小易可以把一个棋子拿起并将其移动到原格子的上、下、左、右的任意一个格子中。小易想知道要让棋盘上出现有一个格子中至少有i(1 ≤ i ≤ n)个棋子所需要的最少操作次数.
输入描述:
输入包括三行,第一行一个整数n(1 ≤ n ≤ 50),表示棋子的个数
第二行为n个棋子的横坐标x[i](1 ≤ x[i] ≤ 10^9)
第三行为n个棋子的纵坐标y[i](1 ≤ y[i] ≤ 10^9)
输出描述:
输出n个整数,第i个表示棋盘上有一个格子至少有i个棋子所需要的操作数,以空格分割。行末无空格
如样例所示:
对于1个棋子: 不需要操作
对于2个棋子: 将前两个棋子放在(1, 1)中
对于3个棋子: 将前三个棋子放在(2, 1)中
对于4个棋子: 将所有棋子都放在(3, 1)中
示例1
输入
复制
4
1 2 4 9
1 1 1 1
输出
复制
0 1 3 10
思路:这题是www.lintcode.com/problem/best-meeting-point/进化版,观察得x,y轴是独立的,所以转换视角,x,y的距离分开求。当i取某个值时(最终i个重合的棋子个数),候选的格子可选为 出现的x,y点的不同组合(当不取这些格子时可等于前面的某个候选格子)。计算所有的候选格子,与其他最近i个棋子的距离。他们的最小值,一定为取i值是的棋子所需要的最少操作次数。
import java.util.*;
public class Main{
public static void main(String[] args){
Scanner sc=new Scanner(System.in);
while(sc.hasNext()){
int n=sc.nextInt();
long[][] points=new long[n][2];
for(int i=0;i<n;i++){
points[i][0]=sc.nextLong();
}
for(int i=0;i<n;i++){
points[i][1]=sc.nextLong();
}
for(int k=0;k<n;k++){
long ret=Long.MAX_VALUE;
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
PriorityQueue<Long> pq=new PriorityQueue<>();
long x=points[i][0];
long y=points[j][1];
long sum=0;
for(int l=0;l<n;l++){
long dis=Math.abs(x-points[l][0])+
Math.abs(y-points[l][1]);
pq.offer(-dis);
if(pq.size()>k+1) pq.poll();
}
while(pq.size()>0) sum+=-pq.poll();
ret=Math.min(sum,ret);
}
}
System.out.print(ret+" ");
}
}
}
}