首页 > 代码库 > XDOJ_1164_逆向DFS
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
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。