旅行计划
题意
有一棵树,从给定的节点出发,遍历所有节点,不要求路径不重叠。要求每一次前往的点与当前点的路径上没有访问过的节点数量最多,若有多个这样的节点,优先取字典序最小的。求遍历顺序。
思路
将出发节点设为树的根节点,dfs一遍,统计根节点到每个叶节点路径上的节点数量。(显然每一次路径的终点都是一个叶节点)
每一次寻找下一个节点,然后暴力更新路径上的节点数量。
本题有个坑,存在只有一个点的情况,注意特判。
代码
#include <bits/stdc++.h>
using namespace std;
namespace StandardIO {
template<typename T>inline void read (T &x) {
x=0;T f=1;char c=getchar();
for (; c<'0'||c>'9'; c=getchar()) if (c=='-') f=-1;
for (; c>='0'&&c<='9'; c=getchar()) x=x*10+c-'0';
x*=f;
}
template<typename T>inline void write (T x) {
if (x<0) putchar('-'),x*=-1;
if (x>=10) write(x/10);
putchar(x%10+'0');
}
}
using namespace StandardIO;
namespace Project {
const int N=50005;
int n,k;
int cnt;
int head[N];
struct node {
int to,next;
} edge[N<<1];
int tot;
int dep[N],leaf[N],vis[N],f[N];
inline bool cmp (int x,int y) {
return (dep[x]==dep[y])?x<y:dep[x]>dep[y];
}
inline void add (int a,int b) {
edge[++cnt].to=b,edge[cnt].next=head[a],head[a]=cnt;
}
void dfs (int now,int fa) {
dep[now]=dep[fa]+1,f[now]=fa;
int tot_son=0;
for (register int i=head[now]; i; i=edge[i].next) {
int to=edge[i].to;
if (to==fa) continue;
++tot_son,dfs(to,now);
}
if (!tot_son) leaf[++tot]=now;
}
inline void MAIN () {
read(n),read(k);
for (register int i=1,x; i<=n-1; ++i) {
read(x);
add(x,i),add(i,x);
}
dfs(k,k);
sort(leaf+1,leaf+tot+1,cmp);
for (register int i=1; i<=tot; ++i) {
int now=leaf[i],ori=now;
dep[now]=0;
while (!vis[now]) {
vis[now]=1,now=f[now],++dep[ori];
if (now==k) break;
}
}
sort(leaf+1,leaf+tot+1,cmp);
write(k),putchar('\n');
if (leaf[1]==0) return;
for (register int i=1; i<=tot; ++i) {
write(leaf[i]),putchar('\n');
}
}
}
int main () {
// freopen(".in","r",stdin);
// freopen(".out","w",stdout);
Project::MAIN();
}