萌新刚学OI 对输入数据有疑问
查看原帖
萌新刚学OI 对输入数据有疑问
13550
叶绿素楼主2020/7/24 09:28

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;
}
2020/7/24 09:28
加载中...