标签:mem const 取出 i++ 编号 添加 sizeof bfs continue
#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<cstring>
using namespace std;
const int N=1000100;
int head[N],net[N],ans,ed[N],ver[N],tot,d[N];
bool v[N]={false};
void add(int x,int y,int z)
{
ver[++tot]=y,ed[tot]=z,net[tot]=head[x],head[x]=tot;
}
void dp(int x)
{
v[x]=1;
for(int i=head[x];i;i=net[i])
{
int y=ver[i];
if(v[y])continue;//判断是不是访问过这个节点
dp(y); //向下寻找子节点
ans=max(ans,d[x]+d[y]+ed[i]);//枚举从x节点出发的所有边,找到一个最远的路径,
d[x]=max(d[x],d[y]+ed[i]);//经过枚举后,d[x]就不一定是最长的了。要更新一下
}
}
int main()
{
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
int x,y,z;
cin>>x>>y>>z;//创建无向图
add(x,y,z);
add(y,x,z);
}
dp(1);
printf("%lld",ans);//处处最长路径
}
#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
#include<stdio.h>
using namespace std;
const int N=1010000;
int head[N],ver[N],ed[N],net[N],tot;
int dist[N],v[N],maxd;
int n,p,q,d;
void add(int x,int y,int z)
{
ver[++tot]=y,ed[tot]=z,net[tot]=head[x],head[x]=tot;
}
int bfs(int u)
{
queue<int >q;
while(q.size())q.pop();//把队排空
memset(v,0,sizeof(v));
memset(dist,0,sizeof(dist));
q.push(u);//先把这个节点添加进去
int x,max_num=0;
while(!q.empty()){
x=q.front();q.pop();//取出队顶元素 然后抛出
v[x]=1;//已经访问过了置1
for(int i=head[x];i;i=net[i])//遍历这个节点及其子树
{
int y=ver[i];//下一个节点
if(v[y])continue;
v[y]=1;//没访问过置一
dist[y]=dist[x]+ed[i];//从上往下走,累加
if(dist[y]>maxd){//更新最大值和节点号
maxd=dist[y];
max_num=y;
}
q.push(y);//每个新的节点都要入队
}
}
return max_num;//最远节点的编号
}
int main()
{
cin>>n;
for(int i=1;i<n;i++)
{
cin>>p>>q>>d;//无向图 两边都加
add(p,q,d);
add(q,p,d);
}
int u=bfs(1);//从第一个节点开始选
int s=bfs(u);//第一个节点的最远节点的最远节点
cout<<maxd<<endl;//输出直径
return 0;
}
标签:mem const 取出 i++ 编号 添加 sizeof bfs continue
原文地址:https://www.cnblogs.com/arbor-one/p/12773504.html