leetcode 870. 优势洗牌

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

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

 

示例 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. 1 <= A.length = B.length <= 10000
  2. 0 <= A[i] <= 10^9
  3. 0 <= B[i] <= 10^9
class Solution {
    public int[] advantageCount(int[] A, int[] B) {
       Map> map = new HashMap<>();
        int[] b = Arrays.copyOf(B, B.length);
        Arrays.sort(A);
        Arrays.sort(B);
        int i = 0;
        int j = B.length - 1;
        int k = A.length -1;
        while(j >= 0 && i <= k){
            if(B[j] < A[k]){
                if(map.containsKey(B[j])){
                    map.get(B[j--]).add(A[k--]);
                }else{
                    List l = new ArrayList<>();
                    l.add(A[k]);
                    map.put(B[j--], l); 
                    k--;
                }

            }else{
                if(map.containsKey(B[j])){
                    map.get(B[j--]).add(A[i++]);
                }else{
                    List l = new ArrayList<>();
                    l.add(A[i]);
                    map.put(B[j--], l); 
                    i++;
                }
            }
        }
        for(int l = 0; l < b.length; l++){
            List list = map.get(b[l]);
            A[l] = list.get(0);
            if(list.size() > 1) {
	            	list.remove(0);
	        }
        }
        return A;
    }
}

思路:依次比较,如果A>B,则B这个数对应的就是A的值,如果A<=B,取最小A值给B的值

你可能感兴趣的:(JAVA,leetcode)