P5538 【XR-3】Namid[A]me

题目描述

小 X 给了你一棵 $n$ 个点的树,点有点权。 你需要求出下列式子模 $786433$ 的值: $\sum_{1\leq u\leq v\leq n}f(u,v)^{f(u,v)}$ 其中 $f(u,v)$ 表示 $u$ 到 $v$ 的最短路径上所有点的点权按位与在一起之后的值。 提示:为了方便你的计算,这里我们认为 $0^0=0$。另外,$786433$ 是一个质数,同时也是一个不常用的 NTT 模数,它的原根为 $10$,如果你不知道什么是 NTT 或者不知道什么是原根,你可以忽略这个提示。

输入格式

第一行一个正整数 $n$,表示树的点数。 第二行 $n$ 个正整数 $a_{1\dots n}$,其中 $a_i$ 表示编号为 $i$ 的点的点权。 接下来 $n-1$ 行,每行 $2$ 个正整数 $u,v$,表示编号为 $u$ 和编号为 $v$ 的点之间有一条边。 **数据范围:** - $2 \le n \le 2 \times 10^5$。 - 对于所有满足 $1\le i \le n$ 的 $i$ 都有 $1 \le a_i < 2^{30}$。 - $1 \le u,v \le n, u \ne v$。 - 设 $d$ 为树中叶子(度数为 $1$ 的点)的个数,数据保证 $4\le n \cdot d \le 3 \times 10 ^ 6$。

输出格式

一行一个整数,表示答案对 $786433$ 取模后的值。