75 颜色分类(排序)

1. 问题描述:

给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。

此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意:
不能使用代码库中的排序函数来解决这道题。

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

2. 思路分析:

① 其实题目还是不难理解的,主要的意思是将数组中的数字从小到大进行排序,一个很直接的想法是通过一层for循环对数组中的0,1,2三个数字出现的数字进行计数看一下各自出现的次数是多少,然后再使用一个for循环将对应次数的0,1,2数字填充到原数组中即可,这个方法的好处是简单明了容易解决问题,代码如下:

import java.util.Scanner;
public class Solution {
    public void sortColors(int[] nums) {
        int zero = 0, one = 0, two = 0;
        for (int i = 0; i < nums.length; ++i){
            if (nums[i] == 0){
                zero++;
            }else if (nums[i] == 1){
                one++;
            }else if (nums[i] == 2){
                two++;
            }
        }
        for (int i = 0; i < nums.length; ++i){
            if (i < zero){
                nums[i] = 0;
            }else if (i - zero < one){
                nums[i] = 1;
            }else if (i - zero - one < two){
                nums[i] = 2;
            }
        }
       /* for (int i = 0; i < nums.length; ++i){
            System.out.print(nums[i] + " ");
        }*/
    }
}

② 其实除了上面的解法之外,我们还可以对其进行优化,可以在原数组的基础上进行排序,使用一层for循环就可以将数组中的内部元素进行排序,可以借助于快速排序中多指针的移动的思想来实现排序,将1的元素往中间移动,2的元素往最右边移动,具体实现如下:

1)声明三个指针,一个左指针left,一个当前指向的正在扫描的指针current,另外一个是指向最右端的指针right,当current扫描到的是1的时候那么current直接往右移动,扫描到0的时候判断left指针指向的元素是否是1,假如是1那么交换left与current各自指向的元素,同时扫描到0的时候假如没有交换的时候需要将left指针往右边移动一位,这主要是将0的移到最左边,1的放在中间,当扫描到2的时候将current指针与right指针指向的元素交换,right往前移动一位

2)循环什么时候停止呢?条件为current的位置小于了right的位置,注意这里并不是!,否则有的情况会判断错误的吗,例如测试数据1,0,2就会发生错误,所以要注意一下,写代码的时候最好结合具体的例子写,看一下哪些是需要进行考虑的

3. 代码如下:

使用多指针进行移动代码:

import java.util.Scanner;
public class Solution {
    public void sortColors(int[] nums) {
        int left = 0, right = nums.length - 1, current = 0, temp;
        while(current <= right){
            if (nums[current] == 0){
                /*current与left只存在着两种情况那么就是两者在同一位置要么就是current大于left位置*/
                if (current > left && nums[left] == 1){
                    /*交换元素*/
                    temp = nums[current];
                    nums[current] = nums[left] ;
                    nums[left] = temp;
                    left++;
                }
                if (nums[left] == 0){
                    left++;
                }
                current++;
            }else if (nums[current] == 1){
                current++;
            }else if (nums[current] == 2){
                /*扫描到2*/
                temp = nums[right];
                nums[right] = nums[current];
                nums[current] = temp;
                right--;
            }
        }
    }
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值