例一 颜色分类 (leetccode 75)
https://leetcode-cn.com/problems/sort-colors/https://leetcode-cn.com/problems/sort-colors/
给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
必须在不使用sort函数的情况下解决这个问题。
示例1:
示例2:
题目分析:
本题是经典的「荷兰国旗问题」,由计算机科学家 Edsger W. Dijkstra 首先提出。
根据题目中的提示,我们可以统计出数组中 0, 1, 2 的个数,再根据它们的数量,重写整个数组。这种方法较为简单,也很容易想到,而本题解中会介绍基于指针进行交换的方法。
因为只有 0,1,2 三种数字,所以我们划分出三种区域来存放这三种数字,存放完后即完成了顺序排序。
我们定义两个指针 q,p 分别作为区域的分界线,其中初始 q = 0, p = nums.size() - 1 。
从左往右遍历一遍,如果 nums[ i ] == 0 就与 q 所指向的元素交换位置,同时 q 指针往右移动一个单位;如果 nums[ i ] == 2 就与 p 所指向的元素交换位置,同时 p 往左移动一个单位,实现区域存放。
值得注意的是在我们将 nums[ i ] 交换位置后,此时新的 nums[ i ] 可能仍为 0 或 2 ,但指针已移动,所以我们不会再考虑新的 nums[ i ] ,以至于结果不是我们所想要的。
所以我们需先遍历一遍,找到所有的 2 ,这样新的 nums[ i ] 就不会是 2 ,这样我们只需要找 0 即可,0进行换位,如果是 1,那么就不需要进行任何后续的操作。
由于其中一个指针 p 是从右向左移动的,因此当我们在从左向右遍历整个数组时,如果遍历到的位置超过了 p,那么就可以直接停止遍历了。
看一下代码
题解代码:
class Solution {
public:
void sortColors(vector<int>& nums) {
int n = nums.size();
int q = 0, p = n - 1;
for(int i = 0; i <= p; i ++){
while(i <= p && nums[i] == 2){
swap(nums[i], nums[p]);
p --;
}// 先遍历一遍,找到所有的 2
if(nums[i] == 0){
swap(nums[i], nums[q]);
q ++;
}
}
}
};
例二 救生艇 (leetcode 881)
给定数组 people 。people[ i ] 表示第 i 个人的体重 ,船的数量不限,每艘船可以承载的最大重量为 limit 。
每艘船最多可同时载两人,但条件是这些人的重量之和最多为 limit 。
返回 承载所有人所需的最小船数 。
示例1:
示例2:
示例3:
题目分析:
本题也运用了贪心思想
因为每艘船最多能载两个人,要使船数最少,应使载两人的船尽可能的多。
我们先对数组进行 sort 排序,这样我们能知道体重最轻和体重最重的人(数组头与数组尾),同时我们定义两个指针 q = 0 , p = people.size() - 1 分别指向最轻与最重,将这两者体重相加,若小于等于 limit ,我们就可将两者放到一个船上,这时将 q 向右移动,p 向左移动(得到新的最重与最轻);若大于 limit ,那么只能将最重的单独放到一条船上,这时只将 p 左移。
统计每次移动情况,即可得到最小船数。
题解代码:
class Solution {
public:
int numRescueBoats(vector<int>& people, int limit) {
sort(people.begin(), people.end());
int n = people.size();
int q = 0, p = n - 1;
int count = 0;
while(q <= p){
if(people[q] + people[p] <= limit){
q ++;
p --;
count ++;
}else{
p --;
count ++;
}
}
return count;
}
};