题目描述
自己解法
空间换时间
class Solution:
def relativeSortArray(self, arr1: List[int], arr2: List[int]) -> List[int]:
except_arr = []
in_dict = {}
ans_one,ans_two = [],[]
for val in arr1:
if val in in_dict.keys():
in_dict[val] += 1
else:
in_dict[val] = 1
for val in arr2:
ans_one += [val] * in_dict[val]
in_dict.pop(val)
for key in in_dict.keys():
ans_two += [key] * in_dict[key]
return ans_one + sorted(ans_two)
题解区
桶排序
class Solution { //桶排序
public:
vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2)
{
int a[1001] = { 0 }; //桶
for (int j = 0; j < arr1.size(); j++)
{
if (j < arr2.size())
a[arr2[j]] += 1001; //以1001为分解线,
a[arr1[j]] += 1;
}
//最后a[arr2[j]]>=1001+1 (1)
int tmp = 0;
for (int i = 0; i < arr2.size(); i++)
{
while (a[arr2[i]]-- > 1001 ) //由(1)a[arr2[i]]>=1001 (2)
arr1[tmp++] = arr2[i];
}
for (int i = 0; i < 1001; i++)
while (a[i] > 0 && a[i]-- < 1000)//由(2)a[arr2[i]]>=1000
arr1[tmp++] = i;
return arr1;
}
};
参考:https://leetcode-cn.com/problems/relative-sort-array/solution/tong-pai-xu-ke-0ms-by-lin-xi-zhu-w/