当前位置: 代码迷 >> 综合 >> POJ 3140 树形dp
  详细解决方案

POJ 3140 树形dp

热度:59   发布时间:2024-01-04 09:13:49.0

题意:一棵n个结点的带权无根树,从中删去一条边,使得剩下来的两棵子树的节点权值之和的绝对值最小,并求出得到的最小绝对值。

水题,首先把每棵子树的权值之和处理出来,在dfs一次做一下比较就好

#include <cmath>
#include <queue>
#include <vector>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define LL long long
using namespace std;
const int maxn = 1e5+7;
int n,w[maxn];
LL dp[maxn],ans,sum;
vector<int>q[maxn];
void dfs(int x,int fa)
{dp[x]=w[x];for(int i=0;i<q[x].size();i++){int u=q[x][i];if(u==fa) continue;dfs(u,x);dp[x]+=dp[u];}
}
void dfss(int x,int fa)
{for(int i=0;i<q[x].size();i++){int u=q[x][i];if(u==fa) continue;LL p=sum-2*dp[u];if(p<0) p=-p;ans=min(ans,p);dfss(u,x);}
}
int main()
{int