米哈游笔试第二题

#include
#include
using namespace std;

int main() {
    long long n;
cin >> n;
    vector> f(n);
    for (long long i = 0; i < n-1; i++) {
        long long tmp;
cin >> tmp;
        f[tmp-1].push_back(i+1);
    }
    long long happy[n];
    for (long long i = 0; i < n; i++)
cin >> happy[i];
    for (long long i = 0; i < n; i++) {
        long long tmp;
cin >> tmp;
        happy[i] += tmp;
    }
    long long ans = 0;
    function dfs = [&](long long i, long long hy) -> void{
        if (f[i].empty()) {
            //hy += happy[i];
            ans = max(hy, ans);
            //hy -= happy[i];
            return;
        }
        //hy += happy[i];
        for (long long x : f[i]) {
            hy += happy[x];
            dfs(x, hy);
            hy -= happy[x];
        }
        
    };
    dfs(0, happy[0]);
    cout << ans;
    
}
感觉我这代码没毛病,gpt没指出错误。到底是哪儿有毛病啊有没有佬能指点一下
全部评论
原来不止我一个人这么认为
1 回复 分享
发布于 10-12 17:28 吉林
并不是只有一条路径吧,假设有3个子树,那么这3个子树都可以带来正的happy,那么是都能过关的呀?
点赞 回复 分享
发布于 10-12 17:33 河北
最后一题咋做啊?感觉是分类讨论然后用数学方法直接计算的。
点赞 回复 分享
发布于 10-12 18:15 美国
从叶节点倒着回溯就可以,只要子树是正贡献,那就加上,否则直接截断,有点像路径上的最长子序列之和
点赞 回复 分享
发布于 10-12 18:23 广东
第二题,我用树形dp写的,然后过了
点赞 回复 分享
发布于 10-12 23:54 江西
好复杂啊,看看下面的 int main() { int n; cin >> n; // 表示前驱的条件 vector<int> con(n); // 表示那两个值的和 vector<int> values(n); for (int i = n - 1; i > 0; ++i) { if (values[i] > 0) values[con[i]] += values[i]; } cout << max(0, values[0]) << endl; return 0; }
点赞 回复 分享
发布于 10-13 14:53 湖北

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务