牛客手速月赛47 F(树学的不是很到位)

题目
题意: 给定n个点的有向树,给定m个点和起点k。至多以k为起点向其他顶点连一条边,使得k到这m个点的最短距离之和最小。
思路: 赛时思路求LCA,然后如果k的深度>=LCA的深度,必须往LCA连,否则不能到达所有点;反之,k本身就可以到达所有点,我是写的连接直接儿子的重儿子,但是不对,因为有可能有长长的链,总之欠考虑了。
看了题解的思路。如果k本身可以到达所有点,需要枚举向哪个点连边,贡献的改变类似换根dp的写法。这样,k本身不能到达所有点的情况也可以用此方法实现,都是枚举。更普遍地,可以发现要么以1为根、要么以k为根,这两种情况。可以在建树时不向k连边,这样就将两个树分开,编程容易实现,之后分情况讨论即可。

  1. k = 1或者m个点都在以k为根的子树内,以k为根的子树内dfs即可,贡献的改变量为 (d[x] - d[k] - 1) * sz[x]. (深度差-1)*子树大小,因为子树中所有点(属于那m个点的)都会减少深度差-1的长度。
  2. k本身不能到达m个点,也是dfs,在以1为根的子树内dfs。额外花费是最短路 + 在1中m个点的数量,因为到达任何一点的深度+1,如果某个点的sz[]达不到sz[1],不用考虑。
  3. 题解的路径和写法很妙,val[x]:表示到以x为根的所有m个点的距离和。val[cur] += val[son] + sz[son],当前节点的距离和是儿子节点的距离和 + 子树的大小。因为子树中每个点的深度会+1的。递归求解。其实相当于起点到所有点的看一下深度差,但是这样写说明对递归理解的很到位。
    时间复杂度: O(n)
    代码:
#include<bits/stdc++.h>
using namespace std;
#define mem(a,x) memset(a,x,sizeof(a))
typedef long long ll;
typedef pair<int,int> PII;
const int N = 2e5+10;
int n,m,k,T;
int h[N],e[N],ne[N],idx = 0;
void add(int a,int b)
{
	e[idx] = b,ne[idx] = h[a],h[a] = idx++;
}
PII ee[N];
int dist[N];
int q[N];
int sz[N];
int cnt[N];
int d[N]; //深度
ll val[N];
void dfs(int cur)
{
	if(cnt[cur]) sz[cur] = 1;
	for(int i=h[cur];~i;i=ne[i])
	{
		int j = e[i];
		d[j] = d[cur] + 1;
		dfs(j);
		sz[cur] += sz[j];
		val[cur] += val[j] + sz[j];
	}
}
ll dfs2(int cur)
{
	if(sz[cur] != sz[1]) return 1e18;
	ll ans = val[cur];
	for(int i=h[cur];~i;i=ne[i])
	{
		int j = e[i];
		ans = min(ans,dfs2(j));
	}
	return ans;
}
ll dfs3(int cur)
{
	ll ans = 1ll*(d[cur] - d[k] - 1) * sz[cur];
	for(int i=h[cur];~i;i=ne[i])
	{
		int j = e[i];
		ans = min(ans,dfs3(j));
	}
	return ans;
}
void solve()
{
	cin>>n; mem(h,-1); mem(dist,0x3f);
	for(int i=0;i<n-1;++i)
	{
		cin>>ee[i].first>>ee[i].second;
	}
	cin>>m>>k;
	for(int i=0;i<n-1;++i)
	{
		if(ee[i].second != k)
		{
			add(ee[i].first,ee[i].second);
		}
	}
	for(int i=0;i<m;++i) cin>>q[i],cnt[q[i]]++;
	dfs(1);
	if(k != 1) dfs(k);
	ll ans;
	if(sz[1] == 0 || k == 1) //所有点都在以k为根的子树中
	{
		ans = val[k] - max(0ll,dfs3(k));
	}
	else
	{
		ans = val[k] + sz[1] + dfs2(1);
	}
	cout<<ans;
}
int main(void)
{
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	solve();
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值