牛牛打怪兽

思路

统计从结点 到叶子节点路径权值和小于等于 的叶子数量即可。
时间复杂度:

Code

/**
 * struct Point {
 *    int x;
 *    int y;
 * };
 */

class Solution {
private:
    #define maxn 100010
    vector<int> g[maxn];
    int val[maxn];
    int ans;
public:
    /**
     * 从 1 到 n 每个结点的权值。
     * @param n int整型
     * @param Edge Point类vector (u, v) 集合
     * @param q int整型
     * @param Query Point类vector Point.x 为题中 r, Point.y题中 为 x
     * @return long长整型vector
     */
    void dfs(int x, int fa = 0, int sum = 0) {
        sum += val[x];
        if (sum > 2) return;
        if (g[x].size() == 1 && x != 1) ++ans;
        for (auto v: g[x]) {
            if (v == fa) continue;
            dfs(v, x, sum);
        }
    }
    int solve(int n, vector<Point>& Edge, vector<int>& f) {
        // write code here
        for (int i = 1; i <= n; ++i) {
            g[i].clear();
            val[i] = f[i-1];
        }
        ans = 0;
        for (auto cur: Edge) {
            g[cur.x].push_back(cur.y);
            g[cur.y].push_back(cur.x);
        }
        if (n > 1) dfs(1);
        else ans = 1;
        return ans;
    }
};
全部评论

相关推荐

12-01 12:34
已编辑
广东工业大学 Java
如题,fw🐭🐭,加上准备的太晚,大三上已找不到日常实习,导致连锁反应,下学期的暑期实习找不到好的实习,导致秋招找不到中大厂,现在是中小厂Java还有考公的选择,由于有些中小厂工作强度比肩大厂,钱还少,感觉不如考公如果🐮u们是我现在这种情况,会怎么选?
负债的混子:关注你一段时间了,突然发现你头像名字都改了,想必是这段时间压力很大。关于就业还是考公的选择,就像很多牛友说的:不要美化自己没走过的路。你现在想往互联网发展,发现这条路很难走,然后想往考公发展,但是你没走过考公这条路,所以你不知道这条路的压力如何。你今年大三了,还有时间给你做选择,我希望你能够尽快的决定自己的方向,然后一条路走到黑,而不是在这里徘徊,每个人的道路是不一样的,你无法复刻别人的路,你能做的就是尽力的完善自己。 最后,我想说的是,加油,陌生人!
点赞 评论 收藏
分享
10-30 10:16
南京大学 Java
永远的鹅孝子:给南大✌️跪了
点赞 评论 收藏
分享
评论
1
收藏
分享
牛客网
牛客企业服务