LeetCode刷题笔记 870. 优势洗牌

题目描述

给定两个大小相等的数组 A 和 B，A 相对于 B 的优势可以用满足 A[i] > B[i] 的索引 i 的数目来描述。

返回 A 的任意排列，使其相对于 B 的优势最大化。

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

总结

两个数组先排序，把对位的A的大于B的放进 队列 暂存，小的统一放入另一个，最后循环一边放入返回数组

Sample Code

class Solution {
    public int[] advantageCount(int[] A, int[] B) {
        int[] sortedA = A.clone();
        Arrays.sort(sortedA);
        int[] sortedB = B.clone();
        Arrays.sort(sortedB);
        
        Map<Integer,Deque<Integer>> assigned = new HashMap();
        for(int b:B) assigned.put(b,new LinkedList());	//assigned里面放的是未排序的B的数组的值
        
        Deque<Integer> remaining = new LinkedList();
        
        int j=0;
        for(int a:sortedA){
            if(a>sortedB[j]){
                assigned.get(sortedB[j++]).add(a);
            }else{
                remaining.add(a);
            }
        }
        
        int[] ans=new int[B.length];
        for(int i=0;i<B.length;++i){
            if(assigned.get(B[i]).size()>0)
                ans[i]=assigned.get(B[i]).pop();
            else
                ans[i]=remaining.pop();
        }
        return ans;
    }
}