问题描述:计算两个数组的交
样例
nums1 = [1, 2, 2, 1]
, nums2 = [2, 2]
, 返回 [2,2]
.
实验代码:
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
sort(nums1.begin(), nums1.end());
sort(nums2.begin(), nums2.end());
int i=0, j=0;
int num1 = nums1.size();
int num2 = nums2.size();
vector<int> v;
while(i<num1 && j<num2)
{if(nums1[i] == nums2[j])
{
v.push_back(nums1[i]);
i++;
j++;
}
else if(nums1[i] > nums2[j])
{ j++;}
else
{ i++;}
}
return v;
}
};
个人感想:else if的用法。