试题编号: 201503-4
试题名称: 网络延时
时间限制: 1.0s
内存限制: 256.0MB
求树的直径
两次宽搜或两次深搜
#include<iostream>
#include<vector>
#include<cstring>
using namespace std;
const int maxn=20005;
vector< vector<int> > G(maxn);
bool visit[maxn];
int d[maxn];
void dfs(int s)
{
visit[s]=true;
for(int i=0;i<G[s].size();++i)
{
if(!visit[G[s][i]])
{
d[G[s][i]]=d[s]+1;
dfs(G[s][i]);
}
}
}
int main()
{
int n,m;
cin>>n>>m;
for(int i=2;i<=n;++i)
{
int temp;
cin>>temp;
G[i].push_back(temp);
G[temp].push_back(i);
}
for(int i=n+1;i<=m+n;++i)
{
int temp;
cin>>temp;
G[i].push_back(temp);
G[temp].push_back(i);
}
memset(visit,false,sizeof(visit));
d[1]=0;
dfs(1);
int max=0;
int u=-1;
for(int i=1;i<=m+n;++i)
{
if(max<d[i])
{
max=d[i];
u=i;
}
}
memset(visit,false,sizeof(visit));
d[u]=0;
dfs(u);
max=0;
for(int i=1;i<=m+n;++i)
{
if(max<d[i])
{
max=d[i];
}
}
cout<<max;
}
#include<iostream>
#include<vector>
#include<cstring>
#include<queue>
using namespace std;
const int maxn=20005;
vector <vector<int> > G(maxn);
int visit[maxn];
int d[maxn];
void bfs(int s)
{
queue<int> Q;
d[s]=0;
Q.push(s);
while(!Q.empty())
{
int now=Q.front();
Q.pop();
visit[now]=true;
for(int i=0;i<G[now].size();++i)
{
int temp=G[now][i];
if(!visit[temp])
{
d[G[now][i]]=d[now]+1;
Q.push(G[now][i]);
}
}
}
}
int main()
{
int n,m;
cin>>n>>m;
for(int i=2;i<=n;++i)
{
int temp;
cin>>temp;
G[i].push_back(temp);
G[temp].push_back(i);
}
for(int i=n+1;i<=m+n;++i)
{
int temp;
cin>>temp;
G[i].push_back(temp);
G[temp].push_back(i);
}
memset(visit,false,sizeof(visit));
bfs(1);
int max=0;
int u=-1;
for(int i=1;i<=m+n;++i)
{
if(max<d[i])
{
max=d[i];
u=i;
}
}
memset(visit,false,sizeof(visit));
bfs(u);
max=0;
for(int i=1;i<=m+n;++i)
{
if(max<d[i])
{
max=d[i];
}
}
cout<<max;
}