51Nod-1424-零树

ACM模版

描述

题解

这个题是一个比较简单的树归了，从叶子开始向上求出来不同结点所需要的加减次数，最后输出 add[1]+sub[1] 即可，这里需要注意的是，一定要分开求加减次数，最开始我用 dfs 返回了一个 pair 表示某结点操作次数和偏移量，但是最后结果大了许多，仔细斟酌后发现，不同子树的加减操作如果混合在一起求那么会有很多重复的存在，必须将其分开来求，酱紫的话就不会存在子树之间的加减操作的重复现象了。

这个题十分有趣，如果树归基本功可以的话，题解应该是秒出的，而我就是差了那么点……还有注意

long long

，最开始因为这个还 WA 了一回，粗心大意害死人啊！

代码

#include <iostream>
#include <cstdio>
#include <vector>
#include <cmath>

using namespace std;

typedef long long ll;

const int MAXN = 1e5 + 10;

int n;
int val[MAXN];
ll add[MAXN];
ll sub[MAXN];
vector<int> vi[MAXN];

template <class T>
inline bool scan_d(T &ret)
{
char c;
int sgn;
if (c = getchar(), c == EOF)
{
return 0; //EOF
}
while (c != '-' && (c < '0' || c > '9'))
{
c = getchar();
}
sgn = (c == '-') ? -1 : 1;
ret = (c == '-') ? 0 : (c - '0');
while (c = getchar(), c >= '0' && c <= '9')
{
ret = ret * 10 + (c - '0');
}
ret *= sgn;
return 1;
}

void dfs(int root, int pre)
{
for (int i = 0; i < vi[root].size(); i++)
{
int tmp = vi[root][i];
if (tmp != pre)
{
dfs(tmp, root);
add[root] = max(add[root], add[tmp]);
sub[root] = max(sub[root], sub[tmp]);
}
}

val[root] += add[root] - sub[root];
val[root] < 0 ? add[root] -= val[root] : sub[root] += val[root];
}

int main(int argc, const char * argv[])
{
//    freopen("/Users/zyj/Desktop/input.txt", "r", stdin);

scan_d(n);

int a, b;
for (int i = 1; i < n; i++)
{
scan_d(a), scan_d(b);
vi[a].push_back(b);
vi[b].push_back(a);
}
for (int i = 1; i <= n; i++)
{
scan_d(val[i]);
}

dfs(1, 0);

cout << add[1] + sub[1] << '\n';

return 0;
}