LeetCode 1273. 删除树节点（拓扑排序/DFS）

文章目录

- 1. 题目
- 2. 解题
- - 2.1 取巧解
  - 2.2 拓扑排序
  - 2.3 建图+DFS

1. 题目

给你一棵以节点 0 为根节点的树，定义如下：

节点的总数为 nodes 个；
第 i 个节点的值为 value[i] ；
第 i 个节点的父节点是 parent[i] 。
请你删除节点值之和为 0 的每一棵子树。

在完成所有删除之后，返回树中剩余节点的数目。

示例：
在这里插入图片描述

输入：nodes = 7, 
parent = [-1,0,0,1,2,2,2], 
value = [1,-2,4,0,-2,-1,-1]
输出：2提示：
1 <= nodes <= 10^4
-10^5 <= value[i] <= 10^5
parent.length == nodes
parent[0] == -1 表示节点 0 是树的根。

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/delete-tree-nodes
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

2.1 取巧解

数据很特殊，数据尾部为更深的节点，逆序遍历即是自底向上
该解法不通用

class Solution {
public:int deleteTreeNodes(int nodes, vector<int>& parent, vector<int>& value) {int i, n = parent.size();vector<int> count(n,1);for(i = n-1; i >= 0; --i){if(value[i]==0)count[i] = 0;//自己子树和为0，删除节点if(parent[i] != -1){value[parent[i]] += value[i];//值加给父节点count[parent[i]] += count[i];//父节点底下不为0的节点个数}}return count[0];}
};

60 ms 21.1 MB

2.2 拓扑排序

class Solution {
public:int deleteTreeNodes(int nodes, vector<int>& parent, vector<int>& value) {int i, n = parent.size();vector<int> indegree(n,0);for(i = 0; i < n; ++i)if(parent[i] != -1)indegree[parent[i]]++;queue<int> q;for(i = 0; i < n; ++i)if(indegree[i] == 0)q.push(i);vector<int> count(n,1);while(!q.empty()){int tp = q.front();q.pop();if(value[tp]==0)count[tp] = 0;if(parent[tp] == -1)continue;if(--indegree[parent[tp]]==0)q.push(parent[tp]);value[parent[tp]] += value[tp];count[parent[tp]] += count[tp];}return count[0];}
};

88 ms 22.2 MB

2.3 建图+DFS

class Solution {vector<vector<int>> edges;vector<int> count;
public:int deleteTreeNodes(int nodes, vector<int>& parent, vector<int>& value) {int i, n = parent.size();edges.resize(n);count = vector<int>(n,1);for(i = 0; i < n; ++i)if(parent[i] != -1)edges[parent[i]].push_back(i);dfs(0, parent, value);return count[0];}void dfs(int id, vector<int>& parent, vector<int>& value){for(int next : edges[id]){dfs(next, parent, value);}if(value[id]==0)count[id] = 0;if(parent[id] != -1){value[parent[id]] += value[id];count[parent[id]] += count[id];}}
};