hdu 4607——Park Visit（暑假第一次多校）

树的直径

求出树的直径后分类讨论。

#include<iostream>#include<cstdio>#include<cstring>#include<queue>using namespace std;#define maxn 200005int head[maxn];int v[maxn],next[maxn],cnt;int n,m,r;void add(int x,int y){v[cnt]=y;next[cnt]=head[x];head[x]=cnt++;v[cnt]=x;next[cnt]=head[y];head[y]=cnt++;}int bfs(int x){int dis[maxn];int end;memset(dis,-1,sizeof(dis));queue<int> q;q.push(x);dis[x]=1;while(!q.empty()){int u=q.front();q.pop();end=u;r=dis[u];for(int i=head[u];i!=-1;i=next[i])if(dis[v[i]]==-1){dis[v[i]]=dis[u]+1;q.push(v[i]);}}return end;}int main(){int t;int x,y;scanf("%d",&t);while(t--){scanf("%d%d",&n,&m);cnt=0;memset(head,-1,sizeof(head));for(int i=0;i<n-1;i++){scanf("%d%d",&x,&y);add(x,y);}bfs(bfs(1));int tmp;while(m--){scanf("%d",&tmp);if(tmp<=r)printf("%d\n",tmp-1);elseprintf("%d\n",r-1+(tmp-r)*2);}}return 0;}