【leetcode】 两个数组的交集 IIc++

题目描述:
给定两个数组,编写一个函数来计算它们的交集。

示例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。
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_40315080/article/details/120565924