思路:本题要求先按顺序放入arr2对应元素的前提下,再从小到大放入arr1中出现的其他元素;我们就可以先对arr1中的元素进行排序,因为只要有从小到大,肯定离不开排序,然后再优先放入arr2元素,同时把该元素置为-1,从而保证之后放入arr1其他元素时不再出现该元素,放完arr2就可以放arr1的元素了;因为最大元素和元素个数都在1000以内,所以我用的计数排序以换取更短的时间;
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;
}