数据结构笔记:leetcode 452
考虑所有气球中右边界位置最靠左的那一个,那么一定有一支箭的射出位置就是它的右边界(否则就没有箭可以将其引爆了)。当我们确定了一支箭之后,我们就可以将这支箭引爆的所有气球移除,并从剩下未被引爆的气球中,再选择右边界位置最靠左的那一个,确定下一支箭,直到所有的气球都被引爆。
我们可以写出如下的伪代码:
let points := [[x(0), y(0)], [x(1), y(1)], ... [x(n-1), y(n-1)]],表示 n 个气球
let burst := [false] * n,表示每个气球是否被引爆
let ans := 1,表示射出的箭数
将 points 按照 y 值(右边界)进行升序排序
while burst 中还有 false 值 do
let i := 最小的满足 burst[i] = false 的索引 i
for j := i to n-1 do
if x(j) <= y(i) then
burst[j] := true
end if
end for
end while
return ans
这样的做法在最坏情况下时间复杂度是
O
(
N
2
)
O(N^2)
O(N2)
2如何继续进行优化呢?
事实上,在内层的 j循环中,当我们遇到第一个不满足
x
(
j
)
≤
y
(
i
)
x(j) \leq y(i)
x(j)≤y(i)的 j值,就可以直接跳出循环,并且这个 y(j)就是下一支箭的射出位置。
也就可以写出如下的伪代码:
let points := [[x(0), y(0)], [x(1), y(1)], ... [x(n-1), y(n-1)]],表示 n 个气球
let pos := y(0),表示当前箭的射出位置
let ans := 1,表示射出的箭数
将 points 按照 y 值(右边界)进行升序排序
for i := 1 to n-1 do
if x(i) > pos then
ans := ans + 1
pos := y(i)
end if
end for
return ans
这样就可以将计算答案的时间从
O
(
n
2
)
O(n^2)
O(n2) 降低至
O
(
n
)
O(n)
O(n)
C++代码:
class Solution {
public:
int findMinArrowShots(vector<vector<int>>& points) {
//判断是否为空
if(points.empty())
return 0;
//排序
sort(points.begin(),points.end(),[](vector<int>&a ,vector<int>&b){return a[1]<b[1];});//这里匿名函数一定要传引用不然会报错超时
int num=1;
int pos=points[0][1];
for(vector<int> sub :points){
if(sub[0]>pos)
{num++;
pos=sub[1];
}
}
return num;
}
};
代码解释:
sort(points.begin(), points.end(), [](const vector<int>& u, const vector<int>& v) {
return u[1] < v[1];
});
这段代码是一个使用 lambda 表达式的函数对象,它的作用是对一个二维向量数组 points 进行排序,排序的规则是按照每个向量的第二个元素从小到大排序。
lambda 表达式用于定义一个匿名函数,它可以在代码中直接使用,而不需要事先定义一个函数名。在这个 lambda 表达式中,[](const vector& u, const vector& v) 是函数的参数列表,表示这个函数有两个参数 u 和 v,类型都是 vector 的常量引用。箭头符号 -> 后面的 bool 表示这个 lambda 表达式返回一个 bool 类型的值。
在这个 lambda 表达式中,函数体只有一行代码,即 return u[1] < v[1];,它的作用是比较两个向量的第二个元素的大小。当 u 的第二个元素小于 v 的第二个元素时,返回 true,表示 u 应该排在 v 前面;否则返回 false,表示 v 应该排在 u 前面。这个 lambda 表达式的作用是告诉 sort 函数按照指定的规则进行排序。
for (const vector<int>& balloon: points)
这段代码是一个 C++ 的范围 for 循环(range-based for loop),它用于遍历名为 points 的容器,容器中存储着多个 vector 类型的元素。在每次循环中,循环变量 balloon 会依次引用容器中的每个 vector 元素.
const 关键字表示循环中的 balloon 引用是只读的,即不能通过这个引用修改元素的值。这种方式可以避免在循环中无意中修改元素的值而导致程序出错。