题目描述:
给定两个数组,编写一个函数来计算它们的交集。
示例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。