【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;
}