其他分享
首页 > 其他分享> > 集合框架题:PIPI的乐高积木

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

作者:互联网

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

问题:

在这里插入图片描述
在这里插入图片描述

思路:

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

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

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

需注意的点

代码:

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