题目介绍
给定两个大小相等的数组 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),未使用新内存空间