[题解][Codeforces 1172A~1172D]Codeforces Round #564 (Div. 1) 前四题题解

• 中国场，最后两题为毒瘤数据结构题，还未写，见谅

Address

洛谷 RemoteJudge

• A
• B
• C
• D
• E
• F

Codeforces

• A
• B
• C1
• C2
• D
• E
• F

A

Meaning

• 两个序列 $a$ 和 $b$ ，长度为 $n$ ，每个数均为 $0$ 或者 $1$ 到 $n$ 之间的整数，且 $1$ 到 $n$ 的每个数出现且仅出现一次
• 每次操作可以在第一个序列里抽走一个数放到第二个序列的末尾，并把第二个序列的开头元素拿到第一个序列内
• （从第一个序列中抽走的数的位置是任意的）
• 求把第二个序列变成依次从 $1$ 到 $n$ 的最少步数
• $1\le n\le 2\times 10^5$

Solution

• 先判定 $b$ 是否存在形如 $123\dots x$ 的后缀
• 如果存在，则尝试在第 $i$ 次操作时把数 $x+i$ 放到序列末尾
• 如果尝试成功（操作时对于每个 $i$ 都满足数 $x+i$ 存在于序列 $a$ 内）那么这就是最优方案，操作次数 $n-x$
• 否则回到开始，每一次操作都把 $0$ 放到序列末尾，直到可以放 $1$ 为止
• 当然，在何时放 $1$ 需要满足一定的条件
• 可以发现，如果 $b$ 中 $x$ 在位置 $y$ （ $x̸=0$ ），那么 $1$ 至少需要在第 $y-x+2$ 次操作时被放入序列，否则数 $x$ 需要被放入序列时 $x$ 不在序列 $a$ 内
• 放了 $1$ 之后，就可以依次把 $2$ 到 $n$ 的数放到末尾了
• 于是这时答案为
• n − 1 + max ⁡ b i ≠ 0 { i − b i + 2 } n-1+\max_{b_i\ne 0}\{i-b_i+2\} n−1+bi​̸​=0max​{i−bi​+2}
• 注意 $b$ 全 $0$ 的情况
• $O(n)$

Code

#include <bits/stdc++.h>

inline int read()
{
	int res = 0; bool bo = 0; char c;
	while (((c = getchar()) < '0' || c > '9') && c != '-');
	if (c == '-') bo = 1; else res = c - 48;
	while ((c = getchar()) >= '0' && c <= '9')
		res = (res << 3) + (res << 1) + (c - 48);
	return bo ? ~res + 1 : res;
}

const int N = 2e5 + 5;

int n, a[N], pos1, ans, lst;

int main()
{
	n = read();
	for (int i = 1; i <= n; i++) read();
	for (int i = 1; i <= n; i++) a[i] = read();
	for (int i = 1; i <= n; i++)
	{
		if (a[i] == 1) pos1 = i;
		if (a[i]) lst = i;
	}
	bool is = 1;
	for (int i = pos1; i <= n; i++)
		if (a[i] != i - pos1 + 1) is = 0;
	if (is)
	{
		bool can = 1;
		for (int i = 1; i < pos1; i++) if (a[i] &&
			i - (a[i] - (n - pos1 + 1) - 1) >= 1) can = 0;
		if (can) return std::cout << pos1 - 1 << std::endl, 0;
	}
	pos1 = 2;
	for (int i = 1; i <= n; i++) if (a[i])
		pos1 = std::max(pos1, i - (a[i] - 3));
	std::cout << pos1 + n - 2 << std::endl;
	return 0;
}

B

Meaning

• 给定一棵树
• 求有多少个不同的 $1$ 到 $n$ 排列 $p$
• 使得对于任意两个满足树上存在边 $(p_i,p_j)$ 和 $(p_x,p_y)$ 的二元组 $(i,j)(x,y)$ 都满足圆周上有 $n$ 个点的圆环上点 $i$ 和 $j$ 连成的边以及点 $x$ 和 $y$ 连成的边不在除端点之外的点相交
• 求方案数对 $998244353$ 取模
• $2\le n\le 2\times 10^5$

Solution

• 转化一下
• 定 $1$ 为根，先把 $1$ 在圆环上定一个位置，有 $n$ 种方案
• 然后对每个子树分配环上的一个连续段
• 然后对每个子树（连续段）分别处理
• 定义 $f[u]$ 表示为子树 $u$ 分配位置的方案数（不包括为 $1$ 定位）
• 当 $u=1$ 时显然有 $d_u!$ 种方案（ $d_u$ 为 $u$ 的子节点数）
• 否则在子树 $u$ 所属的连续段上，相当于需要为 $u$ 的所有子树，包括上 $u$ 分别分配一个连续段，共 $d_u+1$ 个连续段，方案数 $(d_u+1)!$
• $$f[u]=d_u!\times (d_u+1{)}^{[u>1]}\prod _{v\in son(u)}f[v]$$
• 这样答案就是所有点的度数（无根树）阶乘之积乘上 $n$
• $O(n)$

Code

#include <bits/stdc++.h>

inline int read()
{
	int res = 0; bool bo = 0; char c;
	while (((c = getchar()) < '0' || c > '9') && c != '-');
	if (c == '-') bo = 1; else res = c - 48;
	while ((c = getchar()) >= '0' && c <= '9')
		res = (res << 3) + (res << 1) + (c - 48);
	return bo ? ~res + 1 : res;
}

const int N = 2e5 + 5, M = N << 1, ZZQ = 998244353;

int n, ecnt, nxt[M], adj[N], go[M], f[N];

void add_edge(int u, int v)
{
	nxt[++ecnt] = adj[u]; adj[u] = ecnt; go[ecnt] = v;
	nxt[++ecnt] = adj[v]; adj[v] = ecnt; go[ecnt] = u;
}

void dfs(int u, int fu)
{
	int cnt = 0, fc = 1, fac = 1;
	f[u] = 1;
	for (int e = adj[u], v; e; e = nxt[e])
		if ((v = go[e]) != fu) dfs(v, u), cnt++;
	for (int i = 1; i <= cnt; i++)
		fac = 1ll * fac * i % ZZQ;
	f[u] = u == 1 ? fac : 1ll * fac * (cnt + 1) % ZZQ;
	for (int e = adj[u], v; e; e = nxt[e])
		if ((v = go[e]) != fu)
			f[u] = 1ll * f[u] * f[v] % ZZQ;
}

int main()
{
	int x, y;
	n = read();
	for (int i = 1; i < n; i++)
		x = read(), y = read(), add_edge(x, y);
	dfs(1, 0);
	std::cout << 1ll * n * f[1] % ZZQ << std::endl;
	return 0;
}

C

Meaning

• 有 $n$ 个二元组，第 $i$ 个形如 $(a_i,w_i)$
• $a_i$ 是一个 $0/1$ 值
• $w_i$ 是一个正整数
• 接下来有 $m$ 次操作
• 每次操作会随机地选择一个二元组，其中选出第 $i$ 个二元组的概率为
• $$\frac{w_i}{{\sum }_{j=1}^n{w}_j}$$
• 如果选出的二元组的 $a$ 为 $0$ ，把这个二元组的 $w$ 减一
• 否则选出的二元组的 $a$ 为 $1$ ，把这个二元组的 $w$ 加一
• 求 $m$ 次操作后每个二元组 $w$ 的期望值
• $1\le n\le 2\times 10^5$ ， $1\le m\le 3000$

Solution

• 介绍一种做法，（可能）比原题解好推一些？
• 先定义状态 $f[i][j]$ 表示 $i$ 次操作中选出 $a$ 为 $0$ 的二元组恰好 $j$ 次的概率
• $$f[0][0]=1$$
• $$f[i+1][j]+=\frac{s_1+i-j}{s_0+s_1+i-2j}f[i][j]$$
• $$f[i+1][j+1]+=\frac{s_0-j}{s_0+s_1+i-2j}f[i][j]$$
• 其中 $s_0={\sum }_{a_i=0}{w}_i$ ， $s_1={\sum }_{a_i=1}{w}_i$
• 乍一看这个 DP 好像没什么用，不妨分析一下性质
• 以 $a_i=1$ 为例，可以发现一个很优秀的性质
• 如果某一次选出了的二元组的 $a$ 为 $0$ ，那么在下一次选出的 a 为 1 的条件下，选到的是第 $i$ 个二元组的条件概率不变，仍然为现在的 $w_i$ 除以现在 $a=1$ 的 $w$ 之和
• $a_i=0$ 同理
• 定义一次操作「对 $x$ 有效」，当且仅当这次操作选出的二元组的 $a$ 和第 $x$ 个二元组的 $a$ 一样
• 再设状态 $g_x[i]$ 表示第 $x$ 个二元组，在 $i$ 次对 $x$ 有效的操作之后 $w_x$ 的期望
• 答案就很好算了
• 如果 $a_x=0$ 则最后 $w_x$ 的期望为
• $$\sum _{i=0}^{m}f[m][i]\times g_x[i]$$
• 否则 $a_x=1$
• $$\sum _{i=0}^{m}f[m][i]\times g_x[m-i]$$
• 考虑怎么求这个 $g_x[i]$ ，直接求和算答案是 $O(nm)$ ，无法通过 C2
• 以 $a_x=1$ 为例
• $g_x[0]=w_x$
• $$g_x[i]=g_x[i-1]+\frac{g_x[i-1]}{s_1+i-1}$$
• $a_x=0$ 时
• $$g_x[i]=g_x[i-1]-\frac{g_x[i-1]}{s_0-i+1}$$
• 即 $i-1$ 次操作之后的期望再加上第 $i$ 次选出第 $i$ 个二元组的概率
• 看上去还是 $O(nm)$ 的，所以我们需要找性质
• 把两个式子整理一下
• $$g_x[i]=g_x[i-1]\times \{\begin{array}{ll}\frac{s_0-i}{s_0-i+1}& a_x=0\\ \frac{s_1+i}{s_1+i-1}& a_x=1\end{array}$$
• 根据常识
• $$g_x[i]=w_x\times \{\begin{array}{ll}\frac{s_0-i}{s_0}& a_x=0\\ \frac{s_1+i}{s_1}& a_x=1\end{array}$$
• 代回答案，当 $a_x=0$ 时答案为
• $$\sum _{i=0}^{m}f[m][i]\times w_x\times \frac{s_0-i}{s_0}=w_x\times (\sum _{i=0}^{m}f[m][i]-\frac{{\sum }_{i=0}^{m}i\times f[m][i]}{s_0})$$
• 同理 $a_x=1$ 时
• $$\sum _{i=0}^{m}f[m][i]\times w_x\times \frac{s_1+m-i}{s_1}=w_x\times (\sum _{i=0}^{m}f[m][i]+\frac{{\sum }_{i=0}^m(m-i)\times f[m][i]}{s_1})$$
• ${\sum }_{i=0}^{m}f[m][i]$ ， ${\sum }_{i=0}^{m}i\times f[m][i]$ ， ${\sum }_{i=0}^m(m-i)\times f[m][i]$ 可以 DP 后预处理直接使用
• $O(n+m^2\log (s_0+s_1+m))$ （DP 时需要求逆元）
• 如果预处理 $s_0+s_1\pm m$ 的逆元可以做到 $O(n+m^2)$

Code

#include <bits/stdc++.h>

inline int read()
{
	int res = 0; bool bo = 0; char c;
	while (((c = getchar()) < '0' || c > '9') && c != '-');
	if (c == '-') bo = 1; else res = c - 48;
	while ((c = getchar()) >= '0' && c <= '9')
		res = (res << 3) + (res << 1) + (c - 48);
	return bo ? ~res + 1 : res;
}

const int N = 2e5 + 5, M = 3005, ZZQ = 998244353;

int n, m, a[N], b[N], s0, s1, f[M][M], ft, tft, tfmt, i0, i1;

int qpow(int a, int b)
{
	int res = 1;
	while (b)
	{
		if (b & 1) res = 1ll * res * a % ZZQ;
		a = 1ll * a * a % ZZQ;
		b >>= 1;
	}
	return res;
}

int main()
{
	n = read(); m = read();
	for (int i = 1; i <= n; i++) a[i] = read();
	for (int i = 1; i <= n; i++)
	{
		b[i] = read();
		if (a[i]) s1 += b[i];
		else s0 += b[i];
	}
	f[0][0] = 1;
	for (int i = 0; i < m; i++)
		for (int j = 0; j <= i; j++)
		{
			int it = qpow(s0 - j + s1 + i - j, ZZQ - 2);
			f[i + 1][j] = (1ll * (s1 + i - j) * it % ZZQ *
				f[i][j] + f[i + 1][j]) % ZZQ;
			f[i + 1][j + 1] = (1ll * (s0 - j) * it % ZZQ *
				f[i][j] + f[i + 1][j + 1]) % ZZQ;
		}
	for (int i = 0; i <= m; i++)
	{
		ft = (ft + f[m][i]) % ZZQ;
		tft = (1ll * i * f[m][i] + tft) % ZZQ;
		tfmt = (1ll * (m - i) * f[m][i] + tfmt) % ZZQ;
	}
	i0 = qpow(s0, ZZQ - 2); i1 = qpow(s1, ZZQ - 2);
	for (int i = 1; i <= n; i++)
		if (a[i]) printf("%d\n", (1ll * b[i] * ft + 1ll * b[i]
			* i1 % ZZQ * tfmt) % ZZQ);
		else printf("%d\n", (1ll * b[i] * ft - 1ll * b[i]
			* i0 % ZZQ * tft % ZZQ + ZZQ) % ZZQ);
	return 0;
}

D

Meaning

• 一个 $n\times n$ 的网格，要求在一些格子上放传送门
• 一对传送门 $(A,B)$ 表示走到 $A$ 会传送到 $B$ ，走到 $B$ 会传送到 $A$
• 现在需要构造一种放置传送门的方案（必须成对放，一个格子不能放超过一个传送门）
• 使得对于所有的 $i$ ，从第 $1$ 列第 $i$ 个格子出发一直向右走最终达到第 $n$ 列第 $r_i$ 个格子，并且从第 $1$ 行第 $i$ 个格子出发一直向下走最终达到第 $n$ 行第 $c_i$ 个格子
• $1\le n\le 1000$
• $r$ 和 $c$ 都是 $1$ 到 $n$ 的排列

Solution

• 考虑如何缩减问题的规模从 $n\times n$ 到 $(n-1)\times (n-1)$
• 如果 $r_1=c_1=1$ 跳过
• 否则如果 $r_x=c_y=1$ ，那么放一对传送门，位置为 $(x,1)(1,y)$
• 这样显然从第 $(x,1)$ 出发会最终到达 $(1,n)$ ， $(1,y)$ 出发最终达到 $(n,1)$
• 从 $(1,1)$ 向右出发会穿越到 $(x,2)$ ，向下出发会穿越到 $(2,y)$
• 这就是一个规模小 $1$ 的问题！！！！！
• $O(n)$
• 但 SPJ 是 $O(n^2)$ 的

Code

#include <bits/stdc++.h>

inline int read()
{
	int res = 0; bool bo = 0; char c;
	while (((c = getchar()) < '0' || c > '9') && c != '-');
	if (c == '-') bo = 1; else res = c - 48;
	while ((c = getchar()) >= '0' && c <= '9')
		res = (res << 3) + (res << 1) + (c - 48);
	return bo ? ~res + 1 : res;
}

const int N = 1005;

int n, m, ir[N], ic[N], a[N], ia[N], b[N], ib[N],
ans1x[N], ans1y[N], ans2x[N], ans2y[N];

int main()
{
	n = read();
	for (int i = 1; i <= n; i++) ir[read()] = a[i] = ia[i] = i;
	for (int i = 1; i <= n; i++) ic[read()] = b[i] = ib[i] = i;
	for (int i = 1; i <= n; i++)
	{
		if (ir[i] == a[i] && ic[i] == b[i]) continue;
		int x = ia[ir[i]], y = ib[ic[i]];
		ans1x[++m] = x; ans1y[m] = ans2x[m] = i;
		ans2y[m] = y;
		a[x] = a[i]; b[y] = b[i];
		ia[a[i]] = x; ib[b[i]] = y;
	}
	std::cout << m << std::endl;
	for (int i = 1; i <= m; i++)
		printf("%d %d %d %d\n", ans1x[i], ans1y[i], ans2x[i], ans2y[i]);
	return 0;
}