【tyvj1520】树的直径(dfs||树形dp)

题目:

我是超链接

题解:

题目给了一种dfs的做法,但是有负边权的时候不能用

还有一种dp的做法:f[i]g[i]分别表示i点到它的子树中的点距离的最大值和次大值,答案就是max(f[i]+g[i])哎为什么不会加重呢?因为每一条边只遍历一遍,如果是链状则只有f[i]的值没有g[i]的值

代码:

dfs:

#include <cstdio>
#define N 50000
using namespace std;
int tot,nxt[N],point[N],v[N],c[N],ans,ansp;
void addline(int x,int y,int cap)
{
	++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=cap;
	++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=cap;
}
void dfs(int x,int fa,int dis)
{
	if (dis>ans) ans=dis,ansp=x;
	for (int i=point[x];i;i=nxt[i])
	  if (v[i]!=fa) dfs(v[i],x,dis+c[i]);
}
void askans(int x,int fa,int dis)
{
	if (dis>ans) ans=dis;
	for (int i=point[x];i;i=nxt[i])
	  if (v[i]!=fa) dfs(v[i],x,dis+c[i]);
}
int main()
{
	int n,i;
	scanf("%d",&n);
	for (i=1;i<n;i++)
	{
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		addline(a,b,c);
	}
	ans=0; ansp=1; dfs(1,0,0);
    ans=0; askans(ansp,0,0);
    printf("%d",ans);
}

dp:

#include <cstdio>
#include <iostream>
#define N 50000
using namespace std;
int tot,nxt[N],point[N],v[N],c[N],ans,f[N],g[N],n,i;
void addline(int x,int y,int cc)
{
	++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=cc;
	++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=cc;
}
void treedp(int x,int fa)
{
	for (int i=point[x];i;i=nxt[i])
	  if (v[i]!=fa)
	  {
	  	treedp(v[i],x);
	  	if (f[v[i]]+c[i]>f[x])
	  	{
	  		g[x]=f[x];
	  		f[x]=f[v[i]]+c[i];
		}
		else g[x]=max(g[x],f[v[i]]+c[i]);
	  }
	ans=max(ans,f[x]+g[x]);
}
int main()
{
	scanf("%d",&n);
	for (i=1;i<n;i++)
	{
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		addline(a,b,c); 
	}
	treedp(1,0);
	printf("%d",ans);
}


  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值