题目描述:
给定两个数组,编写一个函数来计算它们的交集。
示例1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2,2]
示例2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[4,9]
说明:
输出结果中每个元素出现的次数,应与元素在两个数组中出现次数的最小值一致。
我们可以不考虑输出结果的顺序。
代码:
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
vector<int> v;
int n=nums1.size(),s=nums2.size();
sort(nums1.begin(),nums1.end());
sort(nums2.begin(),nums2.end());
int x=0,y=0;
int flag1[n],flag2[s];
memset(flag1,0,sizeof(flag1));
memset(flag2,0,sizeof(flag2));
while(true){
if(x==n-1&&y==s-1){
if(nums1[x]==nums2[y]){
if(flag1[x]==0&&flag2[y]==0)
v.push_back(nums1[x]);
break;
}
}
if(nums1[x]==nums2[y]){
cout<<"等于"<<" ";
if(flag1[x]==0&&flag2[y]==0){
cout<<"变flag"<<endl;
v.push_back(nums1[x]);
flag1[x]=1;
flag2[y]=1;
}
if(x<n-1)x++;
if(y<s-1)y++;
}
else if(nums1[x]>nums2[y]){
if(y==s-1)break;
if(y<s-1)y++;
}
else if(nums1[x]<nums2[y]){
if(x==n-1)break;
if(x<n-1)x++;
}
}
return v;
}
};
比上一题的两数组求交集,这一题在说明中增加了”输出结果中每个元素出现的次数,应与元素在两个数组中出现次数的最小值一致“的要求。
例如[1,9,9]和[1,9,9,9,9]的交集为[1,9,9],因为9在两个数组中最少出现的次数为2次,所以交集中只有两个9。即数组a中已经在数组b中匹配过的元素,不能再出现在交集中,即每个元素只能查找匹配一次。
只需给两个数组每个元素增加标志,默认全为0。若该元素已经被当作交集元素加入vector中,则修改标志值为1;只有当两个数组中元素标志值都为0时才可以被加入vector。