[LintCode] 丢失的第一个正整数

转载于 2015-07-07 16:10:00 发布 · 58 阅读

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/jcliBlogger/p/4627311.html

本文介绍了一种高效算法，用于找出未排序整数数组中最小的缺失正整数。通过原地哈希的方式重新排列数组元素，使得每个位置i的元素等于i+1，最后遍历数组找到第一个不匹配的位置即为所求。

 1 class Solution {
 2 public:
 3     /**    
 4      * @param A: a vector of integers
 5      * @return: an integer
 6      */
 7     int firstMissingPositive(vector<int> A) {
 8         // write your code here
 9         int n = A.size();
10         for (int i = 0; i < n; i++)
11             while (A[i] > 0 && A[i] <= n && A[A[i] - 1] != A[i])
12                 swap(A[i], A[A[i] - 1]);
13         for (int i = 0; i < n; i++)
14             if (A[i] != i + 1)
15                 return i + 1;
16         return n + 1;
17     }
18 };