Rinne Loves Edges (树形DP)

Rinne Loves Edges

题目描述: Rinne 最近了解了如何快速维护可支持插入边删除边的图，并且高效的回答一下奇妙的询问。
她现在拿到了一个 n 个节点 m 条边的无向连通图，每条边有一个边权 wi 现在她想玩一个游戏：选取一个 “重要点” S，然后选择性删除一些边，使得原图中所有除 S 之外度为 1 的点都不能到达 S。定义删除一条边的代价为这条边的边权，现在 Rinne 想知道完成这个游戏的最小的代价，这样她就能轻松到达 rk1 了！作为回报，她会让你的排名上升一定的数量.
备注: 2≤S≤N≤10^5 ,M=N−1，保证答案在 C++ long long 范围内。

题解: 通过题目和数据分析这是一颗树 , 所以很明显的树形DP, 定义 dp[ i ] 为以 i 为根的最小代价, 则转移方程 : dp[x] += min(dp[v] , w[i] ) , 其中 v 为 x 的孩子,w[i] 为 x 与 v 之间的边权. 可见每次操作 , 要么以父节点与孩子之间的 w[i] 为代价, 要么以其孩子的孩子们为代价 ;

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const  int INF  = 0x3f3f3f3f;
const int N = 1e5+7;
int h[N],e[N*2],ne[N*2],w[N*2],idx=0;	//无向边
int n,m,s;
int dp[N];
int in[N];

void add(int a,int b,int v){
	e[idx] = b,w[idx] = v,ne[idx] = h[a],h[a] = idx++;
}

void DFS(int x,int f)
{
	if(in[x]==1&&x!=s) dp[x] = INF;		//叶子节点只能断其与父亲之间的边权,所以设为INF.
	for(int i=h[x] ; i!=-1;i = ne[i])
	{
		int v = e[i];
		if(v==f) continue;
		DFS(v,x);
		dp[x] += min(dp[v],w[i]);
	}
}

int main()
{
	cin>>n>>m>>s;
	int x,y,v;
	memset(h,-1,sizeof h);
	for(int i=1;i<=m;i++){
		cin>>x>>y>>v;
		add(x,y,v),add(y,x,v);
		in[x]++,in[y]++;
	}
	DFS(s,0);
	cout<<dp[s]<<endl;
	return 0;
}