Partition的四种写法

最新推荐文章于 2024-07-02 17:34:11 发布

原创

最新推荐文章于 2024-07-02 17:34:11 发布 · 1.4k 阅读

收录于

本文介绍了Partition算法的四种不同写法，它是快速排序的核心，也适用于TopK问题。第一种是通过标志位找到小于pivot的元素，第二种是双向扫描交换元素，第三种是优化的交换方式减少赋值次数，第四种是三向切分法。这些方法在处理无序数组时将元素分为小于和大于pivot的两部分，并返回pivot的新位置。

partition算法是快排的核心，还可用于TopK问题，它的思想是这样：从无序数组中随机选一个 pivot，然后通过一趟扫描，以 pivot 为分界线将数组中其他元素分为两部分，使得左边部分的数小于等于 pivot，右边部分的数大于等于 pivot（左部分或者右部分都可能为空），最后返回 pivot 在新的数组中的位置。

第一种写法:

public static int partition1(int[] a, int begin, int end) {
   
   
	//选a[begin]作为pivot
    int pivot = a[begin];
    int pos = begin;
    for (int i = begin + 1; i <= end; ++i) {
   
   
        if (a[i] < pivot) {
   
   
            pos++;
            if (i != pos) //避免自身交换
                swap(a, pos, i);
        }
    }
    //最后交换pos和begin
    swap(a, pos, begin);
    return pos;
}