[leetcode]2049统计最高分的节点数目,深搜

题意

(题目的说法虽然没什么错误,但我还是觉得十分难理解,所以直接看题目下面示例1的那张图会比较好)第一段就是说一个数组parents代表每个节点的父节点是谁,0节点一定是根节点,所以数组第一个数一定为-1。然后这棵树一定会是二叉树。
第二段就是题目十分绕口的地方,这个二叉树上每个节点都有它自己相对应的分数,这个分数的计算方法:把当前节点删去,原本的二叉树有可能被分为一个或两个或三个二叉树,删除叶子节点后,二叉树还是二叉树,但若是删除有左右孩子的节点,二叉树就可以分出多个二叉树,题目要求的就是剩余(一个或多个)子树节点个数的乘积(看题目示例的图会很好理解的)
在这里插入图片描述

求有最高得分的节点个数。

思路

这里有一个能够建立二叉树的数组,首先就是要建树,因为节点不多且是二叉树,用二维的vector就行,然后0节点一定是根节点建树可以不遍历它。
一个节点被删除后,它的分数可以分为三个区域的乘积,它的左子树的结点个数、右子树的结点个数,以及父节点的“连通块”个数,这三个区域需要非空才能进行相乘,两个子树的个数用深搜,父节点区域就是总节点个数减去当前节点的子节点个数(包含其自身),深搜过程统计最大值和最大值出现的个数。
这里的乘积最大值不用用int,反正我wa了一发(惨)

代码

class Solution {
public:
    int ans = 0;
    long long maxFen = 0;
    int n;
    vector<vector<int>> trie;
    int dfs(int x){
        long long fen = 1;
        int zishu = 1;
        for(int y:trie[x]){
            int z = dfs(y);
            fen *= z;
            zishu += z;
        }
        if(x) fen *= n-zishu;
        if(fen==maxFen)++ans;
        else if(fen>maxFen){
            maxFen = fen;
            ans = 1;
        }
        return zishu;
    }
    int countHighestScoreNodes(vector<int>& parents) {
        this->n = parents.size();
        this->trie = vector<vector<int>>(n);
        for(int i = 1; i < n; ++i){
            trie[parents[i]].emplace_back(i);
        }
        dfs(0);
        return ans;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值