【NOI2019模拟2019.6.29】智慧树（fft单位根性质推导）

Description：

---

1<=n<=8000,m<=57984且m是mo-1的约数

空间限制32MB，时间限制5s

题解：

---

首先思考一下$m=2^{?}$怎么做？

我们知道fft的本质是一个循环卷积，这道题要求$modm$，那么可以利用这个特质而不管溢出，一开始对每一个点的序列进行dft，然后每次转移就是把dft的序列乘起来，最后逆dft回去就好了。

因为每个点一开始的序列只有一个位置有值，所以可以$O(m)$dft，即$A[i]+=w^{i\ast b[x]modm}$。

空间限制非常小，也就是我们不能把dft的序列存下来，考虑到dft后每一位完全独立，所以可以先枚举做哪一位，再在树上dp。

还有一种方法，这是一个树形依赖dp，考虑用树链剖分节省空间，先做重儿子，把重儿子做完的数组的指针给当前点，然后做轻儿子，做完一个就合并，由于轻链+重链条数是log的，所以空间是$O(log\ast m)$的。

然后最后的问题是$m̸=2^{?}$怎么办，此时的m依然是(mo-1)的约数，也就是存在单位复数根，那么fft的性质依然可以用，只是不能用分治做fft，那么不妨看一下式子：
把$a$ dft成$A$
$A[i]={\sum }_{j=0}^{n-1}a[j]\ast w^{ij}$
$A[i]={\sum }_{j=0}^{n-1}a[j]\ast w^{C_{i+j}^2-C_i^2-C_j^2}$

那么可以用一次卷积来完成这个dft的，实际上要做的是逆dft，差不多的。

注意常数优化。

Code：

---

#include<bits/stdc++.h>
#define fo(i, x, y) for(int i = x, B = y; i <= B; i ++)
#define ff(i, x, y) for(int i = x, B = y; i <  B; i ++)
#define fd(i, x, y) for(int i = x, B = y; i >= B; i --)
#define ll long long
#define pp printf
#define hh pp("\n")
using namespace std;

const int mo = 950009857;

ll ksm(ll x, ll y) {
	ll s = 1;
	for(; y; y /= 2, x = x * x % mo)
		if(y & 1) s = s * x % mo;
	return s;
}

const int N = 2e5 + 5;
ll a[N]; int r[N];

void dft(ll *a, int n, int F) {
	ff(i, 0, n) {
		r[i] = r[i / 2] / 2 + (i & 1) * (n / 2);
		if(i < r[i]) swap(a[i], a[r[i]]);
	}
	for(int h = 1; h < n; h *= 2) {
		ll wn = ksm(7, (mo - 1) / 2 / h);
		for(int j = 0; j < n; j += 2 * h) {
			ll b, *l = a + j, *r = a + j + h, w = 1;
			ff(i, 0, h) {
				b = *r * w, *r = (*l - b) % mo, *l = (*l + b) % mo;
				w = w * wn % mo, l ++, r ++;
			}
		}
	}
	if(F == -1) {
		reverse(a + 1, a + n);
		ll v = ksm(n, mo - 2);
		ff(i, 0, n) a[i] = (a[i] + mo) * v % mo;
	}
}
void fft(ll *a, ll *b, int n) {
	dft(a, n, 1); dft(b, n, 1);
	ff(i, 0, n) a[i] = a[i] * b[i] % mo;
	dft(a, n, -1);
 }

int n, m, b[N], x, y;
int fi[N], to[N * 2], nt[N * 2], tot;

void link(int x, int y) {
	nt[++ tot] = fi[x], to[tot] = y, fi[x] = tot;
}

int d[8005], d0, fa[8005];
void dg(int x) {
	d[++ d0] = x;
	for(int i = fi[x]; i; i = nt[i]) if(to[i] != fa[x])
		fa[to[i]] = x, dg(to[i]);
}

int v[N], f[N];
ll w, aw[N];
ll s[N], s2[N * 2], s3[N * 2];

int c[N];

int main() {
	freopen("tree.in", "r", stdin);
	freopen("tree.out", "w", stdout);
	scanf("%d %d", &n, &m);
	w = ksm(7, (mo - 1) / m);
	aw[0] = 1; fo(i, 1, m) aw[i] = aw[i - 1] * w % mo;
	fo(i, 1, n) scanf("%d", &b[i]);
	fo(i, 1, n - 1) {
		scanf("%d %d", &x, &y);
		link(x, y); link(y, x);
	}
	dg(1);
	ff(j, 0, m) {
		fo(i, 1, n) f[i] = aw[c[i]], c[i] = (c[i] + b[i] > m) ? (c[i] + b[i] - m) : (c[i] + b[i]);
		fd(i, d0, 2) {
			s[j] += f[d[i]];
			f[d[i]] ++;
			f[fa[d[i]]] = (ll) f[fa[d[i]]] * f[d[i]] % mo;
		}
		s[j] += f[1];
		s[j] %= mo;
	}
	ll fw = ksm(w, m - 1);
	ff(i, 0, 2 * m) s2[i] = ksm(w, (ll) i * (i - 1) / 2);
	ff(i, 0, m) s[i] = s[i] * ksm(fw, (ll) i * (i - 1) / 2) % mo;
	ff(i, 0, m) s3[m - i] = s[i];
	int tp = 0;
	while(1 << ++ tp <= 3 * m);
	fft(s3, s2, 1 << tp);
	ff(i, 0, m) s[i] = s3[i + m] * ksm(fw, (ll) i * (i - 1) / 2) % mo;
	ff(i, 1, m / 2) swap(s[i], s[m - i]);
	ff(i, 0, m) s[i] = s[i] * ksm(m, mo - 2) % mo;
	
	ff(i, 0, m) pp("%lld ", (s[i] + mo) % mo);
}