[程序员代码面试指南]数组和矩阵-数组的partition调整

Posted 2021-12-03 coding-gaga

题目

补充问题：数组只含0，1，2，对数组排序，要求时间复杂度O(n),额外空间复杂度O(1)

题解

• 维护三个变量,l,idx,r。左区间[0,l],中间区间[l+1,idx],右区间[idx+1,r]。
• 初始化l=-1，r=len,idx=0。idx用来遍历数组。
  • 当arr[idx]=1,idx++;
  • 当arr[idx]=0,swap(arr[l+1,idx]),i++,idx++
  • 当arr[idx]=2,swap(arr[idx,r-1]),r--,idx++
• 当idx=r说明中区间和右区间连上了。排序结束。

todo

原问题待做。

代码

package ArrayAndMatrix;

public class Main {
    public static void main(String args[]) {
        int[] arr= {0,1,2,0,1,2};
        sort(arr);
        for(int elm:arr) {
            System.out.println(elm);
        }
    }
    
    public static void sort(int[] arr) {
        if(arr.length>1) {
            int l=-1;
            int r=arr.length;
            int idx=0;
            while(idx!=r) {
                if(arr[idx]==1) {
                    ++idx;
                }
                else if(arr[idx]==0) {
                    swap(arr,idx++,++l);
                }
                else {
                    swap(arr,idx,--r);
                }
            }
        }
    }
    
    public static void swap(int[] arr,int idx1,int idx2) {
        int temp=arr[idx1];
        arr[idx1]=arr[idx2];
        arr[idx2]=temp;
    }
}