用排序+双指针
如果两个数组是有序的,则可以使用双指针的方法得到两个数组的交集。
首先对两个数组进行排序,然后使用两个指针遍历两个数组。
初始时,两个指针分别指向两个数组的头部。每次比较两个指针指向的两个数组中的数字,如果两个数字不相等,则将指向较小数字的指针右移一位,如果两个数字相等,将该数字添加到答案,并将两个指针都右移一位。当至少有一个指针超出数组范围时,遍历结束。
复杂度分析
时间复杂度:O(m *log m+n *log n)O(mlogm+nlogn),其中 m 和 n 分别是两个数组的长度。对两个数组进行排序的时间复杂度是 O(mlogm+nlogn),遍历两个数组的时间复杂度是 O(m+n)O(m+n),因此总时间复杂度是 O(mlogm+nlogn)。
空间复杂度:O(min(m,n)),其中 m 和 n 分别是两个数组的长度。为返回值创建一个数组 intersection,其长度为较短的数组的长度。不过在 C++ 中,我们可以直接创建一个 vector,不需要把答案临时存放在一个额外的数组中,所以这种实现的空间复杂度为O(1)。
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int cmp(int *a,int *b){
return *a-*b;
}
int* intersect(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize){
qsort(nums1,nums1Size,sizeof(int),cmp);
qsort(nums2,nums2Size,sizeof(int),cmp);
int q=0,p=0,i=0;
int *a=(int *)malloc((nums1Size+nums2Size)*sizeof(int));
while(q<nums1Size&&p<nums2Size){
if(nums1[q]==nums2[p]){
a[i++]=nums2[p];
q++;p++;
}else if(nums1[q]<nums2[p]){
q++;
}else{
p++;
}
}
*returnSize=i;
return a;
}