LeetCode 870. 優(yōu)勢洗牌

給定兩個大小相等的數(shù)組 A 和 B，A 相對于 B 的優(yōu)勢可以用滿足 A[i] > B[i] 的索引 i 的數(shù)目來描述。
返回 A 的任意排列，使其相對于 B 的優(yōu)勢最大化。

示例 1:
輸入：A = [2,7,11,15], B = [1,10,4,11]
輸出：[2,11,7,15]

示例 2:
輸入：A = [12,24,8,32], B = [13,25,32,11]
輸出：[24,32,8,12]

提示：
1 <= A.length = B.length <= 10000
0 <= A[i] <= 10^9
0 <= B[i] <= 10^9

思路 : 田忌賽馬
排序倆個數(shù)組，然后比較
記得使用map記錄排序前的位置信息

    public int[] advantageCount(int[] A, int[] B) {
        if(A.length!=B.length){
            return new int[0];
        }
        Map<Integer,LinkedList<Integer>> map = new HashMap<>();
        for(int x=0;x<B.length;x++){
            if(!map.containsKey(B[x])){
                map.put(B[x],new LinkedList<>());
            }
            map.get(B[x]).add(x);
        }

        Arrays.sort(A);
        Arrays.sort(B);

        int[] answer = new int[A.length];

        int indexA = 0;
        int indexB = 0;
        int pos = 0;
        int last = A.length-1;

        while(indexA<A.length){
            if(A[indexA]>B[indexB]){
                answer[pos++] = A[indexA++];
                indexB++;
            }else{
                answer[last--] = A[indexA++];
            }
        }

        int[] result = new int[answer.length];

        for(int x=0;x<result.length;x++){
            result[map.get(B[x]).pop()] = answer[x];
        }

        return result;
    }

總結(jié)

*仔細觀察 找到規(guī)律

反饋與建議

• QQ：[3441242166]
• 郵箱：3441242166@qq.com