我是靠谱客的博主 过时冰棍,最近开发中收集的这篇文章主要介绍【简单LeetCode】数组的相对排序,觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

给你两个数组,arr1 和 arr2,

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

示例:

输入:arr1 = [2,3,1,3,2,4,6,7,9,2,19], arr2 = [2,1,4,3,9,6]
输出:[2,2,2,1,4,3,3,9,6,7,19]

提示:

arr1.length, arr2.length <= 1000
0 <= arr1[i], arr2[i] <= 1000
arr2 中的元素 arr2[i] 各不相同
arr2 中的每个元素 arr2[i] 都出现在 arr1 中

1、C++

先遍历arr1中的元素个数,然后遍历arr2按照arr1中对应元素的个数添加进结果数组中,最后将剩余的arr1中的元素按照原顺序的元素个数循环添加。

class Solution {
public:
    vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2) {
        map<int,int> m;
        int s1 = arr1.size(), s2 = arr2.size();
        vector<int> res;
        for(int i=0;i<s1;i++){
            auto iter = m.find(arr1[i]);
            if(iter!=m.end()){
                int temp = m[arr1[i]]+1;
                m.erase(arr1[i]);
                m.insert(pair<int,int>(arr1[i], temp));
            }else{
                m.insert(pair<int,int>(arr1[i], 1));
            }
        }
        for(int i=0;i<s2;i++){
            auto iter = m.find(arr2[i]);
            if(iter!=m.end()){
                for(int j=0;j<iter->second;j++){
                    res.push_back(iter->first);
                }
                m.erase(iter);
            }
        }
        for(auto i=m.begin();i!=m.end();i++){
            for(int k=0;k<i->second;k++){
                res.push_back(i->first);
            }
        }
        return res;
    }
};

2、桶排序

先统计arr1中元素的数量,注意索引是arr1中的值,然后把arr2中的数按顺序拿完,在把桶中剩下的按顺序拿完。

public int[] relativeSortArray(int[] arr1, int[] arr2) {
        int[] nums = new int[1001];
        int[] res = new int[arr1.length];
        //遍历arr1,统计每个元素的数量
        for (int i : arr1) {
            nums[i]++;
        }
        //遍历arr2,处理arr2中出现的元素
        int index = 0;
        for (int i : arr2) {
            while (nums[i]>0){
                res[index++] = i;
                nums[i]--;
            }
        }
        //遍历nums,处理剩下arr2中未出现的元素
        for (int i = 0; i < nums.length; i++) {
            while (nums[i]>0){
                res[index++] = i;
                nums[i]--;
            }
        }
        return res;
 }

3、python

直接将arr2加上arr1减去arr2中的元素,并按照从小到大排序。然后按照arr2中的下标的顺序对arr1进行排序,其余的按照原来的顺序。

class Solution:
    def relativeSortArray(self, arr1: List[int], arr2: List[int]) -> List[int]:
        arr2 += sorted(set(arr1)-set(arr2))
        arr1.sort(key=arr2.index)
        return arr1

来源:力扣(LeetCode)

最后

以上就是过时冰棍为你收集整理的【简单LeetCode】数组的相对排序的全部内容,希望文章能够帮你解决【简单LeetCode】数组的相对排序所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(57)

评论列表共有 0 条评论

立即
投稿
返回
顶部