This commit is contained in:
chefyuan
2021-03-18 09:28:12 +08:00
parent 6879ed51a2
commit 4ca5164ea9
6 changed files with 344 additions and 73 deletions

View File

@@ -112,7 +112,7 @@ class Solution {
**动图解析**
![](https://img-blog.csdnimg.cn/2021031809231883.gif#pic_center)
**题目代码**

View File

@@ -1,17 +1,17 @@
### leetcode 75 颜色分类
**leetcode 75 颜色分类**
**题目描述**
题目描述
> 给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
>
> 此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
**示例 1**
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
示例 1
> 输入nums = [2,0,2,1,1,0]
> 输出:[0,0,1,1,2,2]
**示例 2**
示例 2
> 输入nums = [2,0,1]
> 输出:[0,1,2]
@@ -26,80 +26,30 @@
> 输入nums = [1]
> 输出:[1]
**两次遍历**
**做题思路**
**解析:**
这个题目我们使用 Arrays.sort() 解决,哈哈,但是那样太无聊啦,题目含义就是让我们将所有的 0 放在前面2放在后面1 放在中间,是不是和我们上面说的荷兰国旗问题一样。我们仅仅将 1 做为 pivot 值。
通过两次遍历,第一遍遍历首先将 0 归位,第二遍遍历将 1 归位,自然 2 也就被归位了,这个方法很容易理解我们直接看代码吧
**题目代码**
```java
class Solution {
public void sortColors(int[] nums) {
int pro = 0;
//将0归位
for (int i = 0; i < nums.length; ++i) {
if (nums[i] == 0) {
swap(nums,i,pro);
pro++;
}
}
//将1归位
for (int j = pro; j < nums.length; ++j) {
if(nums[j] == 1) {
swap(nums,j,pro);
pro++;
}
}
}
public void swap(int[] nums,int i, int j) {
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
}
```
**一次遍历**
**解析**
两次遍历实现是十分容易的,也很容易实现,那么我们可不可以一次遍历就将其归位呢?其实这里也利用了我们双指针的思想,我们首先定义两个指针,一个位于数组头部,一个位于数组尾部,当我们遇到 0 时则给我们头部指针交换,遇到 2 时,则给尾部指针交换。不过里面有两个细节我们需要注意。
1.遍历数组时,指针的上界,不能是 i < nums.length;应是 i <= right;
2.需要注意交换后当前指针指向的仍不为1
干看上面两种注意事项可能不是特别容易理解我们来看我们的视频解析就可以搞懂这两个情况啦
**动图解析**
![leetcode75颜色分类](https://cdn.jsdelivr.net/gh/tan45du/test1@master/20210122/leetcode75颜色分类.5w4sa458rr40.gif)
**题目代码**
下面我们直接看代码吧,和三向切分基本一致
```java
class Solution {
public void sortColors(int[] nums) {
int len = nums.length;
int left = 0;
//这里和三向切分不完全一致
int i = left;
int right = len-1;
//注意看 for 循环的结束条件
for (int i = 0; i <= right; ++i) {
if (nums[i] == 0) {
swap(nums,i,left);
left++;
}
if (nums[i] == 2) {
swap(nums,i,right);
right--;
// 发现此时不为1则指针不移动继续交换
if (nums[i] != 1) {
i--;
}
}
}
while (i <= right) {
if (nums[i] == 2) {
swap(nums,i,right--);
} else if (nums[i] == 0) {
swap(nums,i++,left++);
} else {
i++;
}
}
}
public void swap (int[] nums, int i, int j) {
int temp = nums[i];
@@ -109,3 +59,54 @@ class Solution {
}
```
另外我们看这段代码,有什么问题呢?那就是我们即使完全符合时,仍会交换元素,这样会大大降低我们的效率。
例如:[0,0,0,1,1,1,2,2,2]
此时我们完全符合情况不需要交换元素但是按照我们上面的代码0,2 的每个元素会和自己进行交换,所以这里我们可以根据 i 和 left 的值是否相等来决定是否需要交换,大家可以自己写一下。
下面我们看一下另外一种写法
这个题目的关键点就是,当我们 nums[i] 和 nums[right] 交换后,我们的 nums[right] 此时指向的元素是符合要求的,但是我们 nums[i] 指向的元素不一定符合要求,所以我们需要继续判断。
![细节地方](https://cdn.jsdelivr.net/gh/tan45du/test@master/photo/微信截图_20210305153911.28capmzljy80.png)
我们 2 和 0 交换后,此时 i 指向 0 0 应放在头部,所以不符合情况,所以 0 和 1 仍需要交换。下面我们来看一下动画来加深理解吧。
![leetcode75颜色分类](https://cdn.jsdelivr.net/gh/tan45du/test1@master/20210122/leetcode75颜色分类.5w4sa458rr40.gif)
另一种代码表示
```java
class Solution {
public void sortColors(int[] nums) {
int left = 0;
int len = nums.length;
int right = len - 1;
for (int i = 0; i <= right; ++i) {
if (nums[i] == 0) {
swap(nums,i,left);
left++;
}
if (nums[i] == 2) {
swap(nums,i,right);
right--;
//如果不等于 1 则需要继续判断,所以不移动 i 指针i--
if (nums[i] != 1) {
i--;
}
}
}
}
public void swap (int[] nums,int i, int j) {
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
}
```
好啦,这个问题到这就结束啦,是不是很简单啊,我们明天见!