LeetCode 1245. 樹的直徑（圖的最大直徑結論）

1. 題目

2. 解題

• 結論：求無權無向圖中的最長一條路徑
  先從任意一點P出發，找到離它最遠的點Q
  再從點Q出發，找離它最遠的點W，W到Q的距離就是最長的一條路

• 採用2次BFS遍歷

class Solution {
public:
    int treeDiameter(vector<vector<int>>& edges) {
    	int n = edges.size()+1;
    	unordered_map<int,unordered_map<int,bool>> m;
    	for(auto& e : edges)
    	{
    		m[e[0]][e[1]] = 1;
    		m[e[1]][e[0]] = 1;
    	}
    	int a,cur,size,step = 0;
    	queue<int> q;
    	vector<bool> visited(n,false);
    	q.push(0);
    	visited[0] = true;
    	while(!q.empty())
    	{
			cur = q.front();
			q.pop();
			a = cur;//記錄最後到達的點a
			for(auto it = m[cur].begin(); it != m[cur].end(); ++it)
			{
				if(!visited[it->first])
				{
					q.push(it->first);
					visited[it->first] = true;
				}
			}
    	}
    	vector<bool> visited1(n,false);
    	q.push(a);//從a出發，到達的最遠的就是最大的路徑
    	visited1[a] = true;
    	while(!q.empty())
    	{
    		size = q.size();
    		while(size--)
    		{
    			cur = q.front();
    			q.pop();
				for(auto it = m[cur].begin(); it != m[cur].end(); ++it)
                {
                    if(!visited1[it->first])
                    {
                        q.push(it->first);
                        visited1[it->first] = true;
                    }
                }
    		}
    		step++;
    	}
    	return step-1;
    }
};

136 ms 24.4 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公衆號（Michael阿明），一起加油、一起學習進步！