荷兰国旗问题(三路划分)

最新推荐文章于 2021-11-27 20:15:48 发布

sdccone1

最新推荐文章于 2021-11-27 20:15:48 发布

阅读量197

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/weixin_44718747/article/details/107907602

版权

算法专栏收录该内容

271 篇文章 0 订阅

订阅专栏

本文介绍了一种解决荷兰国旗问题的算法，即快速排序的三路划分实现。该算法将数组中的元素根据目标值分为小于、等于和大于目标值的三个部分，通过移动指针和交换元素来实现。文章详细解释了算法的步骤，包括如何使用cur、l和r三个指针，并提供了Java代码实现。

摘要由CSDN通过智能技术生成

荷兰国旗问题，也是快排三路划分的实现，将nums[]中所有比target小的数放在左变，中间是target，右边是所有比target大的数
思想：只有相等时才移动cur指针，不等时不移动cur指针只进行交换，因为交换来的数我们并不知它与target的大小关系，所以我们要对现在这个cur位置上刚被换过来的数再进行比较
时间复杂度O(n)

import java.util.*;
public class Main{
    public static void swap(int[]nums,int l,int r){
        int temp = nums[l];
        nums[l] = nums[r];
        nums[r] = temp;
    }
    public static void threeWayPartition(int[]nums,int target){
        int len = nums.length;
        if(len<=1)return;
        int l = -1,r = len,cur = 0;//[0,l]表示当前比target的区间，[r,n-1]表示所有比target大的区间
        while(cur<r){
            if(nums[cur]<target){
                swap(nums,++l,cur);
            }else if(nums[cur]>target){
                swap(nums,--r,cur);
            }else{//只有相等时才移动cur指针，不等时不移动cur指针只进行交换，因为交换来的数我们并不知它与target的大小关系，所以我们要对现在这个cur位置上刚被换过来的数再进行比较
                cur++;
            }
        }
    }
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt(),target = in.nextInt();
        int[] nums = new int[n];
        for(int i=0;i<n;i++){
            nums[i] = in.nextInt();
        }
        in.close();
        threeWayPartition(nums,target);
        for(int x:nums){
            System.out.print(x+" ");
        }
    }
}