[BZOJ3451]normal 点分治,NTT

好久没更博了,咕咕咕。

BZOJ3451权限题,上darkbzoj交吧。

一句话题意,求随机点分治的期望复杂度。

考虑计算每个点对的贡献:如果一个点在点分树上是另一个点的祖先,那么这个点对另一个点的贡献就是1,这样的话,这个点就必须是这两个点之间的链上的点中在点分树上深度最浅的点,由于链上每个点成为点分树上最浅的点的概率都是相等的,所以这个点对对最终的期望的贡献就是\(\frac{1}{dis(i, j) + 1}\),这里的\(dis(i, j)\)习惯上认为是边的条数,\(+1\)就变成了点的个数。现在我们要求的就是\(\sum \limits _{i = 1} ^{n} \sum \limits _{j = 1} ^{n} \frac {1} {dis(i, j) + 1}\)。

考虑怎样在树上统计,明显需要点分治,每次统计从当前分治中心出发的所有长度的路径条数,跟先前统计过的子树合并一下就好了。观察合并的式子,设\(f[i]\)表示这次合并统计的长度为\(i\)的路径条数,\(p[i]\)表示以前统计过的子树中长度为\(i\)的路径条数,\(q[i]\)表示这次统计的长度为\(i\)的路径条数,显然有\(f[k] = \sum \limits _{i + j = k} p[i] * q[j]\),明显是一个卷积的形式,可以FFT,但是发现\(f\)数组的值肯定不会超过NTT模数,于是直接NTT。时间复杂度\(O(n (\log n) ^ 2)\)。

实现的时候要注意,每次从分支中心出发统计时,必须先把子树按深度或大小排一遍序,从小往大处理,不然一个扫把图就能把你卡成\(n ^ 2 \log n\),可以自己画图手玩。

#include <cstdio>
#include <cctype>
#include <vector>
#include <cstring>
#include <algorithm>
#define R register
#define I inline
#define D double
#define L long long
#define B 1000000
using namespace std;
const int N = 32777, P = 998244353, G = 3, H = 332748118;
char buf[B], *p1, *p2;
I char gc() { return p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, B, stdin), p1 == p2) ? EOF : *p1++; }
I int rd() {
R int f = 0;
R char c = gc();
while (c < 48 || c > 57)
c = gc();
while (c > 47 && c < 58)
f = f * 10 + (c ^ 48), c = gc();
return f;
}
int s[N], t[N], v[N], p[N], q[N], a[N], b[N], d[N], c[N], h[N], u, r, S, e, F, M;
D o;
vector <int> g[N];
I int max(int x, int y) { return x > y ? x : y; }
I void swp(int &x, int &y) { x ^= y, y ^= x, x ^= y; }
I int cmp(int x, int y) { return t[x] < t[y]; }
void gsz(int x, int f) {
t[x] = 1;
for (R int i = 0, y; i < s[x]; ++i)
if (!v[y = g[x][i]] && y ^ f)
gsz(y, x), t[x] += t[y];
}
void grt(int x, int f, int a) {
R int m = 0, i, y;
for (i = 0; i < s[x]; ++i)
if (!v[y = g[x][i]] && y ^ f)
m = max(m, t[y]), grt(y, x, a);
m = max(m, a - t[x]);
if (m < u)
u = m, r = x;
}
void dfs(int x, int f, int d) {
c[++e] = d;
for (R int i = 0, y; i < s[x]; ++i)
if (!v[y = g[x][i]] && y ^ f)
dfs(y, x, d + 1);
}
I L pwr(L a, L b) {
L r = 1;
for (; b; b >>= 1, a = a * a % P)
if (b & 1)
r = r * a % P;
return r;
}
void ntt(int *f, int v) {
R int i, j, k, t;
L p, q, o;
for (i = 0; i < M; ++i)
if (i < d[i])
swp(f[i], f[d[i]]);
for (i = 1; i < M; i <<= 1) {
t = i << 1, p = pwr(v ? G : H, (P - 1) / t);
for (j = 0; j < M; j += t)
for (q = 1, k = 0; k < i; ++k)
o = q * f[i + j + k] % P, f[i + j + k] = (f[j + k] - o + P) % P, f[j + k] = (f[j + k] + o + P) % P, q = q * p % P;
}
}
void dac(int x) {
R int i, j, y, z;
p[0] = 1, h[0] = 0, u = S, gsz(x, 0), grt(x, 0, t[x]), v[r] = 1, sort(&g[r][0], s[r] + &g[r][0], cmp);
for (i = 0; i < s[r]; ++i)
if (!v[y = g[r][i]]) {
for (M = 1; M <= t[x]; M <<= 1) {}
e = 0, dfs(y, r, 1), F = pwr(M, P - 2), d[0] = 0;
for (z = M >> 1, j = 0; j < M; ++j)
d[j] = (d[j >> 1] >> 1)|((j & 1) ? z : 0);
for (j = 1; j <= e; ++j)
++q[c[j]], h[++h[0]] = c[j];
memcpy(a, p, M * 4), memcpy(b, q, M * 4);
ntt(a, 1), ntt(b, 1);
for (j = 0; j < M; ++j)
a[j] = 1ll * a[j] * b[j] % P;
ntt(a, 0);
for (j = 0; j < M; ++j)
p[j] += q[j], a[j] = 1ll * a[j] * F * 2 % P, o += (D)a[j] / (j + 1);
for (j = 1; j <= e; ++j)
q[c[j]] = 0;
}
for (i = 1; i <= h[0]; ++i)
p[h[i]] = 0;
for (x = r, i = 0; i < s[x]; ++i)
if(!v[y = g[x][i]])
dac(y);
}
int main() {
R int n = rd(), i, x, y;
for (S = 1; S <= n; S <<= 1) {}
for (i = 1; i < n; ++i)
x = rd() + 1, y = rd() + 1, g[x].push_back(y), g[y].push_back(x);
for (i = 1; i <= n; ++i)
s[i] = g[i].size();
o = n, dac(1), printf("%.4lf", o);
return 0;
}

码风也变了。。。

[BZOJ3451]normal 点分治,NTT的更多相关文章

  1. [BZOJ3451]Normal(点分治+FFT)

    [BZOJ3451]Normal(点分治+FFT) 题面 给你一棵 n个点的树,对这棵树进行随机点分治,每次随机一个点作为分治中心.定义消耗时间为每层分治的子树大小之和,求消耗时间的期望. 分析 根据 ...

  2. #565. 「LibreOJ Round #10」mathematican 的二进制(期望 + 分治NTT)

    题面 戳这里,题意简单易懂. 题解 首先我们发现,操作是可以不考虑顺序的,因为每次操作会加一个 \(1\) ,每次进位会减少一个 \(1\) ,我们就可以考虑最后 \(1\) 的个数(也就是最后的和) ...

  3. LOJ2541 PKUWC2018猎人杀(概率期望+容斥原理+生成函数+分治NTT)

    考虑容斥,枚举一个子集S在1号猎人之后死.显然这个概率是w1/(Σwi+w1) (i∈S).于是我们统计出各种子集和的系数即可,造出一堆形如(-xwi+1)的生成函数,分治NTT卷起来就可以了. #i ...

  4. 【BZOJ-3456】城市规划 CDQ分治 + NTT

    题目链接 http://www.lydsy.com/JudgeOnline/problem.php?id=3456 Solution 这个问题可以考虑dp,利用补集思想 N个点的简单图总数量为$2^{ ...

  5. CF960G Bandit Blues 【第一类斯特林数 + 分治NTT】

    题目链接 CF960G 题解 同FJOI2016只不过数据范围变大了 考虑如何预处理第一类斯特林数 性质 \[x^{\overline{n}} = \sum\limits_{i = 0}^{n}\be ...

  6. 洛谷5月月赛T30212 玩游戏 【分治NTT + 多项式求ln】

    题目链接 洛谷T30212 题解 式子很容易推出来,二项式定理展开后对于\(k\)的答案即可化简为如下: \[k!(\sum\limits_{i = 0}^{k} \frac{\sum\limits_ ...

  7. loj2541 「PKUWC2018」猎人杀 【容斥 + 分治NTT】

    题目链接 loj2541 题解 思路很妙啊, 人傻想不到啊 觉得十分难求,考虑容斥 由于\(1\)号可能不是最后一个被杀的,我们容斥一下\(1\)号之后至少有几个没被杀 我们令\(A = \sum\l ...

  8. hdu5279 YJC plays Minecraft 【分治NTT】

    题目链接 hdu5279 题解 给出若干个完全图,然后完全图之间首尾相连并成环,要求删边使得两点之间路径数不超过\(1\),求方案数 容易想到各个完全图是独立的,每个完全图要删成一个森林,其实就是询问 ...

  9. CF960G Bandit Blues 分治+NTT(第一类斯特林数)

    $ \color{#0066ff}{ 题目描述 }$ 给你三个正整数 \(n\),\(a\),\(b\),定义 \(A\) 为一个排列中是前缀最大值的数的个数,定义 \(B\) 为一个排列中是后缀最大 ...

随机推荐

  1. [翻译] SCRecorder

    SCRecorder https://github.com/rFlex/SCRecorder An easy Vine/Instagram like video and/or audio record ...

  2. [翻译] snapshotViewAfterScreenUpdates

    snapshotViewAfterScreenUpdates This method very efficiently captures the current rendered appearance ...

  3. 使用AHKActionSheet

    使用AHKActionSheet https://github.com/fastred/AHKActionSheet 基本配置代码: AHKActionSheet *actionSheet = [[A ...

  4. XGBoost 输出特征重要性以及筛选特征

    1.输出XGBoost特征的重要性 from matplotlib import pyplot pyplot.bar(range(len(model_XGB.feature_importances_) ...

  5. EF 实体类的制定属性不生成数据库字段

    添加一个标签即可 [NotMapped] 没什么营养,就是防忘记

  6. 上手 Kubernetes

    一.Kubernetes架构 二.重要的几个概念 1.Pod:最小单元 1.1 一个Pod里可以有多个Container 1.2这些container共享这个pod的同一个网络地址,同一个文件系统,通 ...

  7. kubeadm init 时从本地私有仓库下载镜像

    #kubeadm init 时从本地私有仓库下载镜像 images=( gcr.io/google_containers/kube-proxy-amd64:v1.6.1 gcr.io/google_c ...

  8. Python自动化之跨域访问jsonp

    这里提到了JSONP,那有人就问了,它同JSON有什么区别不同和区别呢,接下我们就来看看,百度百科有以下说明: ''' 1. JSON(JavaScript Object Notation) 是一种轻 ...

  9. python自动化之上传文件的3种方法和图片预览

    ajax上传文件的三种方法 第一种xhr提交 function xhrSubmit(){ // $('#fafafa')[0] var file_obj = document.getElementBy ...

  10. Qt在线/离线安装包下载网址和说明

    截至到2018年3月27日,Qt最新版本的安装需要使用在线安装工具:qt-unified-windows-x86-3.0.2-online.exe,该安装个工具的下载地址为: http://downl ...