[LOJ#6617][THUPC2019]摆家具（矩阵乘法 + 子集和变换）

Address

• LOJ #6617

Solution

Task 1

• 先解决一个小问题
• 如何求 $T$ 次操作之后，对于所有的 $0\le i\le k$ ，求出最后恰好有 $i$ 个家具不在原来的房间内的方案数
• 容易设计一个 DP
• $f[i][j]$ 表示 $i$ 次操作之后，恰好有 $j$ 个家具不在原来的房间内的方案数
• 转移（自行理解）
• $$f[0][0]=1$$
• $$f[i][j]=f[i-1][j]\times j\times (n-2)+f[i-1][j-1]\times (k-j+1)\times (n-1)+f[i-1][j+1]\times (j+1)$$
• 可以预处理转移矩阵 $A$ 和初始向量 $St$
• 取参数 $S$ 使得 $S^2+S-1\le \max T$ （ $\max T=998244352$ ）
• 然后对于所有的 $0\le i\le S$ 预处理出矩阵 $A^i$ 和 $(A^S{)}^i$
• 那么
• $$A^n\times St=(A^S{)}^{\lfloor \frac{n}{S}\rfloor }\times (A^{n\mathrm{m}\mathrm{o}\mathrm{d}S}\times St)$$
• 可以 $O(S{k}^3)$ 预处理 $O(k^2)$ 单次询问，其中 $S=O(\sqrt{P})$

Task 2

• 对于每个 $S$ 和 $i$ ，预处理出与方案 $S$ 恰好有 $i$ 个家具的摆放位置不同的所有方案的得分之和
• 考虑和子集和变换一样，设 $F[S][i][j]$ 表示方案的前 $i$ 个家具中恰好有 $j$ 个家具摆放位置与 $S$ 不同，后 $k-i$ 个家具的摆放位置与 $S$ 都一样的方案数
• $$F[S][0][0]=a_S$$
• （ $a_S$ 为方案 $S$ 的得分）
• $$F[S][i][j]=F[S][i-1][j]+\sum _{replace(S,i,h)̸=S}F[replace(S,i,h)][i-1][j-1]$$
• $replace(S,i,h)$ 表示把方案 $S$ 中第 $i$ 件家具移到房间 $h$ 得到的方案
• $\sum$ 内的东西可以利用前缀和等方法求
• $O(k^2{n}^k)$

Solution

• 回到问题
• 根据常识， $T$ 次操作之后，对于每个 $i$ ，都满足：对于所有的方案 $S$ 满足 $a$ 和 $S$ 中恰好 $i$ 个家具的所处房间不同，从初始状态 $a$ 到达末状态 $S$ 的方案数都一样
• 于是对于一个特定的状态 $S$ ，如果 $a$ 和 $S$ 中恰好 $i$ 个家具摆放位置不同
• 那么从状态 $a$ 到达状态 $S$ 的方案数可以用 Task 1 中的矩阵乘法算出，为
• $$\frac{f[T][i]}{C_k^i(n-1{)}^i}$$
• 其中 $C$ 为组合数
• 于是一次询问 $(a,T)$ 答案为
• $$\sum _{i=0}^k\frac{f[T][i]}{C_k^i(n-1{)}^i}\times F[S][k][i]$$
• 总复杂度 $O(\sqrt{P}{k}^3+k^2{n}^k+q{k}^2)$

Code

• 常数巨大，见谅

#include <bits/stdc++.h>

template <class T>
inline void read(T &res)
{
	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);
	if (bo) res = ~res + 1;
}

const int N = 1e6 + 5, L = 4e4 + 5, M = 22, ZZQ = 998244353;

int n, k, q, a[N], lstans = 1, f[N][M], pw1[M], pw[M], C[M][M], p1[M], p2[M],
dw1[M], sum[N][M];

inline void add(int &a, const int &b)
{
	a += b;
	if (a >= ZZQ) a -= ZZQ;
}

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;
}

struct matrix
{
	int n, m, a[M][M];
	
	matrix() {}
	matrix(int _n, int _m) :
		n(_n), m(_m) {memset(a, 0, sizeof(a));}
	
	friend inline matrix operator * (const matrix &a, const matrix &b)
	{
		matrix res = matrix(a.n, b.m);
		for (int i = 1; i <= res.n; i++)
			for (int j = 1; j <= res.m; j++)
				for (int k = 1; k <= a.m; k++)
					add(res.a[i][j], 1ll * a.a[i][k] * b.a[k][j] % ZZQ);
		return res;
	}
} s0[L], s1[L], St, Tr;

int main()
{
	int S, T;
	read(n); read(k); read(q);
	for (int i = 0; i <= k; i++) C[i][0] = 1;
	for (int i = 1; i <= k; i++)
		for (int j = 1; j <= i; j++)
			C[i][j] = (C[i - 1][j] + C[i - 1][j - 1]) % ZZQ;
	pw[0] = pw1[0] = dw1[0] = 1;
	for (int i = 1; i <= k; i++)
	{
		pw[i] = pw[i - 1] * n; pw1[i] = pw1[i - 1] * (n - 1);
		dw1[i] = qpow(pw1[i], ZZQ - 2);
		C[k][i] = qpow(C[k][i], ZZQ - 2);
	}
	for (int i = 0; i < pw[k]; i++) read(a[i]);
	for (int i = 0; i < pw[k]; i++) f[i][0] = a[i];
	for (int i = 1; i <= k; i++)
	{
		if (n == 2) for (int j = i; j >= 1; j--)
			for (int S = 0; S < pw[k]; S++)
				add(f[S][j], f[S ^ (1 << i - 1)][j - 1]);
		else for (int j = 0; j <= i; j++)
			for (int S = 0; S < pw[k]; S++)
			{
				int id = S / pw[i - 1] % n;
				sum[S][j] = f[S][j];
				if (id) add(sum[S][j], sum[S - pw[i - 1]][j]);
				add(f[S][j], sum[S + (n - 1 - id) * pw[i - 1]][j - 1]);
				add(f[S][j], ZZQ - sum[S][j - 1]);
				if (id) add(f[S][j], sum[S - pw[i - 1]][j - 1]);
			}
	}
	s0[0] = Tr = matrix(k + 1, k + 1);
	for (int i = 1; i <= k + 1; i++) s0[0].a[i][i] = 1;
	s1[0] = s0[0];
	for (int i = 0; i <= k; i++)
	{
		if (i) Tr.a[i + 1][i] = (k - i + 1) * (n - 1);
		Tr.a[i + 1][i + 1] = i * (n - 2);
		if (i < k) Tr.a[i + 1][i + 2] = i + 1;
	}
	St = matrix(k + 1, 1); St.a[1][1] = 1;
	for (int i = 1; i <= 33333; i++)
		s0[i] = s0[i - 1] * Tr;
	for (int i = 1; i <= 33333; i++)
		s1[i] = s1[i - 1] * s0[33333];
	while (q--)
	{
		read(S); read(T);
		T = 1ll * T * lstans % ZZQ;
		matrix A = s1[T / 33333] * (s0[T % 33333] * St);
		lstans = 0;
		for (int i = 0; i <= k; i++)
			add(lstans, 1ll * A.a[i + 1][1] * dw1[i] % ZZQ * C[k][i] % ZZQ
			* f[S][i] % ZZQ);
		printf("%d\n", lstans);
	}
	return 0;
}