牛客第二场 E MAZE —— 线段树 + 矩阵

最新推荐文章于 2021-10-27 16:22:01 发布

原创最新推荐文章于 2021-10-27 16:22:01 发布 · 317 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#牛客第二场 # E MAZE #线段树 #矩阵

ACM - 线段树专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种结合矩阵动态规划(matrix DP)与线段树(segment tree)的算法，用于解决一类路径计数问题，特别是在二维网格中进行多次查询与更新的场景下。通过将相邻行的状态转移视为矩阵乘法，利用线段树高效地维护状态转移矩阵，实现了O(qm^3logn)的时间复杂度。

题目链接：点我啊╭(╯^╰)╮

题目大意：

    一张 $n \times m$ 的图， $0$ 表示可走， $1$ 不可走
    只能向下、左右走，不能回走
    多次查询，两种操作：
    将某个位置反转
    问从第一行 $a_i$ 到第 $n$ 行 $b_i$ 的的方案数

解题思路：

$d p [i] [j] = s u m (d p [i - 1] [k]$ for $(k < j$ and $b_{ik}=b_{ik+1}=...=b_{ij}=0))+$
$s u m (d p [i - 1] [k]$ for $(k > j$ and $b_{ik}=b_{ik-1}=...=b_{ij}=0))$

    所以相邻两行就可以用矩阵相乘计算
    然后用线段树维护
    每个节点表示一段区间的系数转移矩阵
    时间复杂度： $O(qm^3logn)$
    详细可看：这里

核心：线段树维护系数矩阵

#include<bits/stdc++.h>
#define rint register int
#define deb(x) cerr<<#x<<" = "<<(x)<<'\n';
using namespace std;
typedef long long ll;
typedef pair <int,int> pii;
const ll mod = 1e9 + 7;
const int maxn = 5e4 + 5;
int n, m, q, op;
int b[maxn][15];
struct node{
	ll s[15][15];
	void init() {memset(s, 0, sizeof(s));}
	node operator + (const node &a) const {
		node ret; ret.init();
		for(int i=1; i<=m; i++)
			for(int j=1; j<=m; j++)
				for(int k=1; k<=m; k++)
					ret.s[i][j] = (ret.s[i][j]+s[i][k]*a.s[k][j]%mod) % mod;
		return ret;
	}
} t[maxn<<2];  

void get(node &h, int r){
	h.init();
	for(int i=1; i<=m; i++)
		if(!b[r][i]){
			h.s[i][i] = 1;
			for(int j=i-1; j && !b[r][j]; j--) h.s[j][i] = 1;
			for(int j=i+1; j<=m && !b[r][j]; j++) h.s[j][i] = 1;
		}
}

void build(int l, int r, int rt){
	if(l == r){
		get(t[rt], l);
		return;
	}
	int m = l + r >> 1;
	build(l, m, rt<<1);
	build(m+1, r, rt<<1|1);
	t[rt] = t[rt<<1] + t[rt<<1|1];
}
void update(int pos, int l, int r, int rt){
	if(l>pos || r<pos) return;
	if(l == r){
		get(t[rt], l);
		return;
	}
	int m = l + r >> 1;
	update(pos, l, m, rt<<1);
	update(pos, m+1, r, rt<<1|1);
	t[rt] = t[rt<<1] + t[rt<<1|1];
}
int main() {
	scanf("%d%d%d", &n, &m, &q);
	for(int i=1; i<=n; i++){
		char s[15];
		scanf("%s", s+1);
		for(int j=1; j<=m; j++)
			if(s[j]=='1') b[i][j] = 1;
	}
	build(1, n, 1);
	while(q--){
		int u, v;
		scanf("%d%d%d", &op, &u, &v);
		if(op & 1) b[u][v] ^= 1, update(u, 1, n, 1);
		else printf("%d\n", t[1].s[u][v]);
	}
}