Description
小Q最近学习了一些图论知识。根据课本,有如下定义。树:无回路且连通的无向图,每条边都有正整数的权值来表示其长度。如果一棵树有N个节点,可以证明其有且仅有N-1
条边。 路径:一棵树上,任意两个节点之间最多有一条简单路径。我们用 dis(a,b)
表示点a和点b的路径上各边长度之和。称dis(a,b)为a、b两个节点间的距离。
直径:一棵树上,最长的路径为树的直径。树的直径可能不是唯一的。
现在小Q想知道,对于给定的一棵树,其直径的长度是多少,以及有多少条边满足所有的直径都经过该边。
Input
第一行包含一个整数N,表示节点数。 接下来N-1行,每行三个整数a, b, c ,表示点 a和点b之间有一条长度为c 的无向边。
Output
共两行。第一行一个整数,表示直径的长度。第二行一个整数,表示被所有 直径经过的边的数量。
Sample Input
6
3 1 1000
1 4 10
4 2 100
4 5 50
4 6 100
Sample Output
1110
2
HINT
【样例说明】
直径共有两条,3 到2的路径和3到6的路径。这两条直径都经过边(3, 1)和边(1, 4)。 HINT
对于100%的测试数据:2≤N≤200000,所有点的编号都在1..N的范围内,
边的权值≤10^9。
题解
第一问不用说了吧。。
关于第二问,有两个性质
1:满足答案的边都在一条直径上(因为每条直径上面都有嘛)
2:这些边一定是连续的一条链,这个yy一下就知道了的。。
如果直径上有一个点,这个点能引出两条或以上的相同长度且最长的链,那么这个点之后的所有点都不可能成为答案
于是我们可以跑两遍dfs,同时记录答案
然后就没了。。
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
struct node
{
int x,y,next,other;LL c;
}a[410000];int len,last[210000];
void ins(int x,int y,LL c)
{
int k1,k2;
k1=++len;
a[len].x=x;a[len].y=y;a[len].c=c;
a[len].next=last[x];last[x]=len;
k2=++len;
a[len].x=y;a[len].y=x;a[len].c=c;
a[len].next=last[y];last[y]=len;
a[k1].other=k2;a[k2].other=k1;
}
int G,lx,ly;
LL d[210000];
int n;
void getmx(int x,int fa)
{
if(d[x]>d[G])G=x;
for(int k=last[x];k;k=a[k].next)if(a[k].y!=fa)d[a[k].y]=d[x]+a[k].c,getmx(a[k].y,x);
}
int pre[210000],nxt[210000],tmp[210000];
bool bk;
void getpath(int x,int ed,int fa)
{
if(x==ed){bk=true;return ;}
for(int k=last[x];k;k=a[k].next)
{
int y=a[k].y;
if(y!=fa)
{
pre[y]=x;tmp[y]=k;
getpath(y,ed,x);
}
if(bk)return;
}
}
bool vis[210000];
LL f[210000];
void dfs(int x,int fa)
{
f[x]=0;bool bk=false;
for(int k=last[x];k;k=a[k].next)
{
int y=a[k].y;
if(y!=fa)
{
dfs(y,x);LL sum=f[y]+a[k].c;
if(f[x]>sum)continue;
if(f[x]<sum){bk=false;f[x]=sum;}
else bk=true;
}
}
if(bk==true)vis[x]=true;//这里开始有两个相等的最长
}
bool v[210000];
int main()
{
scanf("%d",&n);
for(int i=1;i<n;i++)
{
int x,y;LL c;scanf("%d%d%lld",&x,&y,&c);
ins(x,y,c);
}
G=0;getmx(1,0);
lx=G;d[lx]=0;G=0;
getmx(lx,0);
ly=G;
printf("%lld\n",d[ly]);
bk=false;getpath(lx,ly,0);
memset(v,false,sizeof(v));
int u=ly;
while(u!=lx)nxt[pre[u]]=u,u=pre[u];
memset(vis,false,sizeof(vis));
dfs(lx,0);
u=lx;
while(u!=ly)
{
if(!vis[u])v[u]=true;
else {v[u]=true;break;}
u=nxt[u];
}
if(u==ly && !vis[u])v[u]=true;
memset(vis,false,sizeof(vis));
dfs(ly,0);
u=ly;
int ans=0;
while(u!=lx)
{
if(!vis[u])
{
if(v[u])ans++;
}
else
{
if(v[u])ans++;
break;
}
u=pre[u];
}
if(u==lx && !vis[u] && v[u])ans++;
printf("%d\n",ans-1);
return 0;
}