【Bzoj1187】神奇游乐园（插头DP）

#include <cstdio>
#include <algorithm>
#define N 10
#define Inf 0x7fffffff
using namespace std;

int A[N],now,op[N],k[N],L[N],Ans=-Inf;
int n,m,g[110][N],dp[2][20000],q[2][20000],t[2];

inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void push(int x,int y){
    if(dp[now^1][x]==-Inf) q[now^1][++t[now^1]]=x;
    dp[now^1][x]=max(dp[now^1][x],y);
}

void clear(){
    for(int i=1;i<=t[now];++i) dp[now][q[now][i]]=-Inf;
    t[now]=0;
    now^=1;
}

void f(int S){
    int tp=0;
    for(int i=0;i<=m;++i){
        k[i]=S&3,S>>=2;//
        if(k[i]==1) L[++tp]=i;
        else if(k[i]==2) op[op[i]=L[tp--]]=i;
    }
}

void bfs(int x,int y){
    clear();
    for(int i=1;i<=t[now];++i){
        int S=q[now][i];
        f(S);
        int l=k[y-1],r=k[y],upd=dp[now][S]+g[x][y];
        if(!l&&!r){
            push(S,upd-g[x][y]);
            push(S+A[y-1]+(A[y]<<1),upd);
        }else if(l&&r){   
            int tS=S-l*A[y-1]-r*A[y];
            if(l==2&&r==1) push(tS,upd);
            else if(l==1&&r==1) push(tS-A[op[y]],upd);
            else if(l==2&&r==2) push(tS+A[op[y-1]],upd);
            else if(!tS) Ans=max(Ans,upd);
        }
        else{
            push(S,upd);
            push(S-l*A[y-1]-r*A[y]+l*A[y]+r*A[y-1],upd);
        }           
    }
}

int main(){
    A[0]=1;for(int i=1;i<N;++i)A[i]=A[i-1]<<2;//四进制
    for(int i=0;i<2;++i)for(int j=0;j<20000;++j)dp[i][j]=-Inf;
    n=read(),m=read();
    for(int i=1;i<=n;++i)for(int j=1;j<=m;++j)g[i][j]=read();
    push(0,0);
    for(int i=1;i<=n;++i){
        clear();
        for(int j=1;j<=t[now];++j) if(q[now][j]<A[m]) push(q[now][j]<<2,dp[now][q[now][j]]);
        for(int j=1;j<=m;++j) bfs(i,j);
    }
    printf("%d\n",Ans);
    return 0;
}