【bzoj3125】CITY 插头dp

f[i][j][S]表示走到第i行第j列状态为S的方案数
0表示没有插头，1表示左括号，2表示右括号

好像插头dp都差不多是这样的？

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<iostream>
#define maxn 600010 

using namespace std;

int head[60010],to[maxn],next[maxn],hash[2][maxn],tot[2];
int a[20][20],bit[50];
long long f[2][maxn];
int n,m,now,pre,num,xx,yy;
char s[21];
long long ans;

void add(int s,long long x)
{
	int pos=s%60000;
	for (int p=head[pos];p;p=next[p])
	  if (hash[now][to[p]]==s)
	  {
	  	f[now][to[p]]+=x;
	  	return;
	  }
	tot[now]++;
	hash[now][tot[now]]=s;
	f[now][tot[now]]=x;
	to[++num]=tot[now];next[num]=head[pos];head[pos]=num;
}

int find_pos(int s,int i)
{
	return (s/(1<<bit[i-1]))%4;
}

int find_l(int s,int k)
{
	int cnt=0;
	for (int i=k;i>=1;i--)
	{
		int p=find_pos(s,i);
		if (p==2) cnt++;
		else if (p==1) cnt--;
		if (!cnt) return i;
	}
}

int find_r(int s,int k)
{
	int cnt=0;
	for (int i=k;i<=m+1;i++)
	{
		int p=find_pos(s,i);
		if (p==1) cnt++;
		else if (p==2) cnt--;
		if (!cnt) return i;
	}
}

void dp()
{
	now=1;pre=0;
	tot[now]=1;
	hash[now][1]=0;
	f[now][1]=1;
	for (int i=1;i<=n;i++)
	{
		for (int j=1;j<=tot[now];j++) hash[now][j]<<=2;
		for (int j=1;j<=m;j++)
		{
			swap(now,pre);
			num=0;
			tot[now]=0;
			memset(f[now],0,sizeof(f[now]));
			memset(head,0,sizeof(head));
			for (int k=1;k<=tot[pre];k++)
			{
				int s=hash[pre][k];
				long long num=f[pre][k];
				if (!num) continue;
				int p=find_pos(s,j),q=find_pos(s,j+1);
				if (!a[i][j])
				{
					if (!p && !q) add(s,num);
				}
				else if (a[i][j]==2)
				{
					if (p==1 && !q) add(s-(1<<bit[j-1])+(1<<bit[j]),num);
					if (p==2 && !q) add(s-(1<<bit[j-1]+1)+(1<<bit[j]+1),num);
				}
				else if (a[i][j]==3)
				{
					if (!p && q==1) add(s-(1<<bit[j])+(1<<bit[j-1]),num);
					if (!p && q==2) add(s-(1<<bit[j]+1)+(1<<bit[j-1]+1),num);
				}
				else 
				{
					if (p && q)
					{
						if (p==1 && q==1)
						{
							int r=find_r(s,j+1);
							add(s-(1<<bit[j-1])-(1<<bit[j])-(1<<bit[r-1]+1)+(1<<bit[r-1]),num);
						}
						else if (p==2 && q==2)
						{
							int l=find_l(s,j);
							add(s-(1<<bit[j-1]+1)-(1<<bit[j]+1)-(1<<bit[l-1])+(1<<bit[l-1]+1),num);
						}
						else if (p==2 && q==1)
						{
							add(s-(1<<bit[j-1]+1)-(1<<bit[j]),num);
						}
						else
						{
							s-=(1<<bit[j-1])+(1<<bit[j]+1);
							if (xx==i && yy==j) add(s,num);
						}
					}
					else if (p)
					{
						if (p==1)
						{
							s-=(1<<bit[j-1]);
							if (j<m) add(s+(1<<bit[j]),num);
							if (i<n) add(s+(1<<bit[j-1]),num);
						}
						else
						{
							s-=(1<<bit[j-1]+1);
							if (j<m) add(s+(1<<bit[j]+1),num);
							if (i<n) add(s+(1<<bit[j-1]+1),num);
						}
					}
					else if (q)
					{
						if (q==1)
						{
							s-=(1<<bit[j]);
							if (j<m) add(s+(1<<bit[j]),num);
							if (i<n) add(s+(1<<bit[j-1]),num);
						}
						else
						{
							s-=(1<<bit[j]+1);
							if (j<m) add(s+(1<<bit[j]+1),num);
							if (i<n) add(s+(1<<bit[j-1]+1),num);
						}
					}
					else
					{
						if (i<n && j<m) add(s+(1<<bit[j-1])+(1<<bit[j]+1),num);
					}
				}
			}
		}
	}
}

int main()
{
	for (int i=1;i<=20;i++) bit[i]=i*2;
	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;
	      else if (s[j]=='-') a[i][j]=2;
	      else if (s[j]=='|') a[i][j]=3;
	}
	for (int i=1;i<=n;i++)
	  for (int j=1;j<=m;j++)
	  {
	    if (a[i][j]==2 && (j==1 || j==m)) {printf("0\n");return 0;}
	    if (a[i][j]==3 && (i==1 || i==n)) {printf("0\n");return 0;}
	    if (a[i][j]==1) xx=i,yy=j;
	  }
	dp();
	for (int i=1;i<=tot[now];i++) if (hash[now][i]==0) {printf("%lld\n",f[now][i]);return 0;}
	printf("0\n");
	return 0;
}