牛客多校训练营2 E.MAZE(线段树+矩阵)

原题地址:https://ac.nowcoder.com/acm/contest/882/E

题意:给出一个矩阵,0表示可走,1表示不可走,给出起点和终点,问从起点到终点的方案数。

基本都是看了大佬的博客

思路:主要就是对矩阵的理解。他每一行的dp状态都是可以通过上一行的状态转移过来了,所以就可以利用矩阵来转移。有n行,所以就会有n-1个矩阵,答案就是将这n-1个矩阵乘起来,然后取值就可以。

所以这里如果没有修改操作,完全就不需要线段树。
所以因为有修改操作,才需要用上线段树。因为修改操作是单点修改,所以每次最多只会影响一个矩阵,放到线段树里面也就是最多影响log个矩阵。

#include <bits/stdc++.h>
#define eps 1e-8
#define INF 0x3f3f3f3f
#define PI acos(-1)
#define lson l,mid,rt<<1
#define rson mid+1,r,(rt<<1)+1
#define CLR(x,y) memset((x),y,sizeof(x))
#define fuck(x) cerr << #x << "=" << x << endl
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int seed = 131;
const int maxn = 1e5 + 5;
const int mod = 1e9 + 7;
int n, m, q;
char str[maxn];
int a[50005][105];//记录初始矩阵,用于构建矩阵
const int SZ = 10;
struct mat {//SZ是矩阵大小
    ll m[SZ][SZ];
    mat operator *(const mat a)const {
        mat ans;
        for (int i = 0; i < SZ; i++) {
            for (int j = 0; j < SZ; j++) {
                ll sum = 0;
                for (int k = 0; k < SZ; k++) {
                    sum = (sum + m[i][k] * a.m[k][j]) % mod;
                }
                ans.m[i][j] = sum;
            }
        }
        return ans;
    }
} mt[maxn << 2];
void push_up(int rt) {
    mt[rt] = mt[rt << 1] * mt[rt << 1 | 1];
}
void mat_upd(int rt, int x) { //第x行,初始化构造的矩阵
    CLR(mt[rt].m, 0);
    //这边的第i个是矩阵的第i列
    for (int i = 1; i <= m; i++) { //第i列
        if (a[x][i]) continue;
        mt[rt].m[i][i] = 1;
        for (int j = i + 1; j <= m; j++) {
            if (a[x][j]) break;
            mt[rt].m[j][i] = 1;
        }
        for (int j = i - 1; j >= 1; j--) {
            if (a[x][j]) break;
            mt[rt].m[j][i] = 1;
        }
    }
}

void build(int l, int r, int rt) {
    if (l == r) {
        mat_upd(rt, l);
        return ;
    }
    int mid = (l + r) / 2;
    build(lson);
    build(rson);
    push_up(rt);
}
void update(int l, int r, int rt, int x, int y) {
    if (l == r) {
        mat_upd(rt, l);
        return;
    }
    int mid = (l + r) / 2;
    if (mid >= x) update(lson, x, y);
    else update(rson, x, y);
    push_up(rt);
}
int main() {
    scanf("%d%d%d", &n, &m, &q);
    for (int i = 1; i <= n; i++) {
        scanf("%s", str + 1);
        for (int j = 1; j <= m; j++) {
            a[i][j] = str[j] - '0';
        }
    }
    build(1, n, 1);
    while (q--) {
        int op, l, r;
        scanf("%d%d%d", &op, &l, &r);
        if (op == 1) {
            a[l][r] ^= 1;
            update(1, n, 1, l, r);
        } else {
            printf("%lld\n", mt[1].m[l][r]);
        }
    }
    return 0;
}



  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值