来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/erect-the-fence
题目描述
在一个二维的花园中,有一些用 (x, y) 坐标表示的树。由于安装费用十分昂贵,你的任务是先用最短的绳子围起所有的树。只有当所有的树都被绳子包围时,花园才能围好栅栏。你需要找到正好位于栅栏边界上的树的坐标。
示例 1:
输入: [[1,1],[2,2],[2,0],[2,4],[3,3],[4,2]]
输出: [[1,1],[2,0],[4,2],[3,3],[2,4]]
解释:
示例 2:
输入: [[1,2],[2,2],[4,2]]
输出: [[1,2],[2,2],[4,2]]
解释:
即使树都在一条直线上,你也需要先用绳子包围它们。
注意:
所有的树应当被围在一起。你不能剪断绳子来包围树或者把树分成一组以上。
输入的整数在 0 到 100 之间。
花园至少有一棵树。
所有树的坐标都是不同的。
输入的点没有顺序。输出顺序也没有要求。
解题思路
第一次遇到这种类型的题,正好借此机会学习了三种凸包算法:
Jarvis算法:Jarvis算法的原理十分的简单,首先,对于一堆点,找到边缘处的点。
如图,我找到了最左侧的点A,这个点在边缘,所以一定会在栅栏上面,然后随笔选取一点B做向量AB
接下来,遍历每一个点,找到向量AB最右侧的点C(关于如何找最右侧,可以使用向量叉乘,如果叉乘和小于0,那么说明AC是AB顺时针旋转得到的,所以点C肯定会在向量AB的右侧,将C点当做B点继续遍历其他的点,就可以找到向量AB最右侧的点C了,注意,利用向量叉乘会出现0的情况,细节如何处理下文会介绍)
点AC就可以确定是外轮廓线,也就是栅栏的位置,接下来使用C作为起点重命名为A继续上述的步骤,直到C回到起始点A为止,外轮廓也就是栅栏就确定下来了。
在这个过程中会有一些特殊的情况,入下图
如果起始点从A遍历到B的时候,下一个点如果选中的是A,那么利用叉乘求最右侧点时候,如果先遍历的是C,那么就会出现叉乘为0而漏掉C的情况,在某些情况还会出现死循环,所以第一个选择的点需要一些技巧,这里每次指定这个点是数组中循环的下一个点而不是每次从数组第一个点开始遍历。
但是当起点是A点时候,如果第一个点选中的是C, 由于AB和AC叉乘为0,所以B点会被漏选,所以在每次选中最右点后,要判断是否有点与起点和最右点共线,将这些点也应该加入篱笆。
此算法对于每个点都要遍历一次其余各点,所以时间复杂度为O(n2)。
代码展示
Jarvis算法:
class Solution { public: vector<vector<int>> outerTrees(vector<vector<int>>& trees) { if(trees.size() < 4) return trees; vector<vector<int>> vviRet; unordered_set<int> setiVistied; int iStart = 0, iCur = 0; /* find leftest tree*/ for(int i = 0; i < trees.size(); i++) { if (trees[i][0] < trees[iStart][0]) { iStart = i; } } iCur = iStart; do { int iNext = (iCur + 1) % trees.size(); int iDirX = trees[iNext][0] - trees[iCur][0]; int iDirY = trees[iNext][1] - trees[iCur][1]; for(int i = 0; i < trees.size(); i++) { int iTempX = trees[i][0] - trees[iCur][0]; int iTempY = trees[i][1] - trees[iCur][1]; if(iDirX * iTempY - iDirY * iTempX < 0) { iDirX = iTempX; iDirY = iTempY; iNext = i; } } for(int i = 0; i < trees.size(); i++) { int iTempX = trees[i][0] - trees[iCur][0]; int iTempY = trees[i][1] - trees[iCur][1]; if(iDirX * iTempY - iDirY * iTempX == 0 && setiVistied.find(i) == setiVistied.end()) { setiVistied.emplace(i); vviRet.emplace_back(vector<int>{trees[i][0], trees[i][1]}); } } iCur = iNext; } while(iCur != iStart); return vviRet; } };