标签：柱子 积木 int 乐高 height num HeightAndNum PIPI treeSet

集合框架题：PIPI的乐高积木

问题：

思路：

首先，为了保证最高柱子与最矮柱子的高度差越来越小，每次移动都需要将最高柱子上的一个块移动到最矮柱子上，即最矮柱子高度 + 1，最高柱子高度 - 1。我们需要一种数据结构能使其中的元素一直保持按序排列，红黑树即是我们需要的数据结构，Java中TreeSet即是用红黑树实现的，能满足上述要求。

对于高度的变化，我们只需删除原来最高柱与最矮柱，然后添加两个新柱子即可，高度分别为最矮柱子高度 + 1，最高柱子高度 - 1

由于最后需要输出每次移动的两个柱子的编号，我们还需将编号和高度一同存入TreeSet。每次移动的柱子的编号存入二维数组中即可。

需注意的点：

• 当最高柱高度 - 最矮柱高度 <= 1时，此时已无法再减少两者之差，因此终止移动
• 当高度相同时，我们需选取编号最大的高柱和编号最小的矮柱。创建类保存编号和高度，该类需实现Comparable接口，重写compareTo方法自定义排序规则

代码：

import java.util.Scanner;
import java.util.TreeSet;

public class Main {
    static TreeSet<HeightAndNum> treeSet = new TreeSet<>();
    public static void main(String[] args) {
        int move[][] = new int[5005][2];
        int minBeauty;
        Scanner scanner = new Scanner(System.in);
        int N = scanner.nextInt();
        int k = scanner.nextInt();
        int i;
        for (i = 0;i < N;i++) {
            treeSet.add(new HeightAndNum(scanner.nextInt(), i + 1));
        }
        i = 0;
        while (i < k) {
            if (treeSet.last().height - treeSet.first().height <= 1) {
                break;
            }
            int minNum = treeSet.first().num;
            int maxNum = treeSet.last().num;
            int minPoll = treeSet.pollFirst().height;
            int maxPoll = treeSet.pollLast().height;
            treeSet.add(new HeightAndNum(minPoll + 1, minNum));
            treeSet.add(new HeightAndNum(maxPoll - 1,maxNum));
            move[i][0] = maxNum;
            move[i][1] = minNum;
            i++;
        }
        minBeauty = treeSet.last().height - treeSet.first().height;
        System.out.print(minBeauty + " " + i);
        System.out.print('\n');
        for (int j = 0;j < i;j++) {
            System.out.println(move[j][0] + " " + move[j][1]);
        }
    }

    static class HeightAndNum implements Comparable{
        int height;
        int num;
        public HeightAndNum(int height, int num) {
            this.height = height;
            this.num = num;
        }

        @Override
        public int compareTo(Object o) {
            HeightAndNum other = (HeightAndNum) o;
            if (this.height > other.height) {
                return 1;
            } else if (this.height == other.height) {
                return this.num - other.num;
            }
            return -1;
        }
    }
}

标签：柱子,积木,int,乐高,height,num,HeightAndNum,PIPI,treeSet
来源： https://blog.csdn.net/qq_44709990/article/details/120173692

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。