链接:https://leetcode.cn/problems/advantage-shuffle/solution/-by-xun-ge-v-jjps/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
题目
思路
题意 -> 将 nums1 中的元素放入 nums2 中,原则为:尽可能的保证 nums1 的元素相对于相同位置的 nums2 中的元素大并且大的个数尽可能多
贪心
尽可能多可以用贪心思想,每次我们都寻找 nums2 中比 nums1 大一点点的元素,当 nums2 中没有比 nums1
中更大的元素时,我们将这个大元素添加到 nums2 中的可用的最小值位置。
桶排序
在寻找第一个更大值时,有很多方法,可以先排序nums1,然后使用循环枚举 nums2 中的每一个元素选择第一个更大的元素,但是这种方法每次选择nums1中的一个元素就需要枚举一次nums2,时间销毁非常大。
优化居然nums1可以排序,那么nums2我们也可以排序呀,可以定义一个新的结构体(桶),里面装nums2的元素和对于下标,然后对这个桶按元素值进行排序,就可以快速找到第一个更大元素同时还可以找到对应下标
代码
struct side
{
int val;
int tige;
}; // 这个结构体的目的是用来存B数组的值和索引
int cmp(const void * a, const void * b) // 这个是结构体的比较器
{
struct side * c, * d;
c = (struct side *) a;
d = (struct side *) b;
return c -> val - d -> val;
}
int cmp1(const void * a, const void * b) // A数组的比较器
{
return *(int *) a - *(int *) b;
}
int* advantageCount(int* A, int ASize, int* B, int BSize, int* returnSize){
int i, l = 0, r = BSize - 1; // r作用是记录排好序B数组最大的值的位置
struct side * head;
int *list; // 是用来还回的数组
// 创数组
list = (int *)malloc(sizeof(int) * (ASize));
head = (struct side *)malloc(sizeof(struct side) *(BSize));
for (i = 0; i < BSize; i++) // 把B数组的值赋给结构体
{
head[i].val = B[i];
head[i].tige = i;
}
qsort(A, ASize, sizeof(int), cmp1); // 进行排列
qsort(head, BSize, sizeof(struct side), cmp);
for (i = 0; i < ASize; i++)
{
if(A[i] > head[l] . val) // 如果A数组的值大于B数组的值则把A[i]的值赋给list数组中的B数组的索引。
{
list[head[l] . tige] = A[i];
l++;
}
else
{
list[head[r] . tige] = A[i]; // 如果是小于等于则把A[i]的值放在list数组中B数组最大的索引位置。
r--;
}
}
*returnSize = ASize;
return list;
}
作者:xun-ge-v
链接:https://leetcode.cn/problems/advantage-shuffle/solution/-by-xun-ge-v-jjps/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。