力扣—75颜色分类

题目解析

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

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

必须在不使用库内置的 sort 函数的情况下解决这个问题。

示例 1：

输入：nums = [2,0,2,1,1,0]
输出：[0,0,1,1,2,2]
示例 2：

输入：nums = [2,0,1]
输出：[0,1,2]

提示：
n == nums.length
1 <= n <= 300
nums[i] 为 0、1 或 2

解题思路

我们定义两个变量left=-1和right=0，用right从左到右遍历数组，当遇到nums[right]==1是right直接跳过，当遇到nums[right]==0时，我们就需要将nums[left]和nums[right]做交换，将0放到数组前面，并让left和right同时加加，当遇到nums[right]==2s时我们将nums[right]放到末尾，所以交换nums[right]和nums[numsSize-1]。但是我们不知道交换的nums[numsSize-1]具体是什么，所以right不能加加，但是numsSize需要减减，因为数组最后肯定是要放2的，必须让交换回来的nums[numsSize-1]再判断一次。

代码实现

C语言

void sortColors(int* nums, int numsSize) 
{
	int left=-1;
	int right=0;
	int len=numsSize;
	while(right<len)//必须要小于len因为len下标后面的元素全为2已经排好了
	{
       if(nums[right]==2)
       {
       	len--;
	   	int temp=nums[len];
	   	nums[len]=nums[right];
	   	nums[right]=temp;
	   }
	   else if(nums[right]==0)
	   {
	   	left++;
	   	int temp=nums[left];
	   	nums[left]=nums[right];
	   	nums[right]=temp;
	   	right++;
	   }
	   else
	   {
	   	right++;
	   }
	} 
}

1.变量定义：
2.left 初始值为 -1，用于记录已经排好序的0的右边界。
3.right 初始值为 0，用于遍历数组中的元素。
4.len 初始值为 numsSize，表示数组的长度。
5.循环遍历数组：
6.使用 while 循环遍历数组，条件是 right < len，即右指针 right 小于数组长度 len。
7.处理元素为2的情况：
8.如果 nums[right] 等于 2，将 right 指向的元素与 len-1 处的元素交换位置，然后将 len 减一。这样做是为了确保 right 指向的元素是已经处理过的或者是 1，而不是重新交换过的 2。
9.处理元素为0的情况：
10.如果 nums[right] 等于 0，将 left 的位置右移一位（left++），然后将 left 和 right 指向的元素进行交换。交换后，right 指针右移一位（right++）。
11.处理元素为1的情况：
12.如果 nums[right] 等于 1，直接将 right 指针右移一位（right++）。
13.循环结束：
14.当 right 大于等于 len 时，遍历结束，此时数组已经按照0、1、2的顺序排好。
注意：while(right<len)//必须要小于len因为len下标后面的元素全为2已经排好了
如果没看懂可以自己调试代码一步一步看。

class Solution {
    public void sortColors(int[] nums) {
        int zero = -1;
        int two = nums.length; 
        int i = 0;
        while(i < two){
            int tmp;
            if(nums[i] == 0) {
                zero++;
                tmp = nums[zero];
                nums[zero] = nums[i];
                nums[i++] = tmp;
            }else if(nums[i] == 2){
                two--;
                tmp = nums[i];
                nums[i] = nums[two];
                nums[two] = tmp;
            }else {
                i++;
            }
        }
    }
}

这个方法只能局限于排列三种
方法二：

void sortColors(int* nums, int numsSize) 
{
    int n1=0,n2=0;
    for(int i=0;i<numsSize;i++)
    {
		int num=nums[i];
		nums[i]=2;
		if(num<2)
		{
			nums[n1++]=1;
		}
		if(num<1)
		{
			nums[n2++]=0;
		}
	}
}

这个方法很巧妙
1.变量定义：
2.n1 和 n2 分别初始化为 0，用于记录数组中已经放置好的 1 和 0 的位置。
3.遍历数组：
4.使用 for 循环遍历数组 nums。
5.处理当前元素 num：
6.将 nums[i] 的值赋给 num 变量。
7.将 nums[i] 的值设为 2，这是因为在该位置放置了 2，而原来的值已经被记录在 num 中。
8.根据 num 的值进行处理：
9.如果 num 小于 2，将 1 放置在 nums[n1] 的位置，并将 n1 加一，即记录下一个 1 的位置。
10.如果 num 小于 1，将 0 放置在 nums[n2] 的位置，并将 n2 加一，即记录下一个 0 的位置。
11.循环结束：
12.当 for 循环结束时，数组 nums 中的元素已经按照 0、1、2 的顺序排好。

还是那句话看不懂直接将代码粘贴去调试。
欢迎留言，点赞，收藏。