leetcode: First Missing Positive

来源:互联网 发布:中国话剧 知乎 编辑:程序博客网 时间:2024/06/14 18:31

思想:要求找出第一个missing的正整数,想法当然是排序,但是题目又要求算法复杂度不超过O(n)

于是想到桶排序,但是又要求空间复杂度为O(1),于是要优化桶排序:

遍历数组,若当前元素不应该装载当前桶内:A[i]!=i+1,那么我就就把A[i]放到它应该在的第A[i]-1号桶里,

例如A[0]=5,就应该把置换A[0]和A[4],如此不断置换,直到A[0]=1或者A[0]<=0或者A[0]>=n,接着算A[1];

这样遍历一遍之后,该在的桶里都放好了对应的元素,就找第一个不匹配的桶,就是第一个missing的正整数。

要考虑重复数字死循环的情况,例如A=[1,1]

public class Solution {

    public int firstMissingPositive(int[] A) {
        int i,j;
        int n=A.length;
        int e1,e2,tmp;
        
        for(i=0;i<n;i++)
        {
            while(i<n && (A[i]<=0 || A[i]==i+1))
                i++;
            e1=i;
            if(e1>=n)
            break;
            while(A[e1]!=e1+1)
            {
            if(e1>=n || A[e1]<=0 || A[e1]==e1+1)
                    break;
                e2=A[e1]-1;
                if(e2>=n || e2<0)
                    break;
                if(A[e1]==A[e2])
                break;
                tmp=A[e1];
                A[e1]=A[e2];
                A[e2]=tmp;
            }
        }
        for(i=0;i<n;i++)
        {
            if(A[i]!=i+1)
                break;
        }
        return i+1;
    }
}
0 0
原创粉丝点击