本题来源于Leetcode网,原题链接如下:
两个数组的交集
描述:
给定两个数组 nums1 和 nums2 ,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
示例1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
示例2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
解释:[4,9] 也是可通过的
数据范围:
1 <= nums1.length, nums2.length <= 1000
0 <= nums1[i], nums2[i] <= 1000
题解:
这里我采用的是创建计数数组来记录出现的数,具体的思路过程如下:
根据数据范围申请计数数组为:int count[1001] = {0};
然后动态申请一个数组ret用以返回,这个数组的大小可以取所给两个数组中数组长度较小的一个,此部分代码如下;
int *ret = (int*)malloc(sizeof(int)*(nums1Size > nums2Size ? nums1Size : nums2Size));
接着遍历数组1,记录出现的元素,记录方法是把出现的数作为数组count对应的下标,然后该下标对应count中的元素置为1(如nums[0] = 2, count[2] = 1),需要注意的是这里是置1而不是让对应元素++,原因后面会提到。此部分代码如下:
for(i = 0; i < nums1Size; i++) //遍历数组1,记录出现的元素
{
count[nums1[i]] = 1; //这里置1,而不++,以便后面操作
}
接着遍历数组2,和遍历数组1使用同样的方法原理,只不过此次遍历是查看计数数组中出现了哪些数,从而找到两集合中相同的数(交集)。当判断出现了某个数时就将其存入返回数组中并改变记录值。这样在下一次判断时,如果数组2中的数作为下标对应count中的元素不为1,那么此时该数要么就是在数组1中没有出现(不属于交集部分);要么就是已经出现过一次了;如此一来就保证了返回数组中每个元素唯一的要求。此部分代码如下:
for(i = 0; i < nums2Size; i++) //遍历数组2,找到相同的数
{
if(count[nums2[i]] == 1)
{
ret[(*returnSize)] = nums2[i];
count[nums2[i]]++; //找到相同的数后改变记录值,确保每个元素的唯一
(*returnSize)++;
}
}
- 完整代码如下:
int* intersection(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize)
{
int count[1001] = {0};
int i = 0;
*returnSize = 0;
int *ret = (int*)malloc(sizeof(int)*(nums1Size > nums2Size ? nums1Size : nums2Size));
//返回的数组的大小最大也不会超过原两个数组中最小的
for(i = 0; i < nums1Size; i++) //遍历数组1,记录出现的元素
{
count[nums1[i]] = 1; //这里置1,而不++,以便后面操作
}
for(i = 0; i < nums2Size; i++) //遍历数组2,找到相同的数
{
if(count[nums2[i]] == 1)
{
ret[(*returnSize)] = nums2[i];
count[nums2[i]]++; //找到相同的数后改变记录值,确保每个元素的唯一
(*returnSize)++;
}
}
return ret;
}
看完觉得有觉得帮助的话不妨点赞收藏鼓励一下,有疑问或看不懂的地方或有可优化的部分还恳请朋友们留个评论,多多指点,谢谢朋友们!🌹🌹🌹