1122. Relative Sort Array

Posted 2020-12-11 habibah-chang

问题：

给定数组arr2，所含元素唯一，

给定数组arr1，对arr1排序，使得arr1的元素相对顺序为arr2的顺序，不存在于arr2中的元素，按照升序排列。

Example 1:
Input: arr1 = [2,3,1,3,2,4,6,7,9,2,19], arr2 = [2,1,4,3,9,6]
Output: [2,2,2,1,4,3,3,9,6,7,19]
 

Constraints:
arr1.length, arr2.length <= 1000
0 <= arr1[i], arr2[i] <= 1000
Each arr2[i] is distinct.
Each arr2[i] is in arr1.

　　

解法：

由于限制：所有元素大小：0 <= arr1[i], arr2[i] <= 1000

因此可以考虑使用桶排序。

计数arr1中每个元素出现的次数ncout[]，

首先，根据arr2的元素，遍历，输出ncout[a2]个a2，

同时更新ncout[a2]=0，

再次，遍历ncout，输出计数不为0的元素。

 

代码参考：

 1 class Solution {
 2 public:
 3     vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2) {
 4         vector<int> res;
 5         int ncout[1001]={0};
 6         for(int a1:arr1){
 7             ncout[a1]++;
 8         }
 9         for(int a2:arr2){
10             if(ncout[a2]>0){
11                 vector<int> tmp(ncout[a2], a2);
12                 res.insert(res.end(), tmp.begin(), tmp.end());
13                 ncout[a2]=0;
14             }
15         }
16         for(int i=0; i<1001; i++){
17             if(ncout[i]>0){
18                 vector<int> tmp(ncout[i], i);
19                 res.insert(res.end(), tmp.begin(), tmp.end());
20             }
21         }
22         return res;
23     }
24 };

另，不使用vector合并insert方法，使用push_back的时候：

但这时花费memory大于上面的合并vector方法。

 1 class Solution {
 2 public:
 3     vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2) {
 4         vector<int> res;
 5         int ncout[1001]={0};
 6         for(int a1:arr1){
 7             ncout[a1]++;
 8         }
 9         for(int a2:arr2){
10             while(ncout[a2]>0){
11                 res.push_back(a2);
12                 ncout[a2]--;
13             }
14         }
15         for(int i=0; i<1001; i++){
16             while(ncout[i]>0){
17                 res.push_back(i);
18                 ncout[i]--;
19             }
20         }
21         return res;
22     }
23 };