问题描述:
返回两个数组的交
解题思路:
对两个向量里的元素排序,然后将两个向量里的元素进行比较,如果找到有两个相等的,在判断是否还有,将找到的元素插入到另一个新的向量里,然后用temp记录当前元素,便于后面的判断,两个元素不相等,继续向下找。
实现代码:
class Solution {
public:
/**
* @param nums1 an integer array
* @param nums2 an integer array
* @return an integer array
*/
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
// Write your code here
vector<int> nums3;
int i=0,j=0;
int m=nums1.size();
int n=nums2.size();
if(m==0||n==0) {
return nums3;
}
sort(nums1.begin(),nums1.end());
sort(nums2.begin(),nums2.end());
int temp;
while(i<=m-1&&j<=n-1){
if(nums1[i]==nums2[j])
{
if(nums1[i]!=temp)
{
nums3.push_back(nums1[i]);
temp=nums1[i];
}
i++;
j++;
}
else if(nums1[i]<nums2[j])
{
i++;
}
else j++;
}
return nums3;
}
};
感想:不算太难吧,就是遍历找到相同的元素,找不到,继续向下,主要是while循环里面的代码以及temp的设定。