1122 数组的相对排序

给你两个数组,arr1 和 arr2arr2 中的元素各不相同,arr2 中的每个元素都出现在 arr1 中。

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

class Solution {
public:
    std::vector<int> relativeSortArray(std::vector<int>& arr1, std::vector<int>& arr2) {
        vector<int> ans;
        unordered_map<int, int> mp;

        // 统计 arr1 中每个数字出现的次数
        for (auto &num : arr1) {
            mp[num]++;
        }

        // 按照 arr2 的顺序将元素加入结果集
        for (auto &num : arr2) {
            //insert 在  pos插入    n 个   elem
            ans.insert(ans.end(), mp[num], num);
            mp[num] = 0; // 清零,表示这些元素已经处理过了
        }

        // 收集不在 arr2 中的元素
        vector<int> remaining;
        for (auto &num : arr1) {
            if (mp[num] > 0) {
                remaining.push_back(num);
                mp[num]--;
            }
        }

        // 对剩余元素进行排序
        sort(remaining.begin(), remaining.end());

        // 将排序后的剩余元素追加到结果集中
        ans.insert(ans.end(), remaining.begin(), remaining.end());

        return ans;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值