题意:
给定一棵带权树和它的根,要求删除一些边使得叶子节点无法到达根节点,求删除边的权值之和的最小值。
思路:
树型dp,其本质为搜索,即遍历树并在返回时维护相关的值。
将原树拆分成以其孩子为根的多个子树,对于一棵子树有两种情况:
1.删除子树的根与其双亲相连的边。
2.删除子树的边。
只要对这两种情况取最小值再回代即可。
代码:
#include <iostream> #include <queue> #include <set> #include <map> #include <vector> #include <stack> #include <cmath> #include <algorithm> #include <cstdio> #include <cctype> #include <functional> #include <string> #include <cstring> #include <sstream> #include <deque> using namespace std; typedef long long ll; typedef pair<int,int> P; typedef pair<int,P> Q; const int inf1=2e9+9; const ll inf2=2e18+9; const int maxn=1e5+9; const int maxx=2e5+9; int n,m,s; vector<P> ar[maxn]; ll f[maxn]; void init() { scanf("%d%d%d",&n,&m,&s); for(int i=0;i<m;i++) { int u,v,w; scanf("%d%d%d",&u,&v,&w); ar[u].push_back(P(v,w)); ar[v].push_back(P(u,w)); } } void dfs(int t,int pre) { if(t!=s&&ar[t].size()==1) { f[t]=inf2; return; } for(int i=0;i<ar[t].size();i++) { ll v=ar[t][i].first,c=ar[t][i].second; if(v==pre) continue; dfs(v,t); f[t]+=min(f[v],c); } } int main() { init(); dfs(s,-1); printf("%lld\n",f[s]); }