优势洗牌-leetcode

题目介绍

给定两个大小相等的数组 nums1 和 nums2，nums1 相对于 nums2 的优势
可以用满足 nums1[i] > nums2[i] 的索引 i 的数目来描述。返回 nums1 
的任意排列，使其相对于 nums2 的优势最大化。

示例 1：
输入：nums1 = [2,7,11,15], nums2 = [1,10,4,11]
输出：[2,11,7,15]
示例 2：
输入：nums1 = [12,24,8,32], nums2 = [13,25,32,11]
输出：[24,32,8,12]

分析过程

此问题类似于“田忌赛马”，当别人的马强于自己时，选择最弱的一匹马出战，尽量保存最大实力，当自己的马强于别人时，选择只比对方强一点的一匹马出战，有一点点贪心的意思。
题目中，nums1的值即代表己方，num2的值代表对方

解题步骤：
	1、对num1己方数据排序（从小到大）
	2、按照对方num2顺序选择对应的值，选择时应该遵循以下原则：
		a、对方值小于我方最小值或者大于我方最大值，选择最小值
		b、当对方值n在我方最大、最小值之间时，采用二分查找找到第一个大于n的值
	3、num1中删除此值，并按顺序放入结果中

代码示例

func advantageCount(nums1 []int, nums2 []int) []int {
    sort.Ints(nums1)
    res := make([]int, 0)
    for _, n2 := range nums2 {
        index := binarySearch(nums1, n2)
        res = append(res, nums1[index])
        // 选择末尾
        if index == len(nums1)-1{
            nums1 = nums1[0:index]
        }else if index ==0 {
            //选择首位
            nums1 = nums1[index+1:]
        }else {
        	//选择中间
            nums1 = append(nums1[0:index], nums1[index+1:]...)
        }
    }
    return res
}

//在nums中找到第一个大于n的值
//如果未找到返回最小值
func binarySearch(nums []int, n int) int{
    l := len(nums)-1
    if n < nums[0] || n>=nums[l]{
        return 0
    }
    i:=0
    j:= l
    for i<j {
        mid := (i+j)/2
        if nums[mid] < n {
            i = mid + 1
        }else if nums[mid]>n{
            j = mid -1
        }else{
            break
        }
    }
    // 存在相等的值
    if i< j {
        mid := (i+j)/2
        k := mid +1
        for k<len(nums)&&nums[k]==nums[mid]{
            k++
        }
        return k
    }
    //不存在相等的值，找到第一个大于n的值
    for j<len(nums)&&nums[j]<=n{
        j++
    }
    return j
}

复杂度分析

时间复杂度：O(nlogn)，由于按nums2遍历一遍，复杂度O(n)，且对nums1采用二分查找复杂度O(logn)，二者相乘为O(nlogn)
空间复杂度：O(1)，未使用新内存空间