XDOJ_1164_逆向DFS

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了XDOJ_1164_逆向DFS相关的知识,希望对你有一定的参考价值。

http://acm.xidian.edu.cn/problem.php?id=1164

 

因为存孩子会爆,只能用孩子存父节点,然后逆向的dfs,期间需要一个数组保存累加值。

 

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;

int a[10000005],pre[10000005] = {0},sum[10000005] = {0},n,u,v;
bool vis[10000005] = {0};

long long dfs(int pos)
{
    if(vis[pos])    return 0;
    vis[pos] = 1;
    if(!pre[pos])
    {
        sum[pos] -= a[pos];
        return abs(sum[pos]);
    }
    long long t = dfs(pre[pos]);
    a[pos] += sum[pre[pos]];
    t += abs(a[pos]);
    sum[pos] += sum[pre[pos]]-a[pos];
    return t;
}

int main()
{
    scanf("%d",&n);
    for(int i = 1;i < n;i++)
    {
        scanf("%d%d",&u,&v);
        pre[v] = u;
    }
    for(int i = 1;i <= n;i++)   scanf("%d",&a[i]);
    long long ans = 0;
    for(int i = 1;i <= n;i++)   ans += dfs(i);
    printf("%lld\n",ans);
    return 0;
}

 

以上是关于XDOJ_1164_逆向DFS的主要内容,如果未能解决你的问题,请参考以下文章

XDOJ_1143_DFS

XDOJ_1084_dfs

XDOJ_1104_dfs

XDOJ_1117_状态压缩DP

XDOJ_1037_贪心

XDOJ_1160_set