首页 > 学院 > 开发设计 > 正文

349. Intersection of Two Arrays

2019-11-11 02:15:23
字体:
来源:转载
供稿:网友

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.

class Solution {public: vector<int> intersection(vector<int>& nums1, vector<int>& nums2) { sort(nums1.begin(), nums1.end()); sort(nums2.begin(), nums2.end()); set<int> s; for(int i = 0, j = 0; i < nums1.size() && j < nums2.size(); ){ if(nums1[i] < nums2[j]) ++i; else if(nums1[i] > nums2[j]) ++j; else { s.insert(nums1[i]); ++i; ++j; } } return vector<int>(s.begin(), s.end()); }};
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表