Every day a Leetcode
题目来源:3249. 统计好节点的数目
解法1:深度优先搜索
建树,然后从根节点 0 开始 DFS 这棵树。
DFS 返回子树大小。
对于节点 x,如果其是叶子节点,或者其所有儿子子树大小都一样,那么答案加一。
代码:
/*
* @lc app=leetcode.cn id=3249 lang=cpp
*
* [3249] 统计好节点的数目
*/
// @lc code=start
class Solution
{
public:
int countGoodNodes(vector<vector<int>> &edges)
{
int n = edges.size() + 1; // 节点数
// 构建邻接表
vector<vector<int>> adj(n);
for (vector<int> &edge : edges)
{
int u = edge[0], v = edge[1];
adj[u].push_back(v);
adj[v].push_back(u);
}
int ans = 0;
// parent 为父节点
function<int(int, int)> dfs = [&](int u, int parent) -> int
{
int cnt_sum = 1; // 节点总数,先计入自身
int single_cnt = 0;
bool valid = true;
for (int v : adj[u])
{
// 规避邻接点中已访问过的父节点,不需要 visited 数组
if (v == parent)
continue;
int cnt = dfs(v, u);
cnt_sum += cnt;
// 检测子树的节点数量是否相同
if (single_cnt == 0)
single_cnt = cnt;
else if (single_cnt != cnt)
valid = false;
}
ans += valid;
return cnt_sum;
};
dfs(0, -1); // -1 即不存在父节点
return ans;
}
};
// @lc code=end
结果:
复杂度分析:
时间复杂度:O(n),其中 n 是数组 edges 的长度。
空间复杂度:O(n),其中 n 是数组 edges 的长度。