LeetCode 2049 统计最高分的节点数目[dfs 二叉树] HERODING的LeetCode之路

解题思路

读题可知该题并没有提供构建好的二叉树，而是给了父节点关系的数组，这就需要我们自己构建二叉树。一个比较简单的方式是用二维数组存储每个节点的子节点，这也有利于深度优先搜索。在dfs部分，从根节点出发，不断找其各个子节点所在子树的节点个数。思考一下可以发现，每删去一个节点，最多只有三个子树，左右子树和父节点所在树的连通分量，所以在把左右子树的节点个数相乘后，如果是非根节点那就还有非根节点所在连通分量，这要用总节点个数（n-1，不能包括当前节点）减去左右子树节点个数，这样剩下的就是非根节点所在连通分量，再和之前的分数相乘得到当前节点的最终分数，和maxScore进行比较，更新个数或者更新maxScore，返回值就是当前节点所在子树的个数（n - 非根节点连通分量大小）。
注释已经很详尽了，代码如下：

代码

class Solution {
private:
    int n;
    int ans = 0;
    // 防止溢出
    long maxScore = 0;
    vector<vector<int>> children;
public:
    int countHighestScoreNodes(vector<int>& parents) {
        this->n = parents.size();
        this->children = vector<vector<int>>(n);
        for(int i = 0; i < n; i ++) {
            // 注意根节点没有父节点
            if(parents[i] != -1) {
                children[parents[i]].push_back(i);
            }
        }
        dfs(0);
        return ans;
    }

    int dfs(int i) {
        // 防止溢出
        long score = 1;
        // 用作记录最后剩下的父节点所在的连通个数
        int size = n - 1;
        for(int c : children[i]) {
            int temp = dfs(c);
            // 子树乘积计算分数
            score *= temp;
            size -= temp;
        }
        // 非根节点一定有父节点
        if(i != 0) {
            score *= size;
        }
        // 更新最大分数个数
        if(score == maxScore) {
            ans ++;
        }
        // 更新最大分数
        if(score > maxScore) {
            maxScore = score;
            ans = 1;
        }
        // 返回包括当前节点在内的子树的节点个数
        return n - size;
    }
};