要求用O(n)时间,常数空间解题。所以不能快排(nlogn)。
注意到,对于A[i]>0 && A[i]<=n的那些数,在数组中若都最终满足A[i] == i+1, 那么只需最后遍历数组,找到第一个不满足A[i] == i+1的数,返回i+1即可。
因此遍历数组,在每次下标为i的迭代中,若A[i]>0 && A[i]<=n && A[i]!=i+1 && A[i]!=A[A[i]-1], 就不断地swap(A[i], A[A[i]-1]);
约束条件A[i] != A[A[i]-1]是为了处理输入为[1, 1]的情况,若没有这个条件,则在迭代到i=1时会陷入死循环。
代码:
class Solution
{
public:
int firstMissingPositive(int A[], int n)
{
for (int i = 0; i < n; ++ i)
{
while (A[i]>0 && A[i]<=n && A[i]-1!=i && A[A[i]-1]!=A[i])
{
swap(A[i], A[A[i]-1]);
}
}
for (int i = 0; i < n; ++ i)
{
if (A[i]-1 != i )
{
return i + 1;
}
}
return n+1;
}
};