Java資料結構優先佇列實練

2022-07-21 14:03:11

最後一塊石頭的重量

題目描述

思路詳解

這裡採用最大堆進行解題。

我們首先考慮，每次拿出兩個最大的進行比較，然後大的減去小的重新放入不就完成了嘛。

首先我們建立一個優先佇列，遍歷重量，放入佇列。依次取出重量最大的和第二大的，如果a>b就把a-b重新放入。直到佇列裡面的元素只剩1個的時候，輸出結果。

程式碼與結果

class Solution {
    public int lastStoneWeight(int[] stones) {
        PriorityQueue<Integer> pq = new PriorityQueue<Integer>((a, b) -> b - a);
        for (int stone : stones) {
            pq.offer(stone);
        }
        while (pq.size() > 1) {
            int a = pq.poll();
            int b = pq.poll();
            if (a > b) {
                pq.offer(a - b);
            }
        }
        return pq.isEmpty() ? 0 : pq.poll();
    }
}

裝滿杯子需要的最短總時長

題目描述

思路詳解

這個題也是思考了很久。

分兩種情況：

第一種：很好想，有一種水特別多，那麼答案就是這種水的杯數。

第二種：就是一定可以匹配完成或者匹配到只剩一杯。

我們只需要先排序，在分情況就好。

程式碼與結果

class Solution {
    public int fillCups(int[] amount) {
        Arrays.sort(amount);
        int sum=amount[0]+amount[1]+amount[2];
        if(amount[1]+amount[0]>=amount[2]) sum=(sum+1)/2;
        else sum=amount[2];
        return sum;
    }
}

移除石子的最大得分

題目描述

思路詳解

本題的思路看起來簡單，也不是很好想。

我們先排一下序，兩種情況：

第一種：前兩個的和小於第三個，這時候我們一直拿最後一堆和任意一堆，結果就是a+b。

第二種：前兩個數的和大於第三個數，那麼前兩堆一定可以內部抵消一部分。只需總和除以2就好。

程式碼與結果

class Solution {
	public int maximumScore(int a, int b, int c) {
		int[] arr = new int[] { a, b, c };
		Arrays.sort(arr);
		a = arr[0];
		b = arr[1];
		c = arr[2];
		if (a + b <= c) {
			return a + b;
		} else {
			return (a + b + c) / 2;
		}
	}
}