算法之荷兰国旗问题

题目一:
给定一个数组arr,和一个数num,请把小于等于num的数放在数组的左边,大于num的数放在数组的右边。
要求额外空间复杂度O(1),时间复杂度O(n)。
分析:遍历数组,只要比num小的,往左排,比num大的,跳过本次操作,这样看来,我们只需要交换满足条件的元素便可以实现题目的要求了。首先我们先建立一个在数组边界外的指针。数组从L至R进行遍历,那么我们的指针当前的位置为index = L-1,在遍历过程中和给定条件进行比较,如满足条件,则交换位置,且index向右移动指针。

代码

public class SortAssignArray {

    public static void sort(int[]arr,int num) {
        if(arr == null || arr.length<2) {
            return;
        }
        int index = -1;  //遍历数组从0开始,那么当前指针的左边界就是-1;
        for(int i=0;i<arr.length;i++) {
            if(arr[i]<=num) {
                swap(arr, ++index, i);  //满足条件交换位置,且index向右移动
            }else {
                continue;
            }
        }
    }
    //交换函数
    public static void swap(int[] arr,int i,int j) {
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }
    public static void main(String[]args) {
        int []arr  = {2,1,8,5,3,9,7,4,10};
        int num = 6;
        sort(arr, num);
        System.out.println(Arrays.toString(arr));
    }
}

荷兰国旗问题是面试中经常遇到的算法考题,一起来看题目要求并分析题目求解过程。
问题2:给定一个数组arr,和一个数num,请把小于num的数放在数组的左边,等于num的数组的中间,大于num的数放在数组的右边。
要求额外空间复杂度O(1),时间复杂度O(n)
这道题和上述的题目有些相似,唯一不同的是加了个等于num放中间这个条件,其实如果你理解了上述的题目的含义,这道题也不难理解了。解这道题需要将数组划分为3部分,即小于num、等于num、大于num,这样一来我们建立三个索引,分别是左边界的,当前位置索引,右边界的索引,通过移动索引并交换位置即可得到答案。

代码

import java.util.Arrays;

//荷兰国旗问题
public class NetherlandsFlag {
    
    public static int[] partition(int[]arr,int L,int R,int num) {
        int less = L - 1;  //左边界的索引
        int more = R + 1;  //右边界的索引
        int cur = L;  //当前位置的索引
        while(cur < more) {
            if(arr[cur] < num) {
                swap(arr, ++less, cur++);
            }else if(arr[cur] > num) {
                swap(arr, --more, cur);// cur不自加的原因是交换后的arr[cur]可能等于num,也可能小于,或者是大于,需要再做一次判断
            }else {  
                cur++;
            }
        }
        return new int[] {less+1,more-1};
    }
    
    public static void swap(int [] arr,int i,int j) {
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }
    
    public static void main(String[] args) {
        int[] arr = {23,4,2,5,7,42,34,32,7,8,39,45,63,21};
        int num = 34;
        partition(arr, 0, arr.length-1, num);
        System.out.println(Arrays.toString(arr));
    }
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 1. 找出数组中重复的数字 题目:在一个长度为n的数组里的所有数字都在0到n-1的范围内。数组中某些数字是重复的,...
    BookThief阅读 1,836评论 0 2
  • 1.把二元查找树转变成排序的双向链表 题目: 输入一棵二元查找树,将该二元查找树转换成一个排序的双向链表。 要求不...
    曲终人散Li阅读 3,394评论 0 19
  • 1、用C语言实现一个revert函数,它的功能是将输入的字符串在原串上倒序后返回。 2、用C语言实现函数void ...
    希崽家的小哲阅读 6,377评论 0 12
  • 来日方长
    q村阅读 120评论 0 1
  • 我国的茶文化可以说是源远流长、博大精深,即使是流传到了今天,也有不少人爱在闲暇时刻跟友人一起喝喝茶。不得不说,喝茶...
    小宇杂谈阅读 365评论 0 1