垃圾ACMer的暑假训练220717

垃圾ACMer的暑假训练220717

33.3 Stirling数

[二项式反演] 若$f_n=\sum _{k=0}^n{C}_n^k{g}_k$,则$g_n=\sum _{k=0}^n(-1{)}^{n-k}{C}_n^k{f}_k$.

[证] $\sum _{k=0}^n(-1{)}^{n-k}{C}_n^k{f}_k=\sum _{k=0}^n(-1{)}^{n-k}{C}_n^k\sum _{i=0}^k{C}_n^i{g}_i=\sum _{i=0}^n{g}_i\sum _{k=i}^n(-1{)}^{n-k}{C}_n^k{C}_k^i$

​ $=\sum _{i=0}^n{g}_i\sum _{k=i}^n(-1{)}^{n-k}\cdot \frac{n!}{k!(n-k)!}\cdot \frac{k!}{i!(k-i)!}=\sum _{i=0}^n{g}_i\sum _{k=i}^n(-1{)}^{n-k}\cdot \frac{n!}{i!(n-i)!}{C}_{n-i}^{n-k}$

​ $=\sum _{i=0}^n{C}_n^i{g}_i\sum _{k=i}^n{C}_{n-i}^{n-k}(-1{)}^{n-k}\stackrel{j=n-k}{=}\sum _{i=0}^n{C}_n^i{g}_i\sum _{j=0}^{n-i}{C}_{n-i}^j(-1{)}^j=\sum _{i=0}^n{C}_n^i{g}_i(1-1{)}^{n-i}=g_n{C}_n^n=g_n$.

---

33.3.1 第二类Stirling数

[第二类Stirling数,Stirling子集数,整数的无序分拆数] 将$n$个两两不同的元素划分为$k$个互不区分的非空子集的方案数,记作$\left\{\begin{array}{c}n\\ k\end{array}\right\}$或$S(n,k)$.

[递推式] $\left\{\begin{array}{c}n\\ k\end{array}\right\}=\left\{\begin{array}{c}n-1\\ k-1\end{array}\right\}+k\left\{\begin{array}{c}n-1\\ k\end{array}\right\}$,初始条件$\left\{\begin{array}{c}n\\ 0\end{array}\right\}=[n=0]$.

[证] 每插入一个新元素时,有两种方案:

①将新元素单独放在一个子集,有$\left\{\begin{array}{c}n-1\\ k-1\end{array}\right\}$种方案.

②将新元素放入一个现有的非空子集,有$k\left\{\begin{array}{c}n-1\\ k\end{array}\right\}$种方案.

[性质] $m^n=\sum _{i=1}^m{C}_m^i\left\{\begin{array}{c}n\\ i\end{array}\right\}\cdot i!$.

[证] $\mathrm{LHS}$为将$n$个两两不同的元素划分为$m$个两两不同的子集(允许空集)的方案数.

$\mathrm{RHS}$为枚举非空子集个数$i$求得的答案,其中$i!=A_i^i$是对$i$个非空子集的排序.

[通项公式] $\left\{\begin{array}{c}n\\ m\end{array}\right\}=\sum _{i=0}^m\frac{(-1{)}^{m-i}{i}^n}{i!(m-i)!}$.

[证1] 因$f(m)=m^n=\sum _{i=1}^m{C}_m^i\left\{\begin{array}{c}n\\ i\end{array}\right\}\cdot i!=\sum _{i=1}^m{C}_m^{i}g(i)$,

由二项式反演:$g(m)=\left\{\begin{array}{c}n\\ m\end{array}\right\}\cdot m!=\sum _{i=1}^m(-1{)}^{m-i}{C}_m^i{i}^n$,移项即证.

[证2] 设将$n$个两两不同的元素划分到$k$个两两不同的集合(允许空集)的方案数为$G_n$,将$n$个两两不同的元素划分到$k$个两两不同的非空集合(不允许空集)的方案数$F_n$.

显然$G_n=k^n,G_n=\sum _{j=0}^i{C}_i^j{F}_j$.

由二项式反演:$F_i=\sum _{j=0}^i(-1{)}^{i-j}{C}_i^j{G}_j=\sum _{j=0}^i(-1{)}^{i-j}{C}_i^j{j}^n=\sum _{j=0}^i\frac{(i-1{)}^{i-j}i!j^n}{j!(i-j)!}$.

因第二类Stirling数要求集合间互不区分,则$F_i$是$\left\{\begin{array}{c}n\\ i\end{array}\right\}$的$i!$倍,故$\left\{\begin{array}{c}n\\ m\end{array}\right\}=\frac{F_m}{m!}=\sum _{i=0}^m\frac{(-1{)}^{m-i}{i}^n}{i!(m-i)!}$.

---

33.3.1.1 第二类Stirling数·行

题意 $(0.5\mathrm{s})$

给定整数$n(1\le n\le 2\mathrm{e}5)$,对所有$i\in [0,n]$,输出第二类Stirling数$\left\{\begin{array}{c}n\\ i\end{array}\right\}$,答案对素数$167772161=2^{25}\times 5+1$取模.

思路

$167772161$的原根为$3$.

由通项公式$\left\{\begin{array}{c}n\\ m\end{array}\right\}=\sum _{i=0}^m\frac{(-1{)}^{m-i}{i}^n}{i!(m-i)!}=\sum _{i=0}^m\frac{(-1{)}^{m-i}}{(m-i)!}\cdot \frac{i^n}{i!}$.

设$f(n)=\frac{(-1{)}^n}{n!},g(n)=\frac{i^n}{i!}$,则$\left\{\begin{array}{c}n\\ i\end{array}\right\}$是$(f\ast g)(n)$的第$i$项的系数.用NTT计算卷积即可.时间复杂度$O(n\log n)$.

对$f(i)$:①$i$为偶数时,取系数为$i!$的逆元$ifac[i]$;②$i$为奇数时,取系数为$MOD-ifac[i]$.

代码 (附带ll版的NTT)

const int MAXN = 6e5 + 5;
const ll MOD = 167772161;

int rev[MAXN];  // rev[x]表示二进制数x镜像后的结果

void NTT(ll* y, int lim, int op) {  // op=1时为NTT,op=-1时为INTT
	for (int i = 0; i < lim; i++)  // 做一遍BRP,将系数位置调整为递归后的位置
		if (rev[i] < i) swap(y[i], y[rev[i]]);

	for (int l = 2; l <= lim; l <<= 1) {  // 模拟合并过程,l为当前区间长度
		ll g = qpow(3, (MOD - 1) / l, MOD);  // 注意ll
		for (int i = 0; i < lim; i += l) {
			int cur = 1;
			int k = l >> 1;
			for (int j = 0; j < k; j++, cur = cur * g % MOD) {
				ll tmp = y[i + j + k] * cur % MOD;  // 注意ll
				y[i + j + k] = (y[i + j] - tmp + MOD) % MOD;  // 奇次项
				y[i + j] = (y[i + j] + tmp) % MOD;  // 偶次项
			}
		}
	}

	if (op == -1) {  // INTT
		reverse(y + 1, y + lim);
		ll inv = qpow(lim, MOD - 2, MOD);  // 注意ll
		for (int i = 0; i < lim; i++) y[i] = y[i] * inv % MOD;
	}
}

ll fac[MAXN], ifac[MAXN];  // 阶乘及其逆元

void init(int n) {  // 预处理阶乘及其逆元
	fac[0] = 1;
	for (int i = 1; i <= n; i++) fac[i] = fac[i - 1] * i % MOD;
	ifac[n] = qpow(fac[n], MOD - 2, MOD);
	for (int i = n - 1; i >= 0; i--) ifac[i] = ifac[i + 1] * (i + 1) % MOD;
}

int n;
ll a[MAXN], b[MAXN];  // 多项式f和g的系数

int main() {
	cin >> n;
	
	init(n);  // 预处理阶乘及其逆元

	for (int i = 0; i <= n; i++) {
		a[i] = (i & 1 ? MOD - ifac[i] : ifac[i]);
		b[i] = qpow(i, n, MOD) * ifac[i] % MOD;
	}

	int lim = 1;
	while (lim < (n << 1)) lim <<= 1;

	for (int i = 0; i < lim; i++) 
		rev[i] = (rev[i >> 1] >> 1) + (i & 1) * (lim >> 1);

	NTT(a, lim, 1), NTT(b, lim, 1);
	for (int i = 0; i < lim; i++) a[i] = a[i] * b[i] % MOD;
	NTT(a, lim, -1);

	for (int i = 0; i <= n; i++) cout << a[i] << ' ';
}

---

---

33.3.1.2 染色

题意

有一个长度为$n$的序列和$m$种颜色.现对其染色.若染色后的序列种出现次数恰为$s$的颜色种数为$k$,则会得到$w_k$的愉悦度.对所有染色方案,求可获得的愉悦度之和,答案对$1004535809$取模.

第一行输入整数$n,m,s(1\le n\le 1\mathrm{e}7,1\le m\le 1\mathrm{e}5,1\le s\le 150)$.第二行输入$(m+1)$个整数$w_0,\cdots ,w_m(0\le w_i<1004535809)$.

思路

$1004535809$的原根为$3$.

每种方案中,对愉悦度有贡献的颜色数不超过$color=\min \left\{m,\lfloor \frac{n}{s}\rfloor \right\}$.

设至少有$i$种颜色恰出现$s$次的方案数为$f_i$.从$m$种颜色中选$i$种,乘$C_m^i$.将$n$个位置分为$(i+1)$个部分,其中有$i$种恰出现$s$次的元素,剩下$(m-i)$种颜色共$(n-is)$个.先视为可重排列,乘$\frac{n!}{(s!{)}^i(n-is)!}$.$n$个位置的前$i$个部分都只有一种颜色,最后一个部分每个位置可取$(m-i)$种颜色,乘$(m-i{)}^{n-is}$.故$f_i=C_m^i\cdot \frac{n!}{(s!{)}^i(n-is)!}\cdot (m-i{)}^{n-is}$.

设有$i$种颜色恰出现$s$次的方案数为$an{s}_i$.由容斥原理:$an{s}_i=\sum _{j=i}^{color}(-1{)}^{j-i}{C}_j^i{f}_j$,

​ 则$an{s}_i\cdot i!=\sum _{j=i}^{color}\frac{(-1{)}^{j-i}}{(j-i)!}\cdot (f_j\cdot j!)$,用NTT求即可.

代码

const int MAXN = 3e7 + 5;
const ll MOD = 1004535809;

int rev[MAXN];  // rev[x]表示二进制数x镜像后的结果

void NTT(ll* y, int lim, int op) {  // op=1时为NTT,op=-1时为INTT
	for (int i = 0; i < lim; i++)  // 做一遍BRP,将系数位置调整为递归后的位置
		if (rev[i] < i) swap(y[i], y[rev[i]]);

	for (int l = 2; l <= lim; l <<= 1) {  // 模拟合并过程,l为当前区间长度
		ll g = qpow(3, (MOD - 1) / l, MOD);
		for (int i = 0; i < lim; i += l) {
			int cur = 1;
			int k = l >> 1;
			for (int j = 0; j < k; j++, cur = cur * g % MOD) {
				ll tmp = y[i + j + k] * cur % MOD;
				y[i + j + k] = (y[i + j] - tmp + MOD) % MOD;  // 奇次项
				y[i + j] = (y[i + j] + tmp) % MOD;  // 偶次项
			}
		}
	}

	if (op == -1) {  // INTT
		reverse(y + 1, y + lim);
		ll inv = qpow(lim, MOD - 2, MOD);
		for (int i = 0; i < lim; i++) y[i] = y[i] * inv % MOD;
	}
}

ll fac[MAXN], ifac[MAXN];  // 阶乘及其逆元

void init(int n) {  // 预处理阶乘及其逆元
	fac[0] = 1;
	for (int i = 1; i <= n; i++) fac[i] = fac[i - 1] * i % MOD;
	ifac[n] = qpow(fac[n], MOD - 2, MOD);
	for (int i = n - 1; i >= 0; i--) ifac[i] = ifac[i + 1] * (i + 1) % MOD;
}

ll C(int n, int m) {
	if (n < m) return 0;
	if (n == m) return 1;
	return fac[n] * ifac[m] % MOD * ifac[n - m] % MOD;
}

int n, m, s;  // 序列长度、颜色数、出现次数
ll w[MAXN];  // 愉悦度
ll a[MAXN], b[MAXN];  // 多项式f和g的系数

int main() {
	cin >> n >> m >> s;
	for (int i = 0; i <= m; i++) cin >> w[i];
	
	init(max({ n,m,s }));  // 预处理阶乘及其逆元

	int color = min(m, n / s);

	int lim = 1;
	while (lim < (color + 1 << 1)) lim <<= 1;  // 注意color+1

	for (int i = 0; i <= color; i++) {
		ll inv = qpow(qpow(fac[s], i, MOD) * fac[n - i * s] % MOD, MOD - 2, MOD);  // fi的分母
		a[i] = C(m, i) * fac[n] % MOD * qpow(m - i, n - i * s, MOD) % MOD * inv % MOD * fac[i] % MOD;
		b[i] = ((color - i) & 1 ? MOD - ifac[color - i] : ifac[color - i]);
	}

	for (int i = 0; i < lim; i++) 
		rev[i] = (rev[i >> 1] >> 1) | (i & 1) * (lim >> 1);

	NTT(a, lim, 1), NTT(b, lim, 1);
	for (int i = 0; i < lim; i++) a[i] = a[i] * b[i] % MOD;
	NTT(a, lim, -1);

	int ans = 0;
	for (int i = 0; i <= color; i++) ans = ((ll)ans + (ll)w[i] * a[color + i] % MOD * ifac[i] % MOD) % MOD;
	cout << ans;
}

---

---

Codeforces

Codeforces Round #808 (Div. 2) A. Difference Operations

原题指路:https://codeforces.com/contest/1708/problem/A

题意

给定一个长度为$n(2\le n\le 100)$的序列$a$,问能否进行若干次(可能为$0$次)操作,使得$a_i=0(2\le i\le n)$.操作:选择一个下标$i\in [2,n]$,令$a_i-=a_{i-1}$,

有$t(1\le t\le 100)$组测试数据.每组测试数据第一行输入一个整数$n(2\le n\le 100)$.第二行输入$n$个整数$a_1,\cdots ,a_n(1\le a_i\le 1\mathrm{e}9)$.

对每组测试数据,若能进行若干次(可能为$0$次)操作,使得$a_i=0(2\le i\le n)$,则输出"YES";否则输出"NO".

思路I

有解的充要条件是:$a_2\sim a_n$都能被$a_1$整除.

[证] (充) 显.

(必) $a_2$是$a_1$的倍数,否则$a_2$不能变为$0$.$a_3$是$a_2$的倍数,否则$a_3$不能变为$0$.而$a_2$是$a_1$的倍数,故$a_3$是$a_1$的倍数.以此类推.

思路II

考虑相反的问题:给定一个长度为$n$的序列$b=[b_1,0,\cdots ,0]$和序列$a$,满足$a_1=b_1$,问$b$能否进行若干次操作变为$a$.操作:选择一个下标$2\le i\le n$,令$b_i=b_i+b_{i-1}$.显然有解的充要条件是:$a_2\sim a_n$都能被$a_1$整除.

代码

int main() {
	CaseT{
		int n; cin >> n;
		vi a(n + 2);
		for (int i = 1; i <= n; i++) cin >> a[i];

		bool ok = true;
		for (int i = 2; i <= n; i++) {
			if (a[i] % a[1]) {
				ok = false;
				break;
			}
		}
		cout << (ok ? "YES" : "NO") << endl;
	}
}

---

---

Codeforces Round #808 (Div. 2) B. Difference of GCDs

原题指路:https://codeforces.com/contest/1708/problem/B

题意

给定区间$[l,r]$,构造一个长度为$n$的序列$a_1,\cdots ,a_n(l\le a_i\le r)s.t.\gcd (i,a_i)(1\le i\le n)$相异.

有$t(1\le t\le 1\mathrm{e}4)$组测试数据.每组测试数据输入三个整数$n,l,r(1\le n\le 1\mathrm{e}5,1\le l\le r\le 1\mathrm{e}9)$.数据保证所有测试数据的$n$之和不超过$1\mathrm{e}5$.

对每组测试数据,若能构造出序列$a$,则第一行输出"YES",第二行输出序列$a$;否则输出一行"NO".

思路

因$\gcd (i,a_i)\le i$,而$\gcd (i,a_i)(1\le i\le n)$相异,则$\gcd (i,a_i)=i$,即$a_i$是$i$的倍数.

注意到$\ge l$的最小的$i$的倍数是$tmp=\lceil \frac{l}{i}\rceil \cdot i$,故只需检查$tmp\le r$是否成立.

代码

int main() {
	CaseT{
		int n; cin >> n;
		int l, r; cin >> l >> r;

		if (n == 1) {
			cout << "YES" << endl;
			cout << l << endl;
			continue;
		}

		vi ans(n + 2);
		ans[1] = l;
		int cnt = 1;
		for (int i = 2; i <= n; i++) {
			if (cnt >= n) break;

			int tmp = ceil((double)l / i) * i;
			if (tmp <= r && !ans[i]) {
				ans[i] = tmp;
				cnt++;
			}
		}

		if (cnt != n) {
			cout << "NO" << endl;
			continue;
		}
		else {
			cout << "YES" << endl;
			for (int i = 1; i <= n; i++) cout << ans[i] << ' ';
			cout << endl;
		}
	}
}

---

---

Codeforces Round #808 (Div. 2) C. Doremy’s IQ

原题指路:https://codeforces.com/contest/1708/problem/C

题意

你初始时的IQ为$q$.现你要打$n$场比赛,其中第$i$场比赛的难度为$a_i$,且只能在第$i$天打.你能打比赛的前提是当前的$q>0$.在第$i$天,你可选择打或不打当天的比赛:①若$a_i>q$,则$q--$;②若$a_i\le q$,无事发生.求一个策略,使得你可以打尽量多场比赛.策略用一个长度为$n$的二进制串表示,$1$表示打当天的比赛,$0$表示不打当天的比赛.

有$t(1\le t\le 1\mathrm{e}4)$组测试数据.每组测试数据第一行输入两整数$n,q(1\le n\le 1\mathrm{e}5,1\le q\le 1\mathrm{e}9)$.第二行输入$n$个整数$a_1,\cdots ,a_n(1\le a_i\le 1\mathrm{e}9)$.数据保证所有测试数据的$n$之和不超过$1\mathrm{e}5$.

对每组测试数据,任输出一个可打尽量多场比赛的策略.

思路I

称会降低$q$的比赛为坏比赛,其余比赛为好比赛.

最优策略中存在一个下标$x\ s.t.\ $①$[1,x)$中的好比赛都打了;②$[x,n]$中的比赛都打了.

[证] 对每个策略,设$a$为最后一场没打的比赛,$b$为第一场打的坏比赛.

①若$b<a$,则可不打$b$,改打$a$,这样打的比赛数不变.

②若$b>a$,则$x=b$即满足.

当$x$最小时即为最优策略,$x$的最小值可二分.时间复杂度$O(n\log n)$.

代码I

const int MAXN = 1e5 + 5;
int n, q;
int a[MAXN];
int ans[MAXN];

bool check(int m) {
	for (int i = 1; i <= n; i++) ans[i] = 0;

	int curq = q;
	for (int i = 1; i < m; i++)
		if (a[i] <= curq) ans[i] = 1;

	for (int i = m; i <= n; i++) {
		ans[i] = 1;
		if (!curq) return false;
		if (a[i] > curq)curq--;
	}
	return curq >= 0;
}

int main() {
	CaseT{
		cin >> n >> q;
		for (int i = 1; i <= n; i++) cin >> a[i];

		int l = 1, r = n;
		while (l < r) {
			int mid = l + r >> 1;
			if (check(mid)) r = mid;
			else l = mid + 1;
		}

		check(l);  // 求策略
		for (int i = 1; i <= n; i++) cout << ans[i];
		cout << endl;
	}
}

---

思路II

逆向考虑问题.初始时在最后一场比赛的后一天,且$curq=0$.

①$a_i\le curq$时,则应打第$i$场比赛.

②$a_i>curq$且$curq<q$时,若打第$i$场比赛,则$curq++$;否则无事发生.

​ 注意到最多能打$q$场满足$a_i>curq$的比赛,若能使$curq$尽量大,就能打更多前面的好比赛.故应打第$i$场比赛.

③$a_i>curq$且$curq=q$时,无法打第$i$场比赛.

倒着决定每场比赛是否打即可.时间复杂度$O(n)$.

代码II

int main() {
	CaseT{
		int n, q; cin >> n >> q;
		vi a(n + 1);
		for (int i = 1; i <= n; i++) cin >> a[i];

		int curq = 0;
		vi ans(n + 1);
		for (int i = n; i; i--) {
			if (a[i] <= curq) ans[i] = 1;
			else if (curq < q) curq++, ans[i] = 1;
			else ans[i] = 0;
		}

		for (int i = 1; i <= n; i++) cout << ans[i];
		cout << endl;
	}
}

---

---