求树的直径裸题。可以用两次dfs或者两次bfs,都一样。
树的直径:从任意一点到距离这一点的最远处的一点,最远处的这一点为树的直径的一个端点,从这个端点开始深搜或者广搜求得距离该端点最远的一个结点,也就是树的直径的另一个端点。
#include <iostream>
#include <queue>
#include <vector>
using namespace std;
typedef struct{
int p;
int level;
int pace=0;
vector<int> graph;//储存边,如果用邻接矩阵表示法开最大值是0分哟
}INTER;
vector<INTER> inter(1);
int main(){
int n,m;
cin>>n>>m;
INTER node;
node.level=0;
node.p=1;
inter.push_back(node);
int maxl=0;
for (int a=2;a<=n;a++){
int t;
scanf ("%d",&t);
inter[t].graph.push_back(a);
INTER temp;
temp.level=inter[t].level+1;
if (maxl<temp.level) maxl=temp.level;
temp.p=a;
temp.graph.push_back(t);
inter.push_back(temp);
}
for (int a=n+1;a<=n+m;a++){//既然后面的表示不涉及是交换机还是电脑,干脆直接把电脑的编号根据交换机数目往后顺移,方便计算
int t;
scanf ("%d",&t);
inter[t].graph.push_back(a);
INTER temp;
temp.level=inter[t].level+1;
if (maxl<temp.level) maxl=temp.level;
temp.p=a;
temp.graph.push_back(t);
inter.push_back(temp);
}
queue<INTER> q;//用队列保存深度最大的结点,相当于从根开始找离根最远的结点,也即求树的直径的第一步,任一结点求距离其最远结点
for (auto it=inter.begin();it!=inter.end();it++){
if ((*it).level==maxl) q.push(*it);
}
long long maxpace=0;//保存最大距离
while (!q.empty()){//对于每一个距离根节点最远的子结点
queue<INTER>que;//便于对值修改,开一个临时队列
que.push(q.front());
vector<int> fl(n+m+1);
while (!que.empty()){//bfs求最大距离,即树的直径
auto t=que.front();
fl[t.p]=1;
for (auto it=inter[t.p].graph.begin();it!=inter[t.p].graph.end();it++){
if (!fl[(*it)]){
inter[(*it)].pace=inter[t.p].pace+1;
if (maxpace<inter[(*it)].pace) maxpace=inter[(*it)].pace;
que.push(inter[(*it)]);
fl[(*it)]=1;
}
}
que.pop();
}
for (auto it=inter.begin();it!=inter.end();it++){
(*it).pace=0;
}
q.pop();
}
printf ("%lld",maxpace);//是不是long long我也没试过,反正用了也不会怎么样
}