349. Intersection of Two Arrays
点击查看350. Intersection of Two Arrays II
原题链接
Given two arrays, write a function to compute their intersection.
给两个数组,写一个函数返回交集
Example:
Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2].
Note:
- Each element in the result must be unique.(结果中的每个元素必须是唯一的)
- The result can be in any order.(结果可以是任何顺序)
思路1:
- 设置一个数组res用来存放nums1内每个元素的个数,并将其赋值给元素大小对应的位置(例: nums1里面有5个3,则res[2] = 5)
- 遍历nums2内元素,看元素大小对应位置是否为0,若不是,则将该元素大小在res中对应的位置添加至交集数组,并将其赋值为0
- 最后返回交集数组
- 简单的小数值数组倒是能够用,但是数组内最大元素大于1000,这个算法就不能用了
#include <iostream>
#include <vector>
#include<algorithm>
using namespace std;
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
vector <int> res(1000, 0);//处理不了大量数据,不符合题意
vector <int> intersectionArray;//交集数组
for(char m : nums1)
res[m]++;
for(char n : nums2){
if(res[n] != 0){
intersectionArray.push_back(n);//将两数组内相同的数存入数组intersectionArray中
res[n] = 0;//赋值为0是为了不存入相同元素
//例如: nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2]. 第二个2不存入
}
}
return intersectionArray;
}
};
int main()
{
Solution a;
int A[4] = {2, 8, 11, 15};
int B[4] = {2, 7, 21, 15};
vector<int> nums1(A, A+4);
vector<int> nums2(B, B+4);
vector<int> intersectionArray = a.intersection(nums1, nums2);
for(int i=0; i<intersectionArray.size(); i++)
cout << intersectionArray[i] << " ";
return 0;
}
思路2:
- 由于思路1不能处理大量数据,只能用不定长的数组
- 先将数组排序,同时遍历两个数组,碰见相同值就输入交集数组,并且派出两有序数组前后两个元素均相等的情况。
vector<int> intersection1(vector<int>& nums1, vector<int>& nums2) {
vector <int> intersectionArray;//交集数组
if(nums1.empty() || nums2.empty())//判断数组是否为空
return intersectionArray;
sort(nums1.begin(), nums1.end());//底层使用了快速排序,O(nlogn)
sort(nums2.begin(), nums2.end());
int m = nums1.size();
int n = nums2.size();
for(int i=0,j=0; i<m||j<n; ){//任何一个数组遍历完就结束循环
if(nums1[i] == nums2[j]){
if(i==0 || j==0){//若是数组首元素直接加入交集数组
intersectionArray.push_back(nums1[i]);
i++; j++;
continue;
}
//排除两个数组内均有两个或两个以上相同元素的情况。例如nums1=[1, 2, 2, 1],nums2=[2, 2]
else if(nums1[i]!=nums1[i-1]){
intersectionArray.push_back(nums1[i]);
i++; j++;
continue;
}
//两个数组内均有两个或两个以上相同元素,i++ j++ 直接开始下一个循环
else{
i++; j++;
continue;
}
}
else if(nums1[i] > nums2[j])
j++;
else
i++;
}
return intersectionArray;
}