LeetCode 2122. 还原原数组

最新推荐文章于 2024-04-18 16:05:01 发布

英雄哪里出来

最新推荐文章于 2024-04-18 16:05:01 发布

阅读量468

点赞数 2

分类专栏：《LeetCode算法全集》文章标签：算法数据结构贪心

本文链接：https://blog.csdn.net/WhereIsHeroFrom/article/details/123785756

版权

《LeetCode算法全集》专栏收录该内容

265 篇文章 1916 订阅

订阅专栏

文章目录

一、题目
二、解题报告
三、本题小知识
四、加群须知

一、题目

1、题目描述

Alice有一个下标从 $0$ 开始的数组 arr，由 $n$ 个正整数组成。她会选择一个任意的正整数 $k$ 并按下述方式创建两个下标从 $0$ 开始的新整数数组 lower和 higher：
对每个满足 0 <= i < n的下标 $i$ ，lower[i] = arr[i] - k
对每个满足 0 <= i < n的下标 $i$ ，higher[i] = arr[i] + k
不幸地是，Alice丢失了全部三个数组。但是，她记住了在数组 lower和 higher中出现的整数，但不知道每个整数属于哪个数组。请你帮助 Alice还原原数组。
给你一个由 $2 n$ 个整数组成的整数数组 nums，其中恰好 $n$ 个整数出现在 lower，剩下的出现在 higher，还原并返回原数组 arr。如果出现答案不唯一的情况，返回任一有效数组。
样例输入： [2,10,6,4,8,12]
样例输出： [3,7,11]

2、基础框架

C++ 版本给出的基础框架代码如下：

class Solution {
public:
    vector<int> recoverArray(vector<int>& nums) {
        
    }
};

3、原题链接

LeetCode 2122. 还原原数组

二、解题报告

1、思路分析

$(1)$ 遇到这么一个题，首先看下数据范围， $\le 1000$ ，所以我们可以试着考虑 $O(n^2)$ 的算法。
$(2)$ 在没有任何思路的情况下，我们想一想能不能枚举 $k$ ，然后来求解。枚举 $k$ 的方法有：顺序枚举、二分枚举、通过给定信息来枚举。
$(3)$ 这题没有单调性，所以不能用二分枚举； $k$ 有可能很大，所以必然不能用顺序枚举。所以只能根据题目信息再来看 $k$ 了。
$(4)$ 对每个满足 0 <= i < n的下标 $i$ ，lower[i] = arr[i] - k；对每个满足 0 <= i < n的下标 $i$ ，higher[i] = arr[i] + k。
$(5)$ 基于第 $(4)$ 点，如果将 higher[j] - lower[i]得到的是 2k。也就是说给定的数组nums中的任意两个数相减，是 $k$ 的一个可行解的两倍；所以要求 $k$ ，只要枚举任何两个数的差就行了。
$(6)$ 由于每个数都要被放到lower或者higher，所以最小的那个数，必然是 lower，那么，只要对数组进行排序，然后枚举所有的数减去最小的那个数，得到的就是其中一个 $2 k$ 。
$(7)$ 然后，贪心的去数组里面找所有 lower和higher数组对就行了，找的过程可以用哈希表查找。
$(8)$ 所以，这道题的思想是贪心，算法是排序 + 线性枚举 + 哈希表。

2、时间复杂度

最坏时间复杂度 $O(n^2)$ 。

3、代码详解

class Solution {
    unordered_map <int, int> hash;

public:
    vector<int> recoverArray(vector<int>& nums) {
        sort(nums.begin(), nums.end());

        int i, j, n = nums.size();
        vector <int> ret;

        for(i = 1; i < n; ++i) {
            int k2 = nums[i] - nums[0];
            if(k2 & 1) {
                continue;
            }
            if(k2 == 0) {
                continue;
            }
            int k = k2 / 2;
            hash.clear();
            ret.clear();
            ret.push_back( nums[0] + k );

            for(j = 1; j < n; ++j) {
                if(j == i) {
                    continue;
                }
                ++hash[ nums[j] ];
            }

            for(j = 1; j < n; ++j) {
                if(j == i) {
                    continue;
                }
                if(!hash[ nums[j] ]) {
                    continue;
                }
                
                if(hash[ nums[j] + k2 ] ) {    
                    --hash[ nums[j] ];             // lowwer
                    --hash[ nums[j] + k2 ];        // higher
                    ret.push_back( nums[j] + k );
                }else {
                    break;
                }
            }
            if(j == n) {
                return ret;
            }

        }
        return {};
    }
};

三、本题小知识

这题其实可以用双指针，只是哈希表能够水过去。

四、加群须知

相信看我文章的大多数都是「大学生」，能上大学的都是「精英」，那么我们自然要「精益求精」，如果你还是「大一」，那么太好了，你拥有大把时间，当然你可以选择「刷剧」，然而，「学好算法」，三年后的你自然「不能同日而语」。
那么这里，我整理了「几十个基础算法」 的分类，点击开启：

🌌《算法入门指引》🌌
如果链接被屏蔽，或者有权限问题，可以私聊作者解决。

大致题集一览：
在这里插入图片描述

在这里插入图片描述

在这里插入图片描述

为了让这件事情变得有趣，以及「照顾初学者」，目前题目只开放最简单的算法 「枚举系列」 （包括：线性枚举、双指针、前缀和、二分枚举、三分枚举），当有一半成员刷完 「枚举系列」 的所有题以后，会开放下个章节，等这套题全部刷完，你还在群里，那么你就会成为「夜深人静写算法」专家团 的一员。
不要小看这个专家团，三年之后，你将会是别人 望尘莫及 的存在。如果要加入，可以联系我，考虑到大家都是学生， 没有「主要经济来源」，在你成为神的路上，「不会索取任何」。
🔥联系作者，或者扫作者主页二维码加群，加入刷题行列吧🔥

🔥让天下没有难学的算法🔥
C语言免费动漫教程，和我一起打卡！ 🌞《光天化日学C语言》🌞
让你养成九天持续刷题的习惯 🔥《九日集训》🔥
入门级C语言真题汇总 🧡《C语言入门100例》🧡
组团学习，抱团生长 🌌《算法零基础100讲》🌌
几张动图学会一种数据结构 🌳《画解数据结构》🌳
竞赛选手金典图文教程 💜《夜深人静写算法》💜