leetcode1122.数组的相对排序

这道题思路比较简单,先对于每个数组2中的元素遍历数组1,如果遇到相同的,就直接放入result数组。再对于每个数组1中的元素遍历数组2,如果在数组2中没有找到,就存入一个暂时数组temp,再对temp数组进行一下排序,把排序后的temp数组中的数加入result数组。代码如下:

#include
#include
#include
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;
}

代码评价:空间复杂度还好,时间复杂度比较高,因为存在两个双重循环遍历两个数组。

你可能感兴趣的:(leetcode,leetcode,C++,数组排序)