【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;
    }
};
相关文章
|
2天前
|
算法
leetcode代码记录(寻找两个正序数组的中位数
leetcode代码记录(寻找两个正序数组的中位数
9 2
|
2天前
|
索引
leetcode代码记录(最长重复子数组
leetcode代码记录(最长重复子数组
7 0
|
2天前
leetcode代码记录(两个数组的交集
leetcode代码记录(两个数组的交集
8 1
|
2天前
leetcode代码记录(最大子数组和
leetcode代码记录(最大子数组和
8 2
|
5天前
|
存储 算法
Leetcode 30天高效刷数据结构和算法 Day1 两数之和 —— 无序数组
给定一个无序整数数组和目标值,找出数组中和为目标值的两个数的下标。要求不重复且可按任意顺序返回。示例:输入nums = [2,7,11,15], target = 9,输出[0,1]。暴力解法时间复杂度O(n²),优化解法利用哈希表实现,时间复杂度O(n)。
16 0
|
11天前
|
索引
Leetcode 给定一个数组,给定一个数字。返回数组中可以相加得到指定数字的两个索引
Leetcode 给定一个数组,给定一个数字。返回数组中可以相加得到指定数字的两个索引
|
17天前
[leetcode] 705. 设计哈希集合
[leetcode] 705. 设计哈希集合
|
26天前
【力扣】238. 除自身以外数组的乘积
【力扣】238. 除自身以外数组的乘积
|
2天前
|
算法 C++
【刷题】Leetcode 1609.奇偶树
这道题是我目前做过最难的题,虽然没有一遍做出来,但是参考大佬的代码,慢慢啃的感觉的真的很好。刷题继续!!!!!!
6 0
|
2天前
|
算法 索引
【刷题】滑动窗口精通 — Leetcode 30. 串联所有单词的子串 | Leetcode 76. 最小覆盖子串
经过这两道题目的书写,相信大家一定深刻认识到了滑动窗口的使用方法!!! 下面请大家继续刷题吧!!!
8 0

热门文章

最新文章