LOJ10159旅游规划

题目描述

W 市的交通规划出现了重大问题，市政府下定决心在全市各大交通路口安排疏导员来疏导密集的车流。但由于人员不足，W 市市长决定只在最需要安排人员的路口安排人员。

具体来说，W 市的交通网络十分简单，由 n 个交叉路口和 n-1 条街道构成，交叉路口路口编号依次为 0,1,⋯,n−1 。任意一条街道连接两个交叉路口，且任意两个交叉路口间都存在一条路径互相连接。

经过长期调查，结果显示，如果一个交叉路口位于 W 市交通网最长路径上，那么这个路口必定拥挤不堪。所谓最长路径，定义为某条路径 p=(v1​,v2​,v3​,⋯,vk​)，路径经过的路口各不相同，且城市中不存在长度大于 k 的路径，因此最长路径可能不唯一。因此 Ｗ 市市长想知道哪些路口位于城市交通网的最长路径上。

输入格式

第一行一个整数 n；

之后 n-1 行每行两个整数 u,v，表示 u 和 v 的路口间存在着一条街道。

输出格式

输出包括若干行，每行包括一个整数——某个位于最长路径上的路口编号。为了确保解唯一，请将所有最长路径上的路口编号按编号顺序由小到大依次输出。

样例

样例输入

10
0 1
0 2
0 4
0 6
0 7
1 3
2 5
4 8
6 9

样例输出

0
1
2
3
4
5
6
8
9

数据范围与提示

对于全部数据，1≤n≤2×10^5。

_____________________________________________________________________________________________________________

树形动态规划，典型的求最长链，变化就是加了一个求链上的点。

写的过程中没有注意到点的多次加入，所以错了！

_____________________________________________________________________________________________________________

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 const int maxn=2e5+10;
 4 int n;
 5 struct edge
 6 {
 7     int u,v,nxt;
 8 }e[maxn<<1];
 9 int head[maxn],js;
10 void addage(int u,int v)
11 {
12     e[++js].u=u;e[js].v=v;
13     e[js].nxt=head[u];head[u]=js;
14 }
15 int ml[maxn],sl[maxn];
16 int ans=0;
17 void dp(int u,int fa)
18 {
19     for(int i=head[u];i!=-1;i=e[i].nxt)
20     {
21         int v=e[i].v;
22         if(v!=fa)
23         {
24             dp(v,u);
25             if(ml[v]+1>ml[u])
26             {
27                 sl[u]=ml[u];
28                 ml[u]=ml[v]+1;
29             }
30             else if(ml[v]+1>sl[u])
31                 sl[u]=ml[v]+1;
32         }
33     }
34     if(ml[u]+sl[u]>ans)ans=ml[u]+sl[u];
35 }
36 int q[maxn],jss,bz[maxn];
37 void work(int u,int fa,int len)
38 {
39     if(bz[u]==0)
40     {
41         q[jss++]=u;
42         bz[u]=1;
43     }
44     for(int i=head[u];i!=-1;i=e[i].nxt)
45     {
46         int v=e[i].v;
47         if(v!=fa)
48         {
49             if(ml[v]==len-1)work(v,u,len-1);
50         }
51     }
52 }
53 void dfs(int u,int fa)
54 {
55     if(ml[u]+sl[u]==ans)
56     {
57         if(ml[u]!=sl[u])work(u,fa,ml[u]);
58         work(u,fa,sl[u]);
59     }
60     for(int i=head[u];i!=-1;i=e[i].nxt)
61     {
62         int v=e[i].v;
63         if(v!=fa)dfs(v,u);
64     }
65 }
66 int main()
67 {
68     scanf("%d",&n);
69     memset(head,-1,sizeof(head));
70     for(int u,v,i=1;i<n;++i)
71     {
72         scanf("%d%d",&u,&v);
73         addage(u,v);addage(v,u);
74     }
75     dp(0,-1);
76     dfs(0,-1);
77     sort(q,q+jss);
78     for(int i=0;i<jss;++i)printf("%d\n",q[i]);
79     return 0;
80 }