[BZOJ1187][HNOI2007] 神奇游乐园 - 插头DP

这么裸的题调得我一脸懵逼 - - 我太弱啦

#include"bits/stdc++.h"
#define F(i,l,r) for(register int i=l;i<=r;i++)
#define readi(x) scanf("%d",&x)
using namespace std;
const int N=105,M=7,S=3005,inf=0x3f3f3f3e;
int dp[N][M][S],n,m,w[N][M],lim,st[S],rk[2<<2*M+2],cnt[2<<2*M],ans=-inf;
void build_status(){
	F(i,0,(1<<2*m+2)-1){
		int p=i,s=0;
		bool flag=true;
		while (p){
			if(p%4==3){flag=false;break;}
			if(p%4==1)s++,cnt[i]++;if(p%4==2)s--;
			p>>=2;if(s<0){flag=false;break;}
		}
		if(s)flag=false;
		if(flag)st[++lim]=i,rk[i]=lim;
	} 
}
int main(){
	readi(n),readi(m);build_status();
	F(i,1,n)F(j,1,m)readi(w[i][j]);
	F(i,1,n)F(j,0,m)F(k,1,lim)dp[i][j][k]=-inf;
	dp[1][0][1]=0;
	F(i,1,n){
		F(j,0,m-1)F(k,1,lim)if(dp[i][j][k]>-inf){
			int s1=(st[k]>>2*j)&3; //左插头
			int s2=(st[k]>>2*j+2)&3; //上插头
			if (s1==1 && s2==2){ //插头组成回路 : 若只有一条回路记录答案 多条回路直接continue 
				if (cnt[st[k]]==1) ans=max(ans,dp[i][j][k]+w[i][j+1]);
				else continue;
			} else if (s1==2 && s2==1){ //两区间合并 
				int p=rk[st[k]-(s1<<2*j)-(s2<<2*j+2)];
				dp[i][j+1][p]=max(dp[i][j+1][p],dp[i][j][k]+w[i][j+1]);
			} else if (!s1 && !s2){ //空的 
				int p1=k,p2=rk[st[k]+(1<<2*j)+(2<<2*j+2)];
				dp[i][j+1][p1]=max(dp[i][j+1][p1],dp[i][j][k]);
				dp[i][j+1][p2]=max(dp[i][j+1][p2],dp[i][j][k]+w[i][j+1]);
			} else if (!s1 || !s2){ //一端为空 
				int p1=k,p2=s1?rk[st[k]+(3*s1<<2*j)]:rk[st[k]-(3*s2<<2*j)];
				dp[i][j+1][p1]=max(dp[i][j+1][p1],dp[i][j][k]+w[i][j+1]);
				dp[i][j+1][p2]=max(dp[i][j+1][p2],dp[i][j][k]+w[i][j+1]);
			} else { //均为1或均为2 
				if (s1&1){ //上插头的对应插头变成1  
					int p=j+2;while(((st[k]>>2*p)&3)!=2)p++;
					p=rk[st[k]-(1<<2*p)-(s1<<2*j)-(s2<<2*j+2)];
					dp[i][j+1][p]=max(dp[i][j+1][p],dp[i][j][k]+w[i][j+1]);
				} else { //左插头的对应插头变成2 
					int p=j-1;while(((st[k]>>2*p)&3)!=1)p--;
					p=rk[st[k]+(1<<2*p)-(s1<<2*j)-(s2<<2*j+2)];
					dp[i][j+1][p]=max(dp[i][j+1][p],dp[i][j][k]+w[i][j+1]);
				}
			}
		}
		if(i!=n)F(k,1,lim)if(rk[st[k]<<2])
			dp[i+1][0][rk[st[k]<<2]]=max(dp[i+1][1][rk[st[k]<<2]],dp[i][m][k]);
	}
	cout << ans << endl;
	return 0;
}