杭二集训 tree

全场都AC了的题

Posted by yjjr's blog on January 13, 2018

标签:树形DP

题目

【题目描述】 给定一棵 n 个点的树,树上有些点是黑的,有些点是白的。 你需要删掉一些边,使得不存在黑点和白点连通。 删掉每条边都需要一定的代价,求出最小代价。 【输入数据】 第一行一个整数 n,接下来 n-1 行每行三个整数表示一条边的端点 编号以及删掉的代价。下一行一个整数 m 表示黑点个数,下一行 m 个整数表示黑点编号。下一行一个整数 k 表示白点个数,下一行 k 个 整数表示白点编号。 【输出数据】 一行一个整数表示答案。 【样例输入】 6 1 2 5 2 4 4 2 5 1 1 3 2 3 6 7 1 4 2 5 6 【样例输出】 3 【数据范围】 对于 30%的数据,n<=15。 对于 60%的数据,n<=100。 对于 100%的数据,1<=n<=10^5,点的编号为[1,n]的整数,所有黑点 与白点的编号不会重复,代价为[1,10^9]的整数

分析

树形DP,全场都AC了的题

$f[i][0/1]$分别表示该节点和子树中白/黑节点不联通的最小代价

$f[i][0/1]+=min(f[son][0/1],f[son][1/0]+e[i].w)$

code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#define mem(x,num) memset(x,num,sizeof x)
#define reg(x) for(int i=last[x];i;i=e[i].next)
#define inf 1e9
using namespace std;
inline ll read()
{
	ll f=1,x=0;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
const int maxn=1e5+6;
int n,m1,m2,x,col[maxn],cnt=0,last[maxn];
ll f[maxn][2];
struct edge{int to,next,w;}e[maxn<<1];
void insert(int u,int v,int w){
	e[++cnt]=(edge){v,last[u],w};last[u]=cnt;
	e[++cnt]=(edge){u,last[v],w};last[v]=cnt;
}
void dfs(int x,int fa){
	reg(x){
		if(e[i].to==fa)continue;
		dfs(e[i].to,x);
		f[x][0]+=min(f[e[i].to][0],f[e[i].to][1]+e[i].w);
		f[x][1]+=min(f[e[i].to][1],f[e[i].to][0]+e[i].w);
	}
	if(col[x])f[x][col[x]-1]=inf;
}
int main()
{
	n=read();
	rep(i,1,n-1){
		int u=read(),v=read(),w=read();
		insert(u,v,w);
	}
	m1=read();rep(i,1,m1)x=read(),col[x]=1;
	m2=read();rep(i,1,m2)x=read(),col[x]=2;
	dfs(1,0);
	cout<<min(f[1][0],f[1][1])<<endl;
	return 0;
}
本文可以转载,但必须附上原文链接,否则你会终生找不到妹子!!!欢迎关注我的CSDN: ahyjjr