Bzoj2331[SCOI2011]地板:插頭dp

題目鏈接:[SCOI2011]地板

裸插頭,知道做了這道題才發現我以前插頭學得是個什麼XX樣子啊QAQ

用0表示沒有插頭,1:插頭類型爲直線,2:插頭類型爲L

具體情況都在程序中標註

順便吐槽一句,這道題我wa了千百遍居然只是因爲數據中'_'寫成了'*'!白白浪費了一個上午TAT

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<time.h>
#define LL long long
using namespace std;
const int maxn=2111111;
const int mod=99997;
const int MO=20110520;
const int maxnm=110;
int n,m,a[maxnm][maxnm],b[maxnm][maxnm];
char s[maxnm];
int code[maxnm];

struct hashset{
    int next[maxn],h[mod],tot;
    LL As[maxn]; int val[maxn];
    void push(LL x,int v){
        int now=x%mod; if (now<0) now+=mod;
        for (int i=h[now];i;i=next[i])
        if(As[i]==x){val[i]+=v;if(val[i]>MO)val[i]-=MO;return;}
        ++tot; next[tot]=h[now]; h[now]=tot;
        val[tot]=v; As[tot]=x; return;
    }
    void init(){
        memset(h,0,sizeof(h)); tot=0;
    }
}dp[2];

void decode(LL st){
    for (int i=m;i>=0;--i) code[i]=st&3,st>>=2;
}

LL encode(){
    int cnt=0; LL st=0;
    for (int i=0;i<=m;++i)
        st<<=2,st|=code[i];
    return st;
}

void shit(){
    for (int i=m;i>=1;--i) code[i]=code[i-1];
    code[0]=0;
}
//0沒有插頭,1:插頭類型爲直線,2:插頭類型爲L
void dpblock(int i,int j,int cur){
    for (int k=1;k<=dp[cur].tot;++k){
        decode(dp[cur].As[k]);
        if(j==1){if(code[m])continue;shit();}
        int up=code[j],left=code[j-1];
        if(!up&&!left)dp[cur^1].push(encode(),dp[cur].val[k]);
    }
}

void dpright(int i,int j,int cur){
    for (int k=1;k<=dp[cur].tot;++k){
        decode(dp[cur].As[k]);
        if(j==1){if(code[m])continue;shit();}
        int up=code[j],left=code[j-1];
        if(!up&&!left){
            if (a[i+1][j]&&a[i][j+1]){//製造一個拐點
                code[j]=code[j-1]=2;
                dp[cur^1].push(encode(),dp[cur].val[k]);
            }if (a[i+1][j]){//製造一條線
                code[j-1]=1; code[j]=0;
                dp[cur^1].push(encode(),dp[cur].val[k]);
            }if (a[i][j+1]){//製造一條線
                code[j]=1; code[j-1]=0;
                dp[cur^1].push(encode(),dp[cur].val[k]);
            }
        }else if ((!up&&left)||(!left&&up)){
            if (left&&!up){
                if (a[i][j+1]){//延伸
                    code[j-1]=0; code[j]=left;
                    dp[cur^1].push(encode(),dp[cur].val[k]);
                }if (a[i+1][j]&&left==1){//轉彎
                    code[j-1]=2; code[j]=0;
                    dp[cur^1].push(encode(),dp[cur].val[k]);
                }if (left==2){//停止
                    code[j]=code[j-1]=0;
                    dp[cur^1].push(encode(),dp[cur].val[k]);
                }
            }else if (!left&&up){
                if (a[i][j+1]&&up==1){//轉彎
                    code[j-1]=0; code[j]=2;
                    dp[cur^1].push(encode(),dp[cur].val[k]);
                }if (a[i+1][j]){//延伸
                    code[j-1]=up; code[j]=0;
                    dp[cur^1].push(encode(),dp[cur].val[k]);
                }if (up==2){//停止
                    code[j]=code[j-1]=0;
                    dp[cur^1].push(encode(),dp[cur].val[k]);
                }
            }
        }else if (up&&left){
            if (up==left&&up==1){
                code[j]=code[j-1]=0;
                dp[cur^1].push(encode(),dp[cur].val[k]);
            }else{}//不合法
        }
    }
}

bool check(LL st){
    decode(st);
    for (int i=0;i<=m;++i) if (code[i]==1) return 0;
    return 1;
}

void solve_DP(){
    int cur=0; dp[0].init(); dp[0].push(0,1);
    for (int i=1;i<=n;++i)
        for (int j=1;j<=m;++j){
            dp[cur^1].init();
            if (a[i][j]) dpright(i,j,cur);
            else dpblock(i,j,cur);
            cur^=1;
        }
    LL ans=0;
    for (int i=1;i<=dp[cur].tot;++i)
        if (check(dp[cur].As[i])) ans+=dp[cur].val[i],ans%=MO;
    printf("%lld\n",ans<0?ans+MO:ans);
    /*if (check(dp[cur].As[1])) printf("%d\n",dp[cur].val[1]);
    else printf("0");*/
}

int main(){
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;++i){
        scanf("%s",s+1);
        for (int j=1;j<=m;++j)
            if (s[j]=='_') a[i][j]=1;
    }
    if (n<m){
        for (int i=1;i<=m;++i)
            for (int j=1;j<=n;++j)
                b[i][j]=a[j][i];
        swap(n,m);
        for (int i=1;i<=n;++i)
            for (int j=1;j<=m;++j)
                a[i][j]=b[i][j];
    }
    solve_DP();
}



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章