Java学习手册:(数据结构与算法-数组)Sort Colors(leetcode75)

题目:

给定一个有n个元素的数组,数组中元素的取值只有0,1,2三种可能。为这个数组排序。

方法一:

计数排序:分别统计0、1、2的元素个数。

方法一代码:
package com.haobi;

public class SortColors {
	public static void main(String[] args) {
		int[] arr = {0,1,2,0,2,1,0,2,1,2,2,1,0,1};
		sortColors(arr);
		for(int a : arr) {
			System.out.print(a+" ");
		}
	}
	
	public static void sortColors(int[] nums) {
		int[] count = new int[3];
		for(int i=0;i<nums.length;i++) {
			//加入断言
			assert(nums[i] >= 0 && nums[i] <= 2);
			//题目保证了nums[i]一定是0、1、2
			count[nums[i]]++;
		}
		int index = 0;
		for(int i=0;i<count[0];i++) {
			nums[index++] = 0;
		}
		for(int i=0;i<count[1];i++) {
			nums[index++] = 1;
		}
		for(int i=0;i<count[2];i++) {
			nums[index++] = 2;
		}
	}
}
方法二:

Java学习手册:(数据结构与算法-排序)三路快排
由于刚好有三个元素,可以采样三路快排的思想,对整个数组采取一次三路快拍。
时间复杂度:O(n)
空间复杂度:O(1)

方法二代码:
package com.haobi;

public class SortColors1 {
	public static void main(String[] args) {
		int[] arr = {0,1,2,0,2,1,0,2,1,2,2,1,0,1};
		sortColors(arr);
		for(int a : arr) {
			System.out.print(a+" ");
		}
	}
	
	public static void sortColors(int[] nums) {
		int zero=-1;//nums[0...zero] == 0
		int two = nums.length;//nums[two...n-1] == 2
		int temp = 0;
		for(int i=0;i<two;) {//直到i和two相遇
			if(nums[i] == 1) {
				i++;
			}else if(nums[i] == 2) {
				temp = nums[--two];
				nums[two] = nums[i];
				nums[i] = temp;
			}else {//nums[i] == 0
				assert(nums[i] == 0);
				temp = nums[++zero];
				nums[zero] = nums[i];
				nums[i++] = temp;
			}
		}
	}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值