刷题记录:牛客NC24953[USACO 2008 Jan G]Cell Phone Network

传送门:牛客

题目描述:

John想让他的所有牛用上手机以便相互交流,他需要建立几座信号塔在N块草地中。已知与信号塔相邻的草地
能收到信号。给你N-1个草地(A,B)的相邻关系,问:最少需要建多少个信号塔能实现所有草地都有信号。
输入:
5
1 3
5 2
4 3
3 5
输出:
2

经典的树形dp的题目.与这道战略游戏类似

做这道之前可以先去做那道战略游戏

主要思路:

  1. 首先这道题与那道战略游戏最大的不同就是那道题是覆盖树的边,而这道题是覆盖树的结点.注意,这两个问题是不一样的,比如这样的一颗树
1->2->3->4
如果只是为了覆盖结点的话我们可以选择2,3或者1,4,但是如果覆盖边的话我们就不能选择1,4这种情况了
  1. 所以我们需要使用 d p [ i ] [ 0 / 1 / 2 ] dp[i][0/1/2] dp[i][0/1/2]来记录三种状态(比那道题多一种状态)
    d p [ i ] [ 0 ] dp[i][0] dp[i][0]代表当前结点放信号塔
    d p [ i ] [ 1 ] dp[i][1] dp[i][1]代表当前结点不放信号塔,但是它的儿子放信号塔
    d p [ i ] [ 2 ] dp[i][2] dp[i][2]代表当前结点不放信号塔,但是它的父亲放信号塔,这种情况类似于我之前局的栗子

所以我们可以写出以下转移:

d p [ u ] [ 0 ] + = m i n ( d p [ v ] [ 0 ] , d p [ v ] [ 1 ] , d p [ v ] [ 2 ] ) dp[u][0]+=min(dp[v][0],dp[v][1],dp[v][2]) dp[u][0]+=min(dp[v][0],dp[v][1],dp[v][2])

当前结点放信号塔,那么无论儿子是什么状态都是可以的

d p [ u ] [ 2 ] + = m i n ( d p [ v ] [ 0 ] , d p [ v ] [ 1 ] ) dp[u][2]+=min(dp[v][0],dp[v][1]) dp[u][2]+=min(dp[v][0],dp[v][1])

当前结点的父亲放信号塔,自己没有信号塔,所以儿子不能被自己管,但是儿子可以是除被自己管的其他状态

对于dp[u][1]的状态:

因为当前结点需要被自己的一个儿子管就行了,但是我们有很多个儿子,也就是说我们需要选出一个最佳的儿子.那么对于选择随意的一个儿子,我们有以下贡献:

d p [ v j ] [ 0 ] + dp[v_j][0]+ dp[vj][0]+ ∑ m i n ( d p [ v i ] [ 0 ] , d p [ v i ] [ 1 ] ) \sum\limits{min(dp[v_i][0],dp[v_i][1])} min(dp[vi][0],dp[vi][1]) i不等于j

那么我们随意的选择两个 j j j,比较一下他们的关系,化简一下我们的式子,实际上就是比较

d p [ v j ] [ 0 ] − m i n ( d p [ v j ] [ 0 ] , d p [ v j ] [ 1 ] ) dp[v_j][0]-min(dp[v_j][0],dp[v_j][1]) dp[vj][0]min(dp[vj][0],dp[vj][1])的值



下面是具体的代码部分:
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
#define ll_maxn 0x3f3f3f3f3f3f3f3f
const double eps=1e-8;
int n;int dp[10010][3];
vector<int>edge[maxn];
void dfs(int u,int per_u) {
	int m_son=0;dp[u][0]=1;
	for(int i=0;i<edge[u].size();i++) {
		int v=edge[u][i];
		if(v==per_u) continue;
		dfs(v,u);
		dp[u][0]+=min(dp[v][1],min(dp[v][0],dp[v][2]));
		dp[u][2]+=min(dp[v][1],dp[v][0]);
		if(dp[v][0]-min(dp[v][0],dp[v][1])<dp[m_son][0]-min(dp[m_son][0],dp[m_son][1])) {
			m_son=v;
		}
	}
	dp[u][1]=dp[m_son][0];
	for(int i=0;i<edge[u].size();i++) {
		int v=edge[u][i];
		if(v!=m_son&&v!=per_u) dp[u][1]+=min(dp[v][1],dp[v][0]);
	}
	return ;
}
int main() {
	n=read();int u,v;
	for(int i=1;i<=n-1;i++) {
		u=read();v=read();
		edge[u].push_back(v);
		edge[v].push_back(u);
	}
	dp[0][0]=inf;
	dfs(1,0);
	cout<<min(dp[1][0],dp[1][1])<<endl;
	return 0;
}

你可能感兴趣的:(c++算法,算法,动态规划,c++,树形dp)