【LeetCode1122】数组的相对排序(哈希or自定义sort)

简介: 没错,,简单题,就是说将arr1中有arr2的元素,则按照arr2的元素先排列(特别注意:这里的arr2的元素都是不同的,但是arr1中是可以元素重复的)

一、题目

image.png


二、方法一:哈希

没错,,简单题,就是说将arr1中有arr2的元素,则按照arr2的元素先排列(特别注意:这里的arr2的元素都是不同的,但是arr1中是可以元素重复的),然后arr1中不是arr2的元素则被丢到arr1的尾巴。


可以用哈希,用mp1记录arr1数组每种数字的出现次数,mp2记录arr2数组的是否出现,接着就是上面说的逻辑,,就是脑子清醒点,mp1[arr2[i]]这种东西不要搞乱了。


三、代码

class Solution {
public:
    vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2) {
        //mp1记录arr1数组每种数字的出现次数
        map<int, int>mp1;
        //mp2记录arr2数组的是否出现
        map<int, bool>mp2;
        for(int i = 0; i < arr2.size(); i++){
            mp2[arr2[i]] = true;
        }
        //temp装arr1中没有arr2的元素
        vector<int>temp;
        for(int j = 0; j < arr1.size(); j++){
            //arr1中是arr2的数字
            mp1[arr1[j]] += 1;
            if(mp2[arr1[j]] == false){
                temp.push_back(arr1[j]);
            }
        }
        sort(temp.begin(), temp.end());
        vector<int>ans;
        //遍历arr2
        for(int i = 0; i < arr2.size(); i++){
            for(int j = 0; j < mp1[arr2[i]]; j++){
                ans.push_back(arr2[i]);
            }
        }
        for(int i = 0; i < temp.size(); i++){
            ans.push_back(temp[i]);
        }
        return ans;
    }
};

四、法二:自定义排序

记录一种更方便的方法:排序并自定义比较函数。


由于数组 a r r 2 a r r_{2}arr

2

 规定了比较顺序,因此我们可以使用哈希表对该顺序进行映射:即对于数组 a r r 2 a r r_{2}arr

2

 中的第 i ii 个 元素,我们将 ( arr ⁡ 2 [ i ] , i ) \left(\operatorname{arr}_{2}[i], i\right)(arr

2

[i],i) 这一键值对放入哈希表 rank ⁡ \operatorname{rank}rank 中,就可以很方便地对数组 arr ⁡ 1 \operatorname{arr}_{1}arr

1

 中的元素进行比较。


比较函数的写法有很多种,这里最基础的比较方法,对于元素 x xx 和 y yy :


如果 x xx 和 y yy 都出现在哈希表中,那么比较它们对应的值 rank ⁡ [ x ] \operatorname{rank}[x]rank[x] 和 rank ⁡ [ y ] \operatorname{rank}[y]rank[y];

如果 x xx 和 y yy 都没有出现在哈希表中,那么比较它们本身;

对于剩余的情况,出现在哈希表中的那个元素较小。

class Solution {
public:
    vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2) {
        unordered_map<int, int> rank;
        for (int i = 0; i < arr2.size(); ++i) {
            rank[arr2[i]] = i;
        }
        sort(arr1.begin(), arr1.end(), [&](int x, int y) {
            if (rank.count(x)) {
                return rank.count(y) ? rank[x] < rank[y] : true;
            }
            else {
                return rank.count(y) ? false : x < y;
            }
        });
        return arr1;
    }
};
相关文章
|
1月前
|
算法
Leetcode 初级算法 --- 数组篇
Leetcode 初级算法 --- 数组篇
38 0
|
1月前
【LeetCode-每日一题】 删除排序数组中的重复项
【LeetCode-每日一题】 删除排序数组中的重复项
19 4
|
1月前
|
索引
Leetcode第三十三题(搜索旋转排序数组)
这篇文章介绍了解决LeetCode第33题“搜索旋转排序数组”的方法,该问题要求在旋转过的升序数组中找到给定目标值的索引,如果存在则返回索引,否则返回-1,文章提供了一个时间复杂度为O(logn)的二分搜索算法实现。
18 0
Leetcode第三十三题(搜索旋转排序数组)
|
1月前
|
算法 C++
Leetcode第53题(最大子数组和)
这篇文章介绍了LeetCode第53题“最大子数组和”的动态规划解法,提供了详细的状态转移方程和C++代码实现,并讨论了其他算法如贪心、分治、改进动态规划和分块累计法。
56 0
|
1月前
|
C++
【LeetCode 12】349.两个数组的交集
【LeetCode 12】349.两个数组的交集
16 0
|
3月前
|
存储 算法
LeetCode第83题删除排序链表中的重复元素
文章介绍了LeetCode第83题"删除排序链表中的重复元素"的解法,使用双指针技术在原链表上原地删除重复元素,提供了一种时间和空间效率都较高的解决方案。
LeetCode第83题删除排序链表中的重复元素
|
3月前
|
算法
LeetCode第81题搜索旋转排序数组 II
文章讲解了LeetCode第81题"搜索旋转排序数组 II"的解法,通过二分查找算法并加入去重逻辑来解决在旋转且含有重复元素的数组中搜索特定值的问题。
LeetCode第81题搜索旋转排序数组 II
|
2月前
|
Unix Shell Linux
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
本文提供了几个Linux shell脚本编程问题的解决方案,包括转置文件内容、统计词频、验证有效电话号码和提取文件的第十行,每个问题都给出了至少一种实现方法。
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
|
3月前
|
Python
【Leetcode刷题Python】剑指 Offer 32 - III. 从上到下打印二叉树 III
本文介绍了两种Python实现方法,用于按照之字形顺序打印二叉树的层次遍历结果,实现了在奇数层正序、偶数层反序打印节点的功能。
55 6
|
3月前
|
搜索推荐 索引 Python
【Leetcode刷题Python】牛客. 数组中未出现的最小正整数
本文介绍了牛客网题目"数组中未出现的最小正整数"的解法,提供了一种满足O(n)时间复杂度和O(1)空间复杂度要求的原地排序算法,并给出了Python实现代码。
113 2