线性代数三之状压DP的矩阵加速——Quad Tiling，Bus公交线路

最新推荐文章于 2021-11-13 16:59:53 发布

ikrvxt

最新推荐文章于 2021-11-13 16:59:53 发布

阅读量138

点赞数 1

分类专栏： # 状压DP 数学 # 矩阵加速文章标签：状压DP 矩阵加速线性代数

本文链接：https://blog.csdn.net/Emm_Titan/article/details/118977367

版权

数学同时被 3 个专栏收录

46 篇文章 1 订阅

订阅专栏

状压DP

18 篇文章 0 订阅

订阅专栏

矩阵加速

13 篇文章 0 订阅

订阅专栏

这篇博客探讨了如何利用矩阵快速幂运算解决动态规划问题，以QuadTiling为例，详细解释了状态转移并展示了代码实现。同时，还介绍了另一个问题——公交线路的访问状态优化，同样使用矩阵加速求解，给出了核心思路和代码片段。

摘要由CSDN通过智能技术生成

Quad Tiling

description

solution

设 $dp_{i,S}:$ $i$ 列的状态为 $S$ 的方案数，最后答案为 $dp_{n,(1<<4)-1}$ （最后一列刚好铺满）

$\begin{bmatrix} 0\\ 0\\ 0\\ 0 \end{bmatrix} \rightarrow \begin{bmatrix} 1\\ 0\\ 0\\ 1\\ \end{bmatrix} / \begin{bmatrix} 0\\ 0\\ 0\\ 0\\ \end{bmatrix} / \begin{bmatrix} 0\\ 0\\ 1\\ 1\\ \end{bmatrix} / \begin{bmatrix} 1\\ 1\\ 1\\ 1\\ \end{bmatrix} / \begin{bmatrix} 1\\ 1\\ 0\\ 0\\ \end{bmatrix}$
$\begin{bmatrix} 1\\ 1\\ 1\\ 1\\ \end{bmatrix} \rightarrow \begin{bmatrix} 0\\ 0\\ 0\\ 0\\ \end{bmatrix}$
$\begin{bmatrix} 1\\ 0\\ 0\\ 1 \end{bmatrix} \rightarrow \begin{bmatrix} 0\\ 1\\ 1\\ 0\\ \end{bmatrix} / \begin{bmatrix} 0\\ 0\\ 0\\ 0\\ \end{bmatrix}$
$\begin{bmatrix} 0\\ 1\\ 1\\ 0 \end{bmatrix} \rightarrow \begin{bmatrix} 1\\ 0\\ 0\\ 1\\ \end{bmatrix}$
$\begin{bmatrix} 1\\ 1\\ 0\\ 0 \end{bmatrix} \rightarrow \begin{bmatrix} 0\\ 0\\ 1\\ 1\\ \end{bmatrix} / \begin{bmatrix} 0\\ 0\\ 0\\ 0\\ \end{bmatrix}$
$\begin{bmatrix} 0\\ 0\\ 1\\ 1\\ \end{bmatrix} \rightarrow \begin{bmatrix} 1\\ 1\\ 0\\ 0\\ \end{bmatrix} / \begin{bmatrix} 0\\ 0\\ 0\\ 0\\ \end{bmatrix}$
$1010\rightarrow 0101...$ 等的其余状态都是不可能最后刚好 $n$ 被多米诺骨牌铺满而不凸出去几格的

在这里插入图片描述

code

#include <cstdio>
#include <cstring>
#define int long long
int n, mod;
struct matrix {
	int n, m;
	int c[16][16];
	matrix() {
		memset( c, 0, sizeof( c ) );
	}
	matrix operator * ( matrix &t ) {
		matrix ans;
		ans.n = n, ans.m = t.m;
		for( int i = 0;i < n;i ++ )
			for( int j = 0;j < t.m;j ++ )
				for( int k = 0;k < m;k ++ )
					ans.c[i][j] = ( ans.c[i][j] + c[i][k] * t.c[k][j] ) % mod;
		return ans;
	}
}g, ret;
int id[2][2][2][2];

matrix qkpow( matrix x, int y ) {
	matrix ans;
	ans.n = ans.m = x.m;
	for( int i = 0;i < ans.n;i ++ )
		ans.c[i][i] = 1;
	while( y ) {
		if( y & 1 ) ans = ans * x;
		x = x * x;
		y >>= 1;
	}
	return ans;
}

void rebuild() {
	memset( g.c, 0, sizeof( g.c ) );
	memset( ret.c, 0, sizeof( ret.c ) );
	g.n = g.m = ret.m = 1 << 4, ret.n = 1;
	g.c[id[0][0][0][0]][id[1][1][0][0]] = 1;
	g.c[id[0][0][0][0]][id[0][0][1][1]] = 1;
	g.c[id[0][0][0][0]][id[1][0][0][1]] = 1;
	g.c[id[0][0][0][0]][id[1][1][1][1]] = 1;
	g.c[id[0][0][0][0]][id[0][0][0][0]] = 1;
	g.c[id[0][0][1][1]][id[1][1][0][0]] = 1;
	g.c[id[0][0][1][1]][id[0][0][0][0]] = 1;
	g.c[id[1][1][0][0]][id[0][0][0][0]] = 1;
	g.c[id[1][1][0][0]][id[0][0][1][1]] = 1;
	g.c[id[1][0][0][1]][id[0][0][0][0]] = 1;
	g.c[id[1][0][0][1]][id[0][1][1][0]] = 1;
	g.c[id[0][1][1][0]][id[1][0][0][1]] = 1;
	g.c[id[1][1][1][1]][id[0][0][0][0]] = 1;
	ret.c[0][id[0][0][0][0]] = 1;
}

signed main() {
	for( int i = 0;i < 2;i ++ )
		for( int j = 0;j < 2;j ++ )
			for( int k = 0;k < 2;k ++ )
				for( int w = 0;w < 2;w ++ )
					id[i][j][k][w] = w + ( 1 << 1 ) * k + ( 1 << 2 ) * j + ( 1 << 3 ) * i;
	while( scanf( "%lld %lld", &n, &mod ) ) {
		if( ! n && ! mod ) return 0;
		rebuild();
		g = qkpow( g, n );
		ret = ret * g;
		printf( "%lld\n", ret.c[0][id[0][0][0][0]] );
	}
	return 0;
}

[Hnoi2010]Bus 公交线路

description

solution

设 $dp_{i,S}:$ 前 $i - 1$ 个站都已经被有且访问过一次，然后 $[i, i + p - 1]$ 的站的访问状态为 $S$

（ $S$ 从高到低第 $j$ 位代表 $i + j - 1$ 车站的访问状态）

$dp_{i,S}\rightarrow dp_{i+1,S'}:$ 必须满足 $S$ 的最高位为 $1$ 且车站访问状态翻译过来是一样的

也就是说原来 $S$ 的第 $j$ 位，变成了 $S^{'}$ 中的第 $j - 1$ 位

枚举下一个被访问的车站，要满足该车站的状态原来是 $0$

最后套矩阵加速

code

#include <cstdio>
#include <cstring>
#define mod 30031
struct matrix {
	int n, m;
	int c[130][130];
	matrix() {
		memset( c, 0, sizeof( c ) );
	}
	matrix operator * ( matrix &t ) {
		matrix ans;
		ans.n = n, ans.m = t.m;
		for( int i = 1;i <= n;i ++ )
			for( int j = 1;j <= t.m;j ++ )
				for( int k = 1;k <= m;k ++ )
					ans.c[i][j] = ( ans.c[i][j] + c[i][k] * t.c[k][j] ) % mod;
		return ans;
	}
}g, ret;
int n, k, p, tot;
int s[130];

matrix qkpow( matrix x, int y ) {
	matrix ans;
	ans.n = ans.m = x.n;
	for( int i = 1;i <= ans.n;i ++ )
		ans.c[i][i] = 1;
	while( y ) {
		if( y & 1 ) ans = ans * x;
		x = x * x;
		y >>= 1;
	}
	return ans;
}

int main() {
	scanf( "%d %d %d", &n, &k, &p );
	for( int i = ( 1 << p - 1 );i < ( 1 << p );i ++ ) {
		int cnt = 0;
		for( int j = 0;j < p;j ++ )
			if( 1 << j & i ) cnt ++;
		if( cnt == k ) s[++ tot] = i;
	}
	g.n = g.m = ret.m = tot, ret.n = 1;
	ret.c[1][tot] = 1;
	for( int i = 1;i <= tot;i ++ )
		for( int j = 1;j <= tot;j ++ ) {
			int New = s[i] - ( 1 << p - 1 ) << 1;
			for( int w = 0;w < p;w ++ )
				if( ! ( 1 << w & New ) && New + ( 1 << w ) == s[j] ) {
					g.c[i][j] = 1;
					break;
				}
		}
	g = qkpow( g, n - k );
	ret = ret * g;
	printf( "%d\n", ret.c[1][tot] );
	return 0;
}