問題一覧 > 通常問題

No.2598 Kadomatsu on Tree

レベル : / 実行時間制限 : 1ケース 2.000秒 / メモリ制限 : 512 MB / 標準ジャッジ問題
タグ : / 解いたユーザー数 25
作問者 : hirayuu_ychirayuu_yc / テスター : 👑 AngrySadEightAngrySadEight
2 ProblemId : 10457 / 自分の提出
問題文最終更新日: 2024-01-01 09:16:39

問題文

長さ $n$ の整数列 $a=(a_1,a_2,\dots,a_n)$ について、門松度を以下の条件を満たす $2$ 以上 $n$ 未満の $i$ の個数と定義します。

  • 「 $a_i< a_{i+1}$ かつ $a_i< a_{i-1}$」または「 $a_i> a_{i+1}$ かつ $a_i> a_{i-1}$」の少なくとも一方が成り立つ

ただし、長さ $2$ 以下の任意の整数列の門松度は $0$ であるとします。


$N$ 頂点 $N-1$ 辺の単純連結無向グラフがあります。頂点には $1,2,\dots,N$ と番号がついており、$i$ 番目の辺は頂点 $U_i$ と頂点 $V_i$ を双方向に結びます。

また、長さ $N$ の整数列 $A$ が与えられます。

$f(i,j)$ を、以下の条件をすべて満たす長さ $n$ の唯一の整数列 $p=(p_1,p_2,\dots,p_n)$ から $q=(A_{p_1},A_{p_2},\dots,A_{p_n})$ を得たときの、$q$ の門松度とします。

  • $p_1=i$
  • $p_n=j$
  • $1\leq p_k\leq N(1\leq k\leq n)$
  • $p$ のすべての要素は相違なる
  • $1$ 以上 $n$ 未満のすべての $k$ について、頂点 $p_k$ と頂点 $p_{k+1}$ は直接辺で結ばれている

$\sum_{i=1}^{N}\sum_{j=i}^{N}f(i,j)$ を $998244353$ で割った余りを求めてください。

入力

$N$
$U_1\ V_1$
$U_2\ V_2$
$\vdots$
$U_{N-1}\ V_{N-1}$
$A_1\ A_2\ \dots\ A_N$
  • $3\leq N\leq 2\times 10^5$
  • $1\leq U_i,V_i\leq N(1\leq i\leq N-1)$
  • 与えられるグラフは連結である。
  • $1\leq A_i\leq 10^9$

出力

$1$ 行で、答えを出力してください。

サンプル

サンプル1
入力
5
1 3
2 1
4 1
4 5
2 1 3 1 2
出力
5

$f(i,j)$ が $1$ 以上であるものを以下に示します。

  • $f(1,5)=1\dots p=(1,4,5),q=(2,1,2)$ であり、$q$ の門松度は $1$
  • $f(2,4)=1\dots p=(2,1,4),q=(1,2,1)$ であり、$q$ の門松度は $1$
  • $f(2,5)=2\dots p=(2,1,4,5),q=(1,2,1,2)$ であり、$q$ の門松度は $2$
  • $f(3,5)=1\dots p=(3,1,4,5),q=(3,2,1,2)$ であり、$q$ の門松度は $1$

よって、答えは $5$ です。

サンプル2
入力
6
1 2
2 3
1 5
5 6
4 5
1 1 1 1 1 1
出力
0
サンプル3
入力
10
6 8
6 1
5 10
5 2
6 4
2 4
5 9
8 7
3 9
461924924 987491630 891274602 524053807 460183912 97638725 932784974 244569286 453506405 44117081
出力
50

提出するには、Twitter 、GitHub、 Googleもしくは右上の雲マークをクリックしてアカウントを作成してください。