#P12355. 「HCOI-R2」巡回演出

    ID: 13514 远端评测题 1000ms 512MiB 尝试: 0 已通过: 0 难度: 6 上传者: 标签>洛谷原创O2优化组合数学Catalan 数生成函数洛谷月赛Prüfer 序列

「HCOI-R2」巡回演出

题目背景

小 A 是 HCOI 国著名的演唱家,她打算开展一场巡回演出。

题目描述

HCOI 国由 nn 个城市构成,可以被抽象为一棵 nn 个节点的有根树,首都是树的根 rr。小 A 打算从 rr 开始,遍历全国进行演出。

具体地,设 xx 是小 A 当前在的位置,记录一个序列 ss

  • 初始时,x=rx=rs=[r]s=[r]
  • 如果 xx 存在没有被访问过的儿子,任意选择一个儿子 yy,在 ss 的最后插入 yy,并前往 yy
  • 如果不存在如上的 yy:若 x=rx=r,演出结束;否则前往 xx 的父亲。

我们称最后得到的 ssTT 的一种遍历方法,选择不同的儿子可以得到多种遍历方法。可以发现遍历方法实际上是 TT 的 DFS 序。

小 A 的助手给出一张收费表 {wn1}\{w_{n-1}\},可以如下计算这次巡回演出的收入 w(T,s)w(T,s)

  • 对于 i[1,n]i\in[1,n],小 A 经过 TTsisimodn+1s_{i}\to s_{i\bmod n+1} 的简单路径上的点(包含 sis_{i},包含 simodn+1s_{i\bmod n+1})。
  • 每次到达一个节点时,若当前是第 ii 次到达,小 A 收获 wiw_i 枚金币作为报酬。
  • w(T,s)w(T,s) 为小 A 收到的金币枚数的总和。

小 A 的助手还给出了本次演出的遍历方法 {sn}\{s_n\},定义一棵树 TT 合法当且仅当 {sn}\{s_n\} 是它的一种遍历方法。你需要求出所有不同的合法 TTw(T,s)w(T,s) 之和对 998244353998244353 取模后的结果。

两棵有根树 TTTT' 不相同当且仅当它们的根节点不同或存在一条在 TT 中出现而 TT' 中未出现的边。

输入格式

本题包含多组数据。

第一行一个正整数 TT 代表数据组数。对于每组测试数据:

第一行一个正整数 nn,表示 HCOI 国的城市数。

第二行 nn 个正整数 s1,s2,,sns_1,s_2,\cdots,s_n,表示小 A 本次演出的遍历方法。

第三行 n1n-1 个非负整数 w1,w2,,wn1w_1,w_2,\cdots,w_{n-1},表示演出的收费表。

输出格式

对于每组测试数据,输出一行一个整数表示所有合法的 TTw(T,s)w(T,s) 之和对 998244353998244353 取模后的结果。

7
2
1 2
1
3
1 2 3
1 2
4
1 2 3 4
1 2 3
5
1 3 5 2 4
1 3 2 1
6
6 2 3 4 1 5
3 2 4 7 4
7
1 3 2 4 5 6 7
12 32 84 27 83 11
8
1 7 3 2 8 4 6 5
11 45 14 19 19 8 10
2
10
42
182
1240
41336
124348
2
9
1 2 3 4 5 6 7 8 9
18 48 72 49 83 59 74 80
12
1 12 2 4 3 6 5 7 8 9 11 10
120 938 283 920 462 748 932 784 178 904 442
787978
522215784
1
4
1 2 3 4
1 0 0
20

提示

【样例解释 #1】

下文用 (r,E)(r,E) 来表示一棵有根树,其中 EE 是树的边集,rr 是树的根。

对于第一组数据,唯一一种合法的 TT(1,{(1,2)})(1,\{(1,2)\})。小 A 经过两条路径 121\to 2212\to 1,经过了 1,21,2 各一遍,故答案为 1+1=21+1=2

对于第二组数据,合法的 TT(1,{(1,2),(2,3)}),(1,{(1,2),(2,3)})(1,\{(1,2),(2,3)\}),(1,\{(1,2),(2,3)\})w(T,s)w(T,s) 都是 55,故答案为 5+5=105+5=10

对于第三组数据,合法的 TT 有 $(1,\{(1,2),(1,3),(1,4)\}),(1,\{(1,2),(1,3),(3,4)\}),(1,\{(1,2),$ $(2,3),(1,4)\}),(1,\{(1,2),(2,3),(2,4)\}),(1,\{(1,2),(2,3),(3,4)\})$,w(T,s)w(T,s) 分别为 9,8,8,9,8,8, 9,89,8,故答案为 9+8+8+9+8=429+8+8+9+8=42

对于第四组数据,一种合法的 TT(1,{(1,3),(2,4),(3,2),(3,5)})(1,\{(1,3),(2,4),(3,2),(3,5)\}),它的收入 w(T,s)=13w(T,s)=13

【数据范围与约定】

本题采用捆绑测试。

子任务编号 n\boldsymbol{n\le} n\boldsymbol{\sum n\le} 特殊性质 得分
11 44 10610^6 A 55
22 88 4040 1515
33 1212 6060 1010
44 5050 200200
55 100100 500500
66 10310^3 5×1035\times 10^3 B
77 55
88 10510^5 2×1052\times 10^5 B
99 1515
1010 5×1055\times10^5 10610^6
  • 特殊性质 A:i[1,n]i\in[1,n]si=is_i=i
  • 特殊性质 B:w1=1w_1=1,对于 i[2,n1]i\in[2,n-1]wi=0w_i=0

对于所有数据,1T1051\le T\le 10^52n5×1052\le n\le 5\times10^5n106\sum n\le 10^60wi<9982443530\le w_i<998244353{sn}\{s_n\} 是一个 1n1\sim n 的排列。