目录
题目链接:349. 两个数组的交集 - 力扣(LeetCode)
题目链接:350. 两个数组的交集 II - 力扣(LeetCode)
题目链接:349. 两个数组的交集 - 力扣(LeetCode)
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
vector<int> record(1001);
vector<int> res;
//本来想用++,但不行
//看了题解才明白直接赋值得了
for(int i=0;i<nums1.size();i++)
{
record[nums1[i]]=1;
}
for(int i=0;i<nums2.size();i++)
{
if(record[nums2[i]]==1)
{
record[nums2[i]]=2;
}
}
for(int i=0;i<record.size();i++)
{
if(record[i]==2)
{
res.push_back(i);
}
}
return res;
}
};
题目链接:350. 两个数组的交集 II - 力扣(LeetCode)
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
vector<int> res;
vector<int> record1(1001);
vector<int> record2(1001);
for(int i=0;i<nums1.size();i++)
{
record1[nums1[i]]++;
}
for(int i=0;i<nums2.size();i++)
{
record2[nums2[i]]++;
}
int j;
for(int i=0;i<record1.size();i++)
{
j=1;
while(record1[i]>0&&record2[i]>0&&j<=record2[i]&&j<=record1[i])
{
// cout<<record1[i]<<endl;
res.push_back(i);
j++;
}
}
return res;
}
};
看题解后
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
vector<int> res;
unordered_map<int,int> m;
//计算nums1中的元素出现次数
for(int i=0;i<nums1.size();i++)
{
m[nums1[i]]++;
}
//当nums2中的元素包含在nums1中时,把num加入结果数组,当nums1中的该元素出现次数-完之后,不再加入结果数组
//因为nums2是挨个遍历的,所以nums2的元素次数<nums1时自动不会再加入结果数组了
for(int num:nums2)
{
if(m.count(num))
{
res.push_back(num);
m[num]--;
if(m[num]==0)
{
m.erase(num);
}
}
}
return res;
}
};