LeetCode刷题笔记 870. 优势洗牌

最新推荐文章于 2023-02-24 14:07:55 发布

原创最新推荐文章于 2023-02-24 14:07:55 发布 · 161 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode笔记专栏收录该内容

156 篇文章

订阅专栏

本文介绍了一种算法，通过合理排列数组A，使其相对于数组B的优势最大化。具体实现包括排序、使用Map和Deque数据结构，确保A中元素尽可能大于B中对应位置的元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

给定两个大小相等的数组 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;
    }
}