标签:you ++ 实测 cannot app cto note pre Once
Given two arrays, write a function to compute their intersection.
Example:
Given nums1 = [1, 2, 2, 1]
, nums2 = [2, 2]
, return [2, 2]
.
Note:
Follow up:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2)
1、这道题给定两个vector,要求返回两个vector的交集,比如nums1=[1,2,2,1],nums2=[2,2],返回的交集是[2,2],其中有多少个相同的元素就返回多少个。返回的交集不讲究顺序。
2、这道题看完题意,熟悉leetcode的同学应该会马上想到先排序,排序之后的两个vector来比较,时间复杂度会下降很多。
如果不排序,那就是双重循环的做法,O(n^2),时间复杂度太高了。
先排序再比较的做法,代码如下:(附详解)
vector<int> intersect(vector<int>& nums1, vector<int>& nums2)
{
sort(nums1.begin(),nums1.end());//给nums1排序,升序
sort(nums2.begin(),nums2.end());//给nums2排序,升序
int s1=nums1.size(),s2=nums2.size(),i=0,j=0;//i表示nums1元素的位置,j表示nums2元素的位置
vector<int>res;//存储最后结果的vector
while(i<s1&&j<s2)//两个vector一旦有一个遍历完了,那么就结束比较
{
if(nums1[i]<nums2[j])
{
while(nums1[i]<nums2[j]&&i<s1)//一直找,直到nums1[i]>=nums2[j]
i++;
if(i==s1)//如果i已经到了nums1的外面,那么结束比较
break;
}
else if(nums1[i]>nums2[j])
{
while(nums1[i]>nums2[j]&&j<s2)//一直找,直到nums2[j]>=nums1[i]
j++;
if(j==s2)//如果j已经到了nums2的外面,那么结束比较
break;
}
if(nums1[i]==nums2[j])//如果刚好相等,那么插入到res中,更新i和j的值
{
res.push_back(nums1[i]);
i++;
j++;
}
}
return res;
}
上述代码实测7ms,beats 98.05% of cpp submissions。
leetcode-350-Intersection of Two Arrays II(求两个数组的交集)
标签:you ++ 实测 cannot app cto note pre Once
原文地址:https://www.cnblogs.com/king-3/p/9122890.html