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;
    }
}