力扣 Hot 100 刷题记录 - 缺失的第一个正数

力扣 Hot 100 刷题记录 - 缺失的第一个正数

问题描述

题目链接：缺失的第一个正数

题目难度：困难

题目描述：给定一个未排序的整数数组 nums，请你找出其中没有出现的最小正整数。要求算法的时间复杂度为 O(n)，并且只能使用常数级别的额外空间。

示例 1：

输入: nums = [1, 2, 0]
输出: 3
解释: 数组中没有出现的最小正整数是 3。

示例 2：

输入: nums = [3, 4, -1, 1]
输出: 2
解释: 数组中没有出现的最小正整数是 2。

示例 3：

输入: nums = [7, 8, 9, 11, 12]
输出: 1
解释: 数组中没有出现的最小正整数是 1。

解题思路

原地哈希（In-Place Hashing）

为了满足题目要求的 O(1) 空间复杂度，可以利用数组本身作为哈希表。具体方法是将数组中的每个正整数 x 放到索引 x-1 的位置上。例如，数字 1 应该放在索引 0 的位置，数字 2 应该放在索引 1 的位置，依此类推。

核心思想

1. 原地哈希：将每个正整数 x 放到索引 x-1 的位置上。
2. 检查缺失的正整数：遍历数组，找到第一个索引 i，使得 nums[i] != i + 1。这个索引 i + 1 就是缺失的第一个正整数。

实现步骤

1. 原地哈希：
   • 遍历数组，对于每个元素 nums[i]，如果它是一个正整数且不在正确的位置上（即 nums[i] != i + 1），则将其交换到正确的位置。
   • 重复上述步骤，直到所有正整数都放在正确的位置上，或者无法再进行交换。

for (int i = 0; i < n; ++i) {
    while (nums[i] > 0 && nums[i] <= n && nums[nums[i] - 1] != nums[i]) {
        swap(nums[i], nums[nums[i] - 1]);
    }
}

2. 检查缺失的正整数：
   在完成原地哈希后，数组中的每个正整数 x 应该被放置在索引 x-1 的位置上。接下来，我们需要遍历数组，找到第一个索引 i，使得 nums[i] != i + 1。这个索引 i + 1 就是缺失的第一个正整数。

实现步骤

1. 遍历数组：

   • 从索引 0 开始，依次检查每个位置上的值是否与其索引匹配。
   • 如果 nums[i] != i + 1，则说明 i + 1 是缺失的第一个正整数。

2. 返回结果：

   • 如果找到了不匹配的索引 i，返回 i + 1。
   • 如果所有位置都匹配，说明数组中包含了从 1 到 n 的所有正整数，因此返回 n + 1。

for (int i = 0; i < n; ++i) {
    if (nums[i] != i + 1) {
        return i + 1;  // 找到缺失的第一个正整数
    }
}
return n + 1;  // 如果所有位置都匹配，返回 n + 1

代码

#include <vector>
using namespace std;

int firstMissingPositive(vector<int>& nums) {
    int n = nums.size();

    // 原地哈希
    for (int i = 0; i < n; ++i) {
        while (nums[i] > 0 && nums[i] <= n && nums[nums[i] - 1] != nums[i]) {
            swap(nums[i], nums[nums[i] - 1]);
        }
    }

    // 检查缺失的正整数
    for (int i = 0; i < n; ++i) {
        if (nums[i] != i + 1) {
            return i + 1;  // 找到缺失的第一个正整数
        }
    }

    return n + 1;  // 如果所有位置都匹配，返回 n + 1
}