集合框架题：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