【leetcode】1122. 数组的相对排序（relative-sort-array）（模拟）[简单]

最新推荐文章于 2022-03-01 20:37:09 发布

KroneX

最新推荐文章于 2022-03-01 20:37:09 发布

阅读量122

点赞数

分类专栏： leetcode 题解模拟文章标签：哈希 leetcode

本文链接：https://blog.csdn.net/Krone_/article/details/109695198

版权

题解同时被 3 个专栏收录

217 篇文章 0 订阅

订阅专栏

leetcode

215 篇文章 0 订阅

订阅专栏

模拟

66 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何利用计数排序的思想解决LeetCode上的一个问题——相对排序数组。作者首先阐述了问题背景，然后详细解释了通过建立哈希表进行离散化和计数，以及遍历排序过程来实现解决方案的步骤。在时间复杂度分析中，指出最坏情况下的复杂度为O(max(nlogn,m))，其中n和m分别是两个输入数组的长度。最后，给出了AC代码实现。

摘要由CSDN通过智能技术生成

链接

https://leetcode-cn.com/problems/relative-sort-array/

耗时

解题：12 min
题解：12 min

题意

给你两个数组，arr1 和 arr2，

arr2 中的元素各不相同
arr2 中的每个元素都出现在 arr1 中

对 arr1 中的元素进行排序，使 arr1 中项的相对顺序和 arr2 中的相对顺序相同。未在 arr2 中出现过的元素需要按照升序放在 arr1 的末尾。

思路

计数排序的思想。

用哈希表做离散化，计数 arr1 中每个元素出现的数量。然后遍历 arr2，对于 arr2 中的每个元素，按照哈希表记录的 arr1 中这个元素的数量，把这么多个这个元素放入结果中，并用哈希表记录 arr2 中出现过的元素。最后遍历记录 arr1 元素数量的哈希表，如果哈希表中遍历到的元素没在 arr2 中出现过，则按照这个元素的数量把这个元素放入一个临时数组，待遍历结束后，对临时数组进行排序，并将临时数组的元素放在结果的末尾。

时间复杂度：最坏 $O (m a x (n l o g n, m))$ n 是 arr1 的长度，m 是 arr2 的长度

AC代码

class Solution {
public:
    vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2) {
        unordered_map<int, int> cnt;
        for(auto x : arr1) {
            cnt[x]++;
        }
        vector<int> ans;
        unordered_set<int> flag;
        for(auto x : arr2) {
            for(int i = 0; i < cnt[x]; ++i) {
                ans.push_back(x);
            }
            flag.insert(x);
        }
        vector<int> res;
        for(auto x : cnt) {
            if(flag.count(x.first)) continue;
            for(int i = 0; i < x.second; ++i) {
                res.push_back(x.first);
            }
        }
        sort(res.begin(), res.end());
        ans.insert(ans.end(), res.begin(), res.end());
        return ans;
    }
};