一、题目描述
二、算法原理
思路：BFS 算法
1）找到图中不是0，1值，用个二维数组来存储他们的下标
2）排序，根据下标对应的值的大小升序
3）升序：1 -> 2 -> 3......... 的本质就是 1 —> 2 的最短路径，2——> 3 的最短路径等，所以剩下的
BFS 算法的老套路；
三、代码实现
class Solution { int dx[4] = {0,0,-1,1}; int dy[4] = {1,-1,0,0}; typedef pair<int,int> PII; int wight; int hight; public: int cutOffTree(vector<vector<int>>& ft) { wight = ft.size(); hight = ft[0].size(); vector<PII> index;//存储树的坐标 for(int i = 0; i < wight; i++) { for(int j = 0; j < hight; j++) { if(ft[i][j] > 1) index.push_back({i,j}); } } sort(index.begin(),index.end(),[&](PII x,PII y){//根据树的高度升序 return ft[x.first][x.second] < ft[y.first][y.second]; }); int bx = 0,by = 0;//起始点值 int total = 0; for(auto [ex,ey] : index)//终点值 { int sep = BFS(ft,bx,by,ex,ey);//查找目的值的步数 if(sep < 0) return -1;//找不到 total += sep; bx = ex;//更新起点值 by = ey; } return total; } int BFS(vector<vector<int>>& ft,int bx,int by,int ex,int ey) { //BFS 算法 if(bx == ex & by == ey) return 0;//有可能遍历到起始值 queue<PII> que; que.push({bx,by});//起点值 int ret = 0; bool vis[51][51];//标记是否遍历过入 queue 的下标 memset(vis,false,sizeof(vis));//清空 vis[bx][by] = true;//起点值已经遍历过了 while(que.size()) { int size = que.size(); ret++; while(size--)// 层数 { auto [x,y] = que.front(); que.pop(); for(int i = 0; i < 4; i++)// 上下左右遍历 { int a = x + dx[i]; int b = y + dy[i]; if(a >= 0 && b >= 0 && a < wight && b < hight && ft[a][b] && vis[a][b] == false) { if(a == ex && b == ey) { return ret;// 找到目标值 } que.push({a,b});//更新 vis[a][b] = true; } } } } return -1;//找不到 } };
)
