【算法】【贪心算法】【leetcode】870. 优势洗牌

时间:2024-05-01 13:46:00

在这里插入图片描述
题目地址:https://leetcode.cn/problems/advantage-shuffle/description/

题目描述:

给定两个长度相等的数组 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]

提示:

1 <= nums1.length <= 10^5
nums2.length == nums1.length
0 <= nums1[i] ,nums2[i] <= 10^9

解题思路(典型贪心算法)

田忌赛马的故事大家应该都听说过: 田忌和齐王赛马,两人的马分上中下三等,如果同等级的马对应着比赛,田忌赢不了齐王。但是田忌遇到了孙膑,
孙膑就教他用自己的下等马对齐王的上等马,再用自己的上等马对齐王的中等马,最后用自己的中等马对齐王的下等马,结果三局两胜,田忌赢了。
田忌赛马的核心思路就是打得过就打,打不过就拿自己的垃圾和对方的精锐互换。

把nums1当成是田忌的马,nums2当成是齐威王的马。 讨论田忌的下等马(nums的最小值):

  • 如果它能比过齐威王的下等马(nums的最小值),那这一分田忌直接拿下;

  • 如果它比不过齐威王的下等马,则用田忌的下等马比齐威王的上等马(mums2的最大值)。

去掉这两匹马,问题变成一个规模更小(n-1)的子问题。重复上述过程,即得到了所有马的对应 关系。

代码实现时,由于num2不能排序,我们可以创建一个下标数组ids,对ids排序,即ids[0]对应
nums2中最小值的下标,ids[1]对应num2中第二小值的下标。用双指针操作ids,从而知道
每个下标所要对应的nums1的元素,也就找到了所要求的nums1的排列。

解题思路来自:(灵茶山艾府)https://leetcode.cn/problems/advantage-shuffle/solutions/1/tian-ji-sai-ma-by-endlesscheng-yxm6/

代码实现

public class Solution{
	 public int[] advantageCount(int[] nums1, int[] nums2) {
        //先对nums1进行排序
        Arrays.sort(nums1);

		//对muns2排序 但是mums2不能直接排序 需要额外借助一个数据排序
		int nums2Len = nums2.length;
		int [] ids = new int [nums2Len];
		
		//记录nums2的下标
		for(int i =0;i<n;i++){
			ids[i]=i;
		}

		//将num2进行排序 注意这里不能直接对nums2排序 转对nums2的下标排序代替nums2的顺序
		//升序排列 (降序也是一个样)
		Arrays.sort(ids,(i,j)->nums2[i]-nums2[j]);

		//赛马:打得过就打,打不过就拿自己的垃圾和对方的精锐互换
		int [] ans = new int[nums1.length];
		int right = nums2Len;
		int left = 0;
		for (int x : nums1) {
          ans[x > nums2[ids[left]] ? ids[left++] : ids[right--]] = x;
        }
	return ans;
    }
}

在这里插入图片描述