Leetcode每日一题:1122.relative-sort-array(数组的相对排序)_计数排序
思路:本题要求先按顺序放入arr2对应元素的前提下,再从小到大放入arr1中出现的其他元素;我们就可以先对arr1中的元素进行排序,因为只要有从小到大,肯定离不开排序,然后再优先放入arr2元素,同时把该元素置为-1,从而保证之后放入arr1其他元素时不再出现该元素,放完arr2就可以放arr1的元素了;因为最大元素和元素个数都在1000以内,所以我用的计数排序以换取更短的时间;
Leetcode每日一题:1122.relative-sort-array(数组的相对排序)_i++_02

vector<int> relativeSortArray(vector<int> &arr1, vector<int> &arr2)
{
	int len1 = arr1.size();
	int len2 = arr2.size();

	//找出arr1数组中的最大值,确定用于计数的数组大小
	int arr1_max = 0;
	for (int i = 0; i < len1; i++)
	{
		if (arr1_max < arr1[i])
			arr1_max = arr1[i];
	}

	vector<int> vec(arr1_max + 1, 0);
	//统计arr1中各元素出现的个数
	for (int i = 0; i < len1; i++)
	{
		vec[arr1[i]]++;
	}
	//按顺序先把arr2的元素放进去
	int index = 0;
	for (int i = 0; i < len2; i++)
	{
		while (vec[arr2[i]]--)
		{
			arr1[index++] = arr2[i];
		}
	}
	//再把vec中计数>0的未出现在arr2的元素放入
	for (int i = 0; i <= arr1_max; i++)
	{
		while (vec[i] > 0)
		{
			arr1[index++] = i;
			vec[i]--;
		}
	}
	return arr1;
}