洛谷P5659 树上的数
题目描述
给定一个大小为 nn 的树，它共有 nn 个结点与 n - 1n−1 条边，结点从 1 \sim n1∼n 编号。初始时每个结点上都有一个 1 \sim n1∼n 的数字，且每个 1 \sim n1∼n 的数字都只在恰好一个结点上出现。
接下来你需要进行恰好 n - 1n−1 次删边操作，每次操作你需要选一条未被删去的边，此时这条边所连接的两个结点上的数字将会交换，然后这条边将被删去。
n - 1n−1 次操作过后，所有的边都将被删去。此时，按数字从小到大的顺序，将数字 1 \sim n1∼n 所在的结点编号依次排列，就得到一个结点编号的排列 P_iPi。现在请你求出，在最优操作方案下能得到的字典序最小的 P_iPi。
如上图，蓝圈中的数字 1 \sim 51∼5 一开始分别在结点②、①、③、⑤、④。按照 (1)(4)(3)(2) 的顺序删去所有边，树变为下图。按数字顺序得到的结点编号排列为①③④②⑤，该排列是所有可能的结果中字典序最小的。
输入格式
本题输入包含多组测试数据。
第一行一个正整数 TT，表示数据组数。
对于每组测试数据：
第一行一个整数 nn，表示树的大小。
第二行 nn 个整数，第 i (1 \leq i \leq n)i(1≤i≤n) 个整数表示数字 ii 初始时所在的结点编号。
接下来 n - 1n−1 行每行两个整数 xx, yy，表示一条连接 xx 号结点与 yy 号结点的边。
输出格式
对于每组测试数据，输出一行共 nn 个用空格隔开的整数，表示最优操作方案下所能得到的字典序最小的 P_iPi。
输入输出样例
输入 #1复制
4 5 2 1 3 5 4 1 3 1 4 2 4 4 5 5 3 4 2 1 5 1 2 2 3 3 4 4 5 5 1 2 5 3 4 1 2 1 3 1 4 1 5 10 1 2 3 4 5 7 8 9 10 6 1 2 1 3 1 4 1 5 5 6 6 7 7 8 8 9 9 10
输出 #1复制
1 3 4 2 5 1 3 5 2 4 2 3 1 4 5 2 3 4 5 6 1 7 8 9 10
说明/提示
【数据范围】
| 测试点编号 | n \leqn≤ | 特殊性质 |
|---|---|---|
| 1 \sim 21∼2 | 10 | 无 |
| 3 \sim 43∼4 | 160 | 树的形态是一条链 |
| 5 \sim 75∼7 | 2000 | 同上 |
| 8 \sim 98∼9 | 160 | 存在度数为 n - 1n−1 的结点 |
| 10 \sim 1210∼12 | 2000 | 同上 |
| 13 \sim 1613∼16 | 160 | 无 |
| 17 \sim 2017∼20 | 2000 | 无 |
对于所有测试点：1 \leq T \leq 101≤T≤10，保证给出的是一个树。
上代码：
#include
#include
using namespace std;
#define rep(i,__l,__r) for(int i=__l,i##_end_=__r;i<=i##_end_;++i)
#define fep(i,__l,__r) for(int i=__l,i##_end_=__r;i>=i##_end_;--i)
#define writc(a,b) fwrit(a),putchar(b)
#define mp(a,b) make_pair(a,b)
#define ft first
#define sd second
#define LL long long
#define ull unsigned long long
#define pii pair
// #define FILEOI
#define cg (c=getchar())
templateinline void qread(T& x){
char c;bool f=0;
while(cg<'0'||'9'inline void qread(T& x,Args&... args){qread(x),qread(args...);}
templateinline T Max(const T x,const T y){return x>y?x:y;}
templateinline T Min(const T x,const T y){return xinline T fab(const T x){return x>0?x:-x;}
inline void getInv(int inv[],const int lim,const int MOD){
inv[0]=inv[1]=1;for(int i=2;i<=lim;++i)inv[i]=1ll*inv[MOD%i]*(MOD-MOD/i)%MOD;
}
templatevoid fwrit(const T x){
if(x<0)return (void)(putchar('-'),fwrit(-x));
if(x>9)fwrit(x/10);putchar(x%10^48);
}
const int MAXN=2000;
struct edge{
int to,nxt;
edge(){}
edge(const int T,const int N):to(T),nxt(N){}
}e[(MAXN<<1)+5];
int tail[MAXN+5],ecnt,siz[MAXN+5];
inline void add_edge(const int u,const int v){
++siz[u],++siz[v];
e[++ecnt]=edge(v,tail[u]);tail[u]=ecnt;
e[++ecnt]=edge(u,tail[v]);tail[v]=ecnt;
}
int N,minp;
int pt[MAXN+5];
int pre[MAXN+5][MAXN+5],nxt[MAXN+5][MAXN+5];
//这条边的 前驱,后继
int rt[MAXN+5][MAXN+5][2];
//点 u 的某一条边的 前根节点、后根节点
int len[MAXN+5][MAXN+5];
//这条边 所在链 的链表长度
inline void init(){
qread(N);
ecnt=0;
rep(i,1,N)qread(pt[i]),tail[i]=siz[i]=0;
rep(i,1,N+1)rep(j,1,N+1)len[i][j]=pre[i][j]=nxt[i][j]=rt[i][j][0]=rt[i][j][1]=0;
int u,v;
rep(i,1,N-1){
qread(u,v);
add_edge(u,v);
pre[u][v]=pre[v][u]=nxt[u][v]=nxt[v][u]=0;
rt[u][v][0]=rt[u][v][1]=v;
rt[v][u][0]=rt[v][u][1]=u;
len[u][v]=len[v][u]=1;
}
}
void findPath(const int u,const int p){//当前节点、前一个节点
int a=rt[u][p][0],b=rt[u][p][1],ta,tb;
//a:来的边的前根 ; b:来的边的后根
if(p==N+1){//这是起点
for(int i=tail[u],v;i;i=e[i].nxt){//从哪条边起飞
v=e[i].to;
ta=rt[u][v][0],tb=rt[u][v][1];
if(ta!=v || (pre[u][p]==tb && len[u][ta]