堆棋子

题目描述

小易将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+" ");
            }
        }
    }
}