TopCoder

Cheng0928
NLOJ 管理員,題目若有誤請至 Discord 私訊我!

User's AC Ratio

100.0% (2/2)

Submission's AC Ratio

73.9% (17/23)

Tags

Description

263d80a0-d614-4f46-bb58-1de02faf29fa
在遙遠的魔法王國裡,有一棵古老的智慧之樹,名為「知識之樹」。樹上每個節點都刻有一個神秘數字,象徵不同的魔法符文力量。傳說,樹上隱藏著多把魔法金鑰,只有按照正確的魔法序列才能開啟王國最神秘的寶藏。

樹共有 $n$ 個節點,節點編號為 $1$ 到 $n$,每個節點 $i$ 上的數字為 $val_i$。王國派出 $p$ 位魔法守衛,他們各自站在樹上的某個節點,第 $i$ 位守衛持有的金鑰位於節點 $id_i$,不會有兩位守衛站在同個節點上。

樹由 $n - 1$ 條邊所組成,第 $i$ 條邊連接兩個節點 $u_i,\ v_i$,使整棵樹保持連通,並且保證對於所有節點的子節點的 $val_i$ 皆不相同。

每把金鑰對應一條金鑰序列,即從根節點 $1$ 到金鑰所在節點沿途經過節點的數字組成的序列。例如,如果金鑰位於節點 $5$,根節點到該節點經過 $1 \rightarrow 2 \rightarrow 5$,對應金鑰序列為 $[val_1, val_2, val_5]$。

王國法典規定:
1. 將所有金鑰序列按照字典序排列。
2. 第 $k$ 小的金鑰序列就是打開寶藏的密碼序列。

請你輸出這組密碼序列。

以 $5$ 個節點,以下樣子的樹為例:
image
若 $val = [7,\ 4,\ 2,\ 5,\ 10]$ 且有 $3$ 位守衛分別站在編號 $1,\ 4,\ 5$ 的節點上。
第一位守衛的金鑰序列為 $[7]$,
第二位守衛的金鑰序列為 $[7,\ 5]$,
第三位守衛的金鑰序列為 $[7,\ 10]$,
按照字典序由小到大排,第一小的是第一位守衛的金鑰序列,第二小的是第二位守衛的金鑰序列,第三小的是第三位守衛的金鑰序列,
所以若 $k = 2$,則密碼序列為 $[7,\ 5]$。

字典序定義:
給定兩個序列 $A = [a_1, a_2, \dots, a_p]$ 與 $B = [b_1, b_2, \dots, b_q]$,序列 $A$ 小於 $B$(記作 $A < B$),如果存在最小的索引 $i$,使得 $a_i \ne b_i$,且 $a_i < b_i$。
如果一個序列是另一個序列的前綴,則較短的序列視為較小。例如 $[1,2] < [1,2,3]$。


對於所有測試資料:
$1 \le p \le n \le 10$$6$
$1 \le k \le p$
$1 \le u_i,\ v_i \le n$
$1 \le id_i \le n$
$1 \le val_i \le 10$$9$
保證根節點為 $1$,且對於所有節點的子節點的 $val_i$ 皆不相同。
保證所有 $id_i$ 皆不相同。

評分說明:
每筆測試資料執行時間限制為 7 秒,依正確通過測資筆數給分,其中:
第 1 子題組 30 分:$n \le 100$。
第 2 子題組 70 分:無額外限制。

Input Format

共輸入 $3 + (n - 1)$ 行,
第一行輸入三個數字 $n,\ p,\ k$,
第二行輸入 $p$ 個數字,第 $i$ 個數字為 $id_i$,
第三行輸入 $n$ 個數字,第 $i$ 個數字為 $val_i$,
接下來有 $n - 1$ 行,
第 $i$ 行有兩個數字 $u_i,\ v_i$,代表樹上編號為 $u_i$ 的節點與編號為 $v_i$ 的節點有一條邊。

Output Format

若密碼序列的長度為 $L$,則輸出 $L$ 個數字並以空白隔開,第 $i$ 個數字為密碼序列的第 $i$ 項。

Sample Input 1

5 3 2
1 4 5
7 4 2 5 10
1 2
1 3
1 4
1 5

Sample Output 1

7 5

Sample Input 2

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

Sample Output 2

6 4 7

Hints

Problem Source

Subtasks

No. Testdata Range Score

Testdata and Limits

No. Time Limit (ms) Memory Limit (VSS, KiB) Output Limit (KiB) Subtasks
0 7000 1048576 65536
1 7000 1048576 65536
2 7000 1048576 65536
3 7000 1048576 65536
4 7000 1048576 65536
5 7000 1048576 65536
6 7000 1048576 65536
7 7000 1048576 65536
8 7000 1048576 65536
9 7000 1048576 65536
10 7000 1048576 65536
11 7000 1048576 65536
12 7000 1048576 65536
13 7000 1048576 65536
14 7000 1048576 65536
15 7000 1048576 65536
16 7000 1048576 65536
17 7000 1048576 65536
18 7000 1048576 65536
19 7000 1048576 65536
20 7000 1048576 65536
21 7000 1048576 65536