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 <= A.length = B.length <= 10000
0 <= A[i] <= 10^9
0 <= B[i] <= 10^9

思路：有点像田忌赛马的优化版

先让自己最弱的马和对面最弱的马跑一下，如果能赢，那就他两跑，如果跑不赢，就让这匹马和对面最强的马赛跑

这道题如何实现这个想法呢？

先对两个数组进行排序，但是呢得记录下b的索引，这样才知道一会把马往哪个位置放

如果a最小的数大于等于b最小的数，则把它放在b最小的数的索引处。

反之，则把它放在b最大的数的索引处。

提交的代码：

class Temp implements Comparable<Temp >{
    int num, index; 
    public Temp(int num, int index) {
        this.index = index;
        this.num = num;
    }
    @Override
    public int compareTo(Temp o) {
        return this.num - o.num;
    }
}
public class Solution {
public static int[] advantageCount(int[] A, int[] B) {
        Arrays.sort(A);
        Temp c[] = new Temp[B.length];
        int i=0,j=A.length-1,k=0;
        int[] d = new int[A.length];
        for(i=0;i<B.length;i++)
        {
            c[i] = new Temp(B[i],i);
        }
        Arrays.sort(c);
        i=0;
        while(k<=j)
        {
            if(A[i]>c[k].num)
            {
                d[c[k].index] = A[i];
                i++;
                k++;
            }
            else
            {
                d[c[j].index] = A[i];
                i++;
                j--;
            }
        }
        return d;
    }
}

完整的代码：

import java.util.Arrays;
class Temp implements Comparable<Temp >{
    int num, index; 
    public Temp(int num, int index) {
        this.index = index;
        this.num = num;
    }
    @Override
    public int compareTo(Temp o) {
        return this.num - o.num;
    }
}
public class Solution870 {
public static int[] advantageCount(int[] A, int[] B) {
        Arrays.sort(A);
        Temp c[] = new Temp[B.length];
        int i=0,j=A.length-1,k=0;
        int[] d = new int[A.length];
        for(i=0;i<B.length;i++)
        {
            c[i] = new Temp(B[i],i);
        }
        Arrays.sort(c);
        i=0;
        while(k<=j)
        {
            if(A[i]>=c[k].num)
            {
                d[c[k].index] = A[i];
                i++;
                k++;
            }
            else
            {
                d[c[j].index] = A[i];
                i++;
                j--;
            }
        }
        return d;
    }
public static void main(String[] args)
{
    int a[] = {12,24,8,32};
    int b[] = {13,25,32,11};
    int c[] = new int[a.length];
    c = advantageCount(a,b);
    for(int i = 0;i<c.length;i++)
    {
        System.out.println(c[i]+" ");
    }
}

}