两种思路都利用了输入的数组A,若A中存在i,则给A[i]作为标记。
因为A中的n个元素存在>n和<=0的,所以第一个缺失的正整数一定在[1-n+1]之间。
第一种思路是将标记设为一个特定的数。因为改变数值会影响该位置原来存的值,所以需要在一个循环里依次处理所有“原来的值”。
例如数组为{2,3,4,1}。对第一个数2,我们将位置(2-1)=1标记为-MAX_INT,数组变为{2,-MAX_INT,4,1},丢失了3,所以应记录下数组原来的值,并继续将位置(3-1)=2标记为-MAX_INT。。。
int firstMissingPositive(int A[], int n) { for (int i = 0; i < n; i++) { int k = A[i]; while (k > 0 && k <= n) { int temp = A[k-1]; A[k-1] = - INT_MAX; k = temp; } } for (int i = 0; i < n; i++) if (A[i] != -INT_MAX) return i + 1; return n + 1; }
第二种思路,我们标记时可以不用改变数值,所以可以取消掉for里面的while循环。
如果不改变数值,我们能改变什么那?我们可以改变符号。这种思路里,数组中一开始存在的负数会对最终结果产生影响,所以我们做一个预处理----将所有负数变成一个大于n的数----INT_MAX。
int firstMissingPositive(int A[], int n) { for (int i = 0; i < n; i++) { if (A[i] <= 0) A[i] = INT_MAX; } for (int i = 0; i < n; i++) { int k = abs(A[i]); if (k <= n) A[k-1] = - abs(A[k-1]); } for (int i = 0; i < n; i++) if (A[i] > 0) return i + 1; return n + 1; }
两种方法的时间复杂度都是O(n),准确的说,是最坏的情况下都会遍历数组3遍。
时间: 2024-12-20 18:12:41