标签:degree int 拓扑 edge BFS grap edges 1245 size
1245. 树的直径
给你这棵「无向树」,请你测算并返回它的「直径」:这棵树上最长简单路径的 边数。
我们用一个由所有「边」组成的数组 edges
来表示一棵无向树,其中 edges[i] = [u, v]
表示节点 u
和 v
之间的双向边。
树上的节点都已经用 {0, 1, ..., edges.length}
中的数做了标记,每个节点上的标记都是独一无二的。
示例 1:
输入:edges = [[0,1],[0,2]] 输出:2 解释: 这棵树上最长的路径是 1 - 0 - 2,边数为 2。
示例 2:
输入:edges = [[0,1],[1,2],[2,3],[1,4],[4,5]] 输出:4 解释: 这棵树上最长的路径是 3 - 2 - 1 - 4 - 5,边数为 4。
提示:
0 <= edges.length < 10^4
edges[i][0] != edges[i][1]
0 <= edges[i][j] <= edges.length
edges
会形成一棵无向树
class Solution { public: int treeDiameter(vector<vector<int>>& edges) { int n = edges.size() + 1; vector<vector<int>> grap(n); vector<std::pair<int,int>> ans; vector<int> degree(n); // 构建无向图 for(auto &edge : edges){ grap[edge[0]].push_back(edge[1]); grap[edge[1]].push_back(edge[0]); degree[edge[0]]++; degree[edge[1]]++; } queue<int> q; for (unsigned int i = 0; i < degree.size(); i++) { // 针对无向图入度为1的入队 if (degree[i] == 1) { q.push(i); } } int step = 0; while (q.size() > 2) { // 等于2时找到树的直径 n -= q.size(); int mSize = q.size(); for (int i = 0; i < mSize; i++) { int cur = q.front(); q.pop(); degree[cur] = 0; // 防重复 for (auto &i : grap[cur]){ degree[i]--; if (degree[i]==1) {//叶子节点入队列 q.push(i); } } } step += 2; } return step + n - 1; } };
标签:degree,int,拓扑,edge,BFS,grap,edges,1245,size 来源: https://www.cnblogs.com/MGFangel/p/16294292.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。