2019牛客暑期多校训练营(第二场)E.MAZE(线段树+dp)

题意:给你一个n*m的矩阵 你只能向左向右相下走 有两种操作 q次询问 一种是把一个单位翻转(即可走变为不可走 不可走变为可走) 另一种是询问从(1,x) 走到 (n,y)有多少种方案

思路:题目n为1e5 而m只有10 我们可以考虑在线段树上维护一个m*m的矩阵 当模拟矩阵乘法的时候等效于计算方案数(自己手动模拟一下) 修改操作就相当于单点更新 然后重新构造矩阵

#include <bits/stdc++.h>
using namespace std;
const int N = 5e4+7;
const int inf = 0x3f3f3f3f;
typedef long long ll;
const ll mod = 1e9+7;
ll a[N][12];
int n,m,q;
struct matrix{
    int l,r;
    ll ma[12][12];
};
matrix t[N<<2];
void pushup(int p){
    memset(t[p].ma,0,sizeof(t[p].ma));
    for(int i=1;i<=m;i++)
        for(int j=1;j<=m;j++)
            for(int k=1;k<=m;k++){
                t[p].ma[i][j]=(t[p].ma[i][j]+((t[p<<1].ma[i][k]%mod)*(t[p<<1|1].ma[k][j]%mod))%mod)%mod;   
            }
}
void work(int p,int l){
    memset(t[p].ma,0,sizeof(t[p].ma));
    for(int i=1;i<=m;i++){
        int pos=i;
        while(pos>=1&&a[l][pos]==0){
            t[p].ma[i][pos]=1;
            pos--;
        }
        pos=i;
        while(pos<=m&&a[l][pos]==0){
            t[p].ma[i][pos]=1;
            pos++;
        }
    }
}
void build(int p,int l,int r){
    t[p].l=l; t[p].r=r;
    if(l==r){
        work(p,l);
        return ;
    }
    int mid=(l+r)>>1;
    build(p<<1,l,mid);
    build(p<<1|1,mid+1,r);
    pushup(p);
}
void update(int p,int x){
    if(t[p].l==t[p].r){
        work(p,t[p].l);
        return ;
    }
    int mid=(t[p].l+t[p].r)>>1;
    if(x<=mid) update(p<<1,x);
    else update(p<<1|1,x);
    pushup(p);
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    cin>>n>>m>>q;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            char b; cin>>b;
            a[i][j]=(b!='0');
        }
    }   
    build(1,1,n);
    for(int i=1;i<=q;i++){
        int z,x,y; cin>>z>>x>>y;
        if(z==1){
            a[x][y]^=1;
            update(1,x);
        }else{
            cout<<t[1].ma[x][y]<<"\n";
        }
    }
    return 0;
}

 

posted @ 2019-07-31 09:38  WAKBGAN  阅读(134)  评论(0编辑  收藏  举报