1号数据输入:
4 3
1 2 1
1 3 1
1 4 2
输出:5
树的直径是3,那为什么答案不是4呢?
下面贴一下我的代码,就只A了一个点
#include<bits/stdc++.h>
using namespace std;
const int maxn=200010;
int n,m,head[maxn],cnt,num;
long long ans;
struct edge{
int nxt,v;
long long w;
}e[maxn*2];
void addedge(int u,int v,long long w){
e[++cnt].nxt=head[u];
e[cnt].v=v;
e[cnt].w=w;
head[u]=cnt;
}
long long dep[maxn],len;
void dfs1(int rt,int fa,long long w){
dep[rt]=dep[fa]+w;
if(dep[rt]>len){
len=dep[rt];
num=rt;
}
for(int i=head[rt];i;i=e[i].nxt){
int v=e[i].v;
long long w=e[i].w;
if(v!=fa)
dfs1(v,rt,w);
}
}
int road[maxn];
void dfs2(int rt,int fa,long long w){
dep[rt]=dep[fa]+w;
if(dep[rt]>len){
len=dep[rt];
num=rt;
road[rt]=fa;
}
for(int i=head[rt];i;i=e[i].nxt){
int v=e[i].v;
long long w=e[i].w;
if(v!=fa)
dfs2(v,rt,w);
}
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++){
int u,v;
long long t;
scanf("%d%d%lld",&u,&v,&t);
addedge(u,v,t);
addedge(v,u,t);
}
dfs1(1,0,0);
memset(dep,0,sizeof(dep));
len=0;
dfs2(num,0,0);
ans=len;
int i;
for(i=num;(double)dep[road[i]]>=(double)len/2;i=road[i]);
ans+=max(dep[road[i]],len-dep[i]);
printf("%lld",ans);
return 0;
}