LeetCode 870. 優勢洗牌 JAVA 題解

LeetCode 870. 優勢洗牌 JAVA 題解

題目鏈接：https://leetcode-cn.com/problems/advantage-shuffle/

思想： 田忌賽馬

每次拿A的“當前輪次“的最小值和B的”當前輪次“最小值比較，
若大於，則OK，滿足。（1)
若小於，則將A的值去和B的”最大值“配對。(2)

技巧：

• 將B的“值”和“索引”封裝爲Node對象，便於排序後找到原索引位置，個人認爲比用其他方式好一些。
• 使用LinkedList維護封裝好的B的Node對象，方便移除頭尾節點(結合思想中的兩步，頭尾節點即對應最小，最大值)

代碼

import java.util.*;
class Solution {
    public int[] advantageCount(int[] A, int[] B) {
        Arrays.sort(A);
        LinkedList<Node> listB = new LinkedList<>();
        for(int i = 0; i < B.length; i++){
            listB.add(new Node(B[i],i));
        }
        Collections.sort(listB,new Comparator<Node>(){
            public int compare(Node n1, Node n2){
                return n1.value - n2.value;
            }
        });
        
        // 遍歷A即可，將B數組作爲輸出容器，因爲B的信息已經都存在LinkedList裏了，這裏B數組已經沒用了
        for(int i = 0; i < A.length; i++){
            if(A[i] > listB.getFirst().value){
                B[listB.removeFirst().index] = A[i]; // 對應思想中的（1）
            }else{
                B[listB.removeLast().index] = A[i]; // // 對應思想中的（2）
            }
        }
        return B;
    }
}

class Node{
    int value;
    int index;
    public Node(int value, int index){
        this.value = value;
        this.index = index;
    }
}