这道题思路比较简单,先对于每个数组2中的元素遍历数组1,如果遇到相同的,就直接放入result数组。再对于每个数组1中的元素遍历数组2,如果在数组2中没有找到,就存入一个暂时数组temp,再对temp数组进行一下排序,把排序后的temp数组中的数加入result数组。代码如下:
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
public:
vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2)
{
vector<int> result;
vector<int> temp;
int i, j;
for ( i = 0; i < arr2.size(); i++)
{
for ( j = 0; j < arr1.size(); j++)
{
if (arr1[j] == arr2[i])
result.push_back(arr1[j]);
}
}
for ( i = 0; i < arr1.size(); i++)
{
for ( j = 0; j < arr2.size(); j++)
{
if (arr1[i] == arr2[j])
break;
}
if (j==arr2.size())
temp.push_back(arr1[i]);
}
sort(temp.begin(),temp.end(),less<int>());
i = 0;
while (i < temp.size())
{
result.push_back(temp[i]);
i++;
}
return result;
}
};
int main()
{
Solution s;
int m;
vector<int> arr1, arr2, result;
while (cin >> m)
{
arr1.push_back(m);
if (cin.get() == '\n')
break;
}
while (cin >> m)
{
arr2.push_back(m);
if (cin.get() == '\n')
break;
}
result = s.relativeSortArray(arr1, arr2);
for (int i = 0; i < result.size(); i++)
{
cout << result[i] << " ";
}
return 0;
}
代码评价:空间复杂度还好,时间复杂度比较高,因为存在两个双重循环遍历两个数组。