题目地址:
https://leetcode.com/problems/erect-the-fence/description/
给定平面若干个点,是各个小树的坐标,现在要求造一个周长最小的围栏,可以让每棵树都恰好被围栏所包围(在边界上也算被包围)。问所有围栏边界上的树的坐标。
即求凸包,可以用Andrew算法,参考https://blog.csdn.net/qq_46105170/article/details/127714131。代码如下:
class Solution {
public:
#define cross(x1, y1, x2, y2) (x1 * y2 - x2 * y1)
int area(vector<int>& p1, vector<int>& p2, vector<int>& p3) {
int x1 = p2[0] - p1[0], y1 = p2[1] - p1[1];
int x2 = p3[0] - p1[0], y2 = p3[1] - p1[1];
return cross(x1, y1, x2, y2);
}
vector<vector<int>> outerTrees(vector<vector<int>>& vv) {
sort(vv.begin(), vv.end());
int n = vv.size();
vector<int> stk;
vector<bool> used(n);
for (int i = 0; i < n; i++) {
// 注意这里是面积大于0,因为边界上的树的坐标也是要返回的
while (stk.size() >= 2 &&
area(vv[stk[stk.size() - 2]], vv[stk.back()], vv[i]) > 0) {
used[stk.back()] = false;
stk.pop_back();
}
stk.push_back(i);
used[i] = true;
}
used[0] = false;
for (int i = n - 1; i >= 0; i--) {
if (used[i]) continue;
while (stk.size() >= 2 &&
area(vv[stk[stk.size() - 2]], vv[stk.back()], vv[i]) > 0)
stk.pop_back();
stk.push_back(i);
}
vector<vector<int>> res;
// 注意栈中第一个点存了两次,最后答案不要加两次
for (int i = 0; i < stk.size() - 1; i++) res.push_back(vv[stk[i]]);
return res;
}
};
时间复杂度 ( n log n ) (n\log n) (nlogn),空间 O ( n ) O(n) O(n)。