|
42 | 42 |
|
43 | 43 | <!-- 这里可写通用的实现逻辑 -->
|
44 | 44 |
|
| 45 | +由于本题数据范围是 `[0, 1000]`,因此我们可以开辟一个长度为 1001 的数组 mp,记录 arr1 的元素以及出现的次数。 |
| 46 | + |
| 47 | +接着,遍历 arr2 中每个元素 x,若 `mp[x]` 大于 0,循环将 x 加入到答案数组 arr1 中,并且递减 `mp[x]`。遍历结束后,再从下标 `j = 0` 开始遍历 mp,若遇到 `mp[j]` 大于 0,将 `mp[j]` 个 j 加入到答案数组 arr1 中。 |
| 48 | + |
| 49 | +最后返回 arr1 即可。 |
| 50 | + |
45 | 51 | <!-- tabs:start -->
|
46 | 52 |
|
47 | 53 | ### **Python3**
|
48 | 54 |
|
49 | 55 | <!-- 这里可写当前语言的特殊实现逻辑 -->
|
50 | 56 |
|
51 | 57 | ```python
|
| 58 | +class Solution: |
| 59 | + def relativeSortArray(self, arr1: List[int], arr2: List[int]) -> List[int]: |
| 60 | + mp = {num: i for i, num in enumerate(arr2)} |
| 61 | + arr1.sort(key=lambda x: (mp.get(x, 10000), x)) |
| 62 | + return arr1 |
| 63 | +``` |
52 | 64 |
|
| 65 | +```python |
| 66 | +class Solution: |
| 67 | + def relativeSortArray(self, arr1: List[int], arr2: List[int]) -> List[int]: |
| 68 | + mp = [0] * 1001 |
| 69 | + for x in arr1: |
| 70 | + mp[x] += 1 |
| 71 | + i = 0 |
| 72 | + for x in arr2: |
| 73 | + while mp[x] > 0: |
| 74 | + arr1[i] = x |
| 75 | + mp[x] -= 1 |
| 76 | + i += 1 |
| 77 | + for x, cnt in enumerate(mp): |
| 78 | + for _ in range(cnt): |
| 79 | + arr1[i] = x |
| 80 | + i += 1 |
| 81 | + return arr1 |
53 | 82 | ```
|
54 | 83 |
|
55 | 84 | ### **Java**
|
56 | 85 |
|
57 | 86 | <!-- 这里可写当前语言的特殊实现逻辑 -->
|
58 | 87 |
|
59 | 88 | ```java
|
| 89 | +class Solution { |
| 90 | + public int[] relativeSortArray(int[] arr1, int[] arr2) { |
| 91 | + int[] mp = new int[1001]; |
| 92 | + for (int x : arr1) { |
| 93 | + ++mp[x]; |
| 94 | + } |
| 95 | + int i = 0; |
| 96 | + for (int x : arr2) { |
| 97 | + while (mp[x]-- > 0) { |
| 98 | + arr1[i++] = x; |
| 99 | + } |
| 100 | + } |
| 101 | + for (int j = 0; j < mp.length; ++j) { |
| 102 | + while (mp[j]-- > 0) { |
| 103 | + arr1[i++] = j; |
| 104 | + } |
| 105 | + } |
| 106 | + return arr1; |
| 107 | + } |
| 108 | +} |
| 109 | +``` |
| 110 | + |
| 111 | +### **C++** |
| 112 | + |
| 113 | +```cpp |
| 114 | +class Solution { |
| 115 | +public: |
| 116 | + vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2) { |
| 117 | + vector<int> mp(1001); |
| 118 | + for (int x : arr1) ++mp[x]; |
| 119 | + int i = 0; |
| 120 | + for (int x : arr2) |
| 121 | + { |
| 122 | + while (mp[x]-- > 0) arr1[i++] = x; |
| 123 | + } |
| 124 | + for (int j = 0; j < mp.size(); ++j) |
| 125 | + { |
| 126 | + while (mp[j]-- > 0) arr1[i++] = j; |
| 127 | + } |
| 128 | + return arr1; |
| 129 | + } |
| 130 | +}; |
| 131 | +``` |
60 | 132 |
|
| 133 | +### **Go** |
| 134 | +
|
| 135 | +```go |
| 136 | +func relativeSortArray(arr1 []int, arr2 []int) []int { |
| 137 | + mp := make([]int, 1001) |
| 138 | + for _, x := range arr1 { |
| 139 | + mp[x]++ |
| 140 | + } |
| 141 | + i := 0 |
| 142 | + for _, x := range arr2 { |
| 143 | + for mp[x] > 0 { |
| 144 | + arr1[i] = x |
| 145 | + mp[x]-- |
| 146 | + i++ |
| 147 | + } |
| 148 | + } |
| 149 | + for j, cnt := range mp { |
| 150 | + for cnt > 0 { |
| 151 | + arr1[i] = j |
| 152 | + i++ |
| 153 | + cnt-- |
| 154 | + } |
| 155 | + } |
| 156 | + return arr1 |
| 157 | +} |
61 | 158 | ```
|
62 | 159 |
|
63 | 160 | ### **...**
|
|
0 commit comments