LintCode_189 First Missing Positive

原创于 2016-07-30 04:37:33 发布 · 249 阅读

0 ·

CC 4.0 BY-SA版权

LintCode 专栏收录该内容

58 篇文章

订阅专栏

本文介绍了一种在未排序整数数组中寻找第一个缺失正整数的算法，并提出了解决方案，该算法运行时间为O(n)，且使用常数空间。通过调整数组元素的位置来标记每个位置对应的正整数是否存在，最终找到第一个缺失的正整数。

Given an unsorted integer array, find the first missing positive integer.

Example

Given [1,2,0] return 3,
and [3,4,-1,1] return 2.

Challenge

Your algorithm should run in O(n) time and uses constant space.

转化成一组数中只有一个出现一次的问题。

class Solution {
public:
    /**    
     * @param A: a vector of integers
     * @return: an integer
     */
    int firstMissingPositive(vector<int> A) {
        // write your code here
        for (int i = 0; i < A.size(); i++) {
            if (A[i] <= A.size() && A[i] > 0 && A[i] != i + 1 && A[A[i] - 1] != A[i]) {
                swap(A[A[i] - 1], A[i]);
                i--;
            }
        }
        for (int i = 0; i < A.size(); i++) {
            if (A[i] != i + 1)
                return i + 1;
        }
        return A.size() + 1;
    }
};