这题题意很好理解,就是要找到个最大长度,比赛的时候就在纠结怎么找,一开始想dfs怕超时,后来用并查集乱搞过了,看了标程果然是dfs,但只要2次就可以了,显然。
于是自己又敲了一遍
/* author:jxy lang:C/C++ university:China,Xidian University **If you need to reprint,please indicate the source** */ #include <iostream> #include <cstdio> #include <cstdlib> #include <cstring> #include <queue> #define INF 1E9 using namespace std; vector<int>node[100005]; int d[100005]; void dfs(int u,int f,int dp) { d[u]=dp; for(int i=0;i<node[u].size();i++) if(node[u][i]!=f)dfs(node[u][i],u,dp+1); } int main() { int T,n,m,k,K,i,v,u; scanf("%d",&T); while(T--) { scanf("%d%d",&n,&m); for(i=1;i<=n;i++)node[i].clear(); for(i=1;i<n;i++) { scanf("%d%d",&v,&u); node[u].push_back(v); node[v].push_back(u); } K=1; dfs(K,0,1); for(i=1;i<=n;i++) if(d[i]>d[K])K=i;//直径的一个端点 dfs(K,0,1); for(i=1;i<=n;i++) if(d[i]>d[K])K=i;//另一个端点 K=d[K]; while(m--) { scanf("%d",&k); if(k<=K)printf("%d\n",k-1); else printf("%d\n",K-1+2*(k-K)); } } }
时间: 2024-10-28 22:56:45