【leetcode C语言实现】350.两个数组的交集II

题目描述

给定两个数组，编写一个函数来计算它们的交集。

示例 1：

输入：nums1 = [1,2,2,1], nums2 = [2,2]
输出：[2,2]
示例 2:

输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[4,9]

说明：

输出结果中每个元素出现的次数，应与元素在两个数组中出现次数的最小值一致。
我们可以不考虑输出结果的顺序。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/intersection-of-two-arrays-ii

解题思路

本题需要求两个数组的交集，可考虑将两个数组都进行排序，再通过双指针依次比较排序后的两个数组元素大小，若相等，即为交集中的元素，将其加入一个新的数组，最终返回这个数组。

代码

int cmp_func(const void* a, const void* b)
{
	return (*(int*)a - *(int*)b);
}

int* intersect(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize)
{
	qsort(nums1, nums1Size, sizeof(int), cmp_func);
	qsort(nums2, nums2Size, sizeof(int), cmp_func);

	int new_size = (nums1 < nums2 ? nums1 : nums2);
	int* intersect_arr = (int*)malloc(new_size * sizeof(int));
	int i = 0, j = 0;
	*returnSize = 0;

	while ((i < nums1Size) && (j < nums2Size))
	{
		if (nums1[i] < nums2[j])
		{
			i++;
		}
		else if (nums1[i] > nums2[j])
		{
			j++;
		}
		else
		{
			intersect_arr[(*returnSize)++] = nums1[i];
			i++;
			j++;
		}
	}
	return intersect_arr;
}

测试代码及结果

void print_arr(int *arr, int size)
{
	int i;

	for (i = 0; i < size; i++)
	{
		printf("%d ", arr[i]);
	}
	printf("\n");
}

int main()
{
	// ========功能测试=========
	// 1. 两个数组有交集
	int nums1_1[] = { 1, 2, 2, 1 };
	int nums1_2[] = { 2, 2 };
	int nums1_1_size = sizeof(nums1_1) / sizeof(int);
	int nums1_2_size = sizeof(nums1_2) / sizeof(int);
	int size1 = 0;
	int* res1 = intersect(nums1_1, nums1_1_size, nums1_2, nums1_2_size, &size1);
	printf("数组有交集：\n");
	print_arr(res1, size1);

	// 2. 两个数组没有交集
	int nums2_1[] = { 11, 5, 8, 3 };
	int nums2_2[] = { 2, 7, 6 };
	int nums2_1_size = sizeof(nums2_1) / sizeof(int);
	int nums2_2_size = sizeof(nums2_2) / sizeof(int);
	int size2 = 0;
	int* res2 = intersect(nums2_1, nums2_1_size, nums2_2, nums2_2_size, &size2);
	printf("数组没有交集：\n");
	print_arr(res2, size2);


	// =======特殊输入测试======
	// 输入空指针
	int nums3_1 = NULL;
	int nums3_2[] = { 1, 4 };
	int nums3_1_size = 0;
	int nums3_2_size = sizeof(nums3_2) / sizeof(int);
	int size3 = 0;
	int* res3 = intersect(nums3_1, nums3_1_size, nums3_2, nums3_2_size, &size3);
	printf("空指针：\n");
	print_arr(res3, size3);

	return 0;
}

执行结果

时间复杂度：采用快速排序对两个数组进行排序所使用的时间复杂度为mlog(m) + nlog(n)，遍历比较时时间复杂度为O(m+n)，因此总复杂度为mlog(m) + nlog(n)。
空间复杂度：创建输出数组所用大小为两个数组中数组长度较小的值，因此空间复杂度为O(min(m, n))。