简介
插头DP(轮廓线DP)是用来解决网格图回路问题的一种算法。
插头DP解决的经典问题就是统计经过所有格子的哈密顿回路条数,某些格子有障碍。
如果问题稍微进阶一点的话,不一定要求路径是回路、路径带权等等情况都可能出现。
它的时间复杂度比较高,但是已经属于比较高效的算法了。
基本概念
首先看经典问题:统计一个带障碍的(n*m)网格中((n,mle 12)),经过所有格子的哈密顿回路的条数。
插头DP的状态围绕轮廓线进行转移。
我们的状态是(f(i,j,state)),表示((i,j))格子转移完成后、轮廓线状态为(state)时的情况数。
轮廓线的形状相对于每一个((i,j))是确定的。((i,j))转移之前到转移之后,轮廓线的形状、变化如图所示:
即转移的格子((i,j))原来是轮廓线上的一个凸出点,转移后把轮廓线“从左上往右下拉”,使得((i,j))变成一个凹格。
轮廓线的长度为(m+1),我们需要记录轮廓线的每一条边的上方(对于轮廓线中打竖的那条边,则是左方),是否有边作为接口,即是否有插头。这个信息,我们存在(state)中。
轮廓线所体现的,一是轮廓线上的插头状态,告诉你每一个地方是否应该用一条路径“接上”;二是所有路径的连通性:显然,路径是两两相连,两两成对的,我们分组标号来记录。下面用几幅图来表达一下轮廓线的记录方式:
括号的每个数代表对应轮廓线边的状态,"0"表示没有路径连接,其他数表示一对路径,每对路径的标号一样。
考虑到标号的大小没有保证,会导致状态的储存十分困难。
我们另寻它法:注意到每对路径在轮廓线上的接口不会相交,即不会出现((...,x,...,y,...x,...y,...))的这种诡异情况。所以我们可以用括号序列来表示路径信息。还是上述三个例子:
这样一来,状态(state)可以看做一个(m+1)位的3进制数:0表示无接口,1表示一对路径的左端,2表示一对路径的右端。
转移
(f(i,j,state))的转移来源,是(f(i,j-1,state')),但这样不好考虑和枚举。我们用(f(i,j,state)),转移到(f(i,j+1,state'))。
我们将(f(i,j,sum state))转移到(f(i,j+1,sum state'))称作大转移。
显然,我们只需要关注轮廓线上唯一变动的两条边:我们把轮廓线从左上拉到右下,这两条边的状态会更改,更改后是什么呢?取决于新的一格内路径怎么走:
我们现在关注转移的位点:
大转移的整体步骤是:枚举每一个状态(state),得到(p_a)和(p_b),根据(p_a)和(p_b)的取值,枚举((i,j))的路径走法,对于选择的((i,j))的走法,由(state)修改(p_a)和(p_b)分别变成(p_a')和(p_b')而得到新状态(state'),执行(f(i,j+1,state')+=f(i,j,state))。
根据(p_a,p_b)枚举走法分三大类情况:
(1)(p_a=0;,p_b=0):
这表明((i,j))的左方和上方没有路径连接,则当前格只能选取1号走法。
令(p_a'=1,p_b'=2)得到新状态。(转移前提:(且i<n且j<m),否则将出现连向网格边界的路径)
(2)(p_a)和(p_b)恰好有一个是0:
这表明((i,j))的左方或者上方有一条路径连接。
由于这种走法只是将连进来的路径延长,所以这条路径在原轮廓线和新轮廓线上的性质是一样的,括号表示相同,其值也相同。
令(p_a'=p_a+p_b,;p_b'=0)代表3号((p_a
eq0))或5号((p_b
eq0))走法。(转移前提:(i<n))
令(p_a'=0,;p_b'=p_a+p_b)代表2号((p_a
eq0))或4号((p_b
eq 0))走法。(转移前提:(j<m))
(3)(p_a)和(p_b)皆不为0:
这表明((i,j))的左方和上方都有路径连接,所以((i,j))能填的只有6号走法,令(p_a'=0,;p_b'=0)。可是我们将两条路径连接起来后,其他位置的状态也需要改变,因为括号序列发生了变动。
(p_a=1,;p_b=1):两条左端路径此刻相连,对于(b)对应的右端路径位置(c),应该令(p_c'=1),此时(c)和(a)对应的右端路径(d)是一对路径。
(p_a=2,;p_b=2):两条右端路径此刻相连,同上,对于(a)的左端路径位置(c),令(p_c'=2)。
以上寻找对应括号路径,用括号序列配对的方式实现,复杂度(mathcal O(m))
(p_a=1;,p_b=2):一条回路此时形成。注意!这个转移只能在最后一个非障碍的位置发生,在其他任意位置连成回路都是不合法的方案。而且,答案就是被这种方式转移的量之和。这种情况我们不需要转移了,你可以姑且理解为这已经是最后一步大转移;但是对于进阶问题,如回路不一定要经过所有点,则是因为这种转移不应该被后续过程所利用(因为路径已经形成),并且答案统计可以在任意位置进行。
(p_a=2,;p_b=1):相当于从中间拼接两条路径,这种情况最舒服,因为其他位置都不会有任何变化。
转移种类较多,但是写起来其实是很好实现的。
实现
你可以将(f(i,j,state))设成一个数组,但是这样非常不便,且大转移时要枚举所有状态,非常的缓慢,而在实际中,许多状态是不合法的。
我们用两个哈希表(s_0,s_1)来分别模拟(f(i,j))和(f(i,j+1)):所有有效的(f(i,j,state))存在(s_0)中,大转移开始前,我们可以从(s_0)里提取所有有效状态,逐一转移至(s_1)中,也就是有效的(f(i,j+1,state))。下一步大转移开始前,交换两个哈希表,并清空(s_1)即可。
对于一个轮廓线状态(x),为了操作方便,需要实现两个函数:提取某一位的值、改变某一位的值。上述括号配对表达方式中状态仅有012三种,但我们可以用四进制来表示,因为位运算的效率相对来说会比较高。
总结
插头DP看起来十分难写,但只要自己动手实现一遍,就能理清楚插头DP的基本架构,就会发现它的实现方式其实挺简明的。建议读者还是自我摸索比较好。不过这里还是贴上模板题BZOJ1814的代码:
#include <cstdio>
#define Push(x,y) s[hv].insert((x),(y));
using namespace std;
typedef long long ll;
const int N=13,BAS[13]={1,4,16,64,256,1024,4096,16384,65536,262144,1048576,4194304,16777216};
const int HASH_MOD=4001,S=50000;
int n,m,mp[N][N],ln,lm;
int qcnt,q1[S],hu,hv;
ll q2[S];
ll ans;
char str[N];
inline void swap(int &x,int &y){x^=y^=x^=y;}
inline int get(int st,int x){
st>>=(x-1)<<1;
return st-((st>>2)<<2);
}
inline void mdf(int &st,int x,int y){
st+=(y-get(st,x))*BAS[x-1];
}
int match(int st,int x,int d){
int top=1,stdcol=get(st,x);
for(x+=d;top;x+=d){
int v=get(st,x);
if(v==0) continue;
(v==stdcol)?top++:top--;
}
return x-d;
}
struct Hash{/*{{{*/
int head[S],tot,id[S],nex[S];
ll val[S];
void reset(){
tot=0;
for(int i=0;i<HASH_MOD;i++) head[i]=0;
}
inline int get_hash(int x){return x%HASH_MOD;}
void insert(int x,ll value){
int hs=get_hash(x),u;
for(u=head[hs];u&&id[u]!=x;u=nex[u]);
if(!u){
id[++tot]=x; val[tot]=value;
nex[tot]=head[hs]; head[hs]=tot;
}
else val[u]+=value;
}
void layout(int &n,int *lis1,ll *lis2){
n=0;
for(int i=0;i<HASH_MOD;i++)
for(int u=head[i];u;u=nex[u])
n++,lis1[n]=id[u],lis2[n]=val[u];
}
}s[2];/*}}}*/
void draw(int i,int j){
int a=j,b=j+1,pa,pb;
s[hu].layout(qcnt,q1,q2);
int x; ll y;
while(qcnt){
x=q1[qcnt]; y=q2[qcnt--];
if(!y) continue;
if(j==1) x=(x-get(x,m+1)*BAS[m])<<2;
pa=get(x,a); pb=get(x,b);
if(mp[i][j]==1){
if(!pa&&!pb)
Push(x,y);
continue;
}
if(!pa&&!pb){
if(i<n&&j<m){
mdf(x,a,1); mdf(x,b,2);
Push(x,y);
}
}
else if(pa&&pb){
if(pa==1&&pb==1){
int pos=match(x,b,1);
mdf(x,pos,1);
mdf(x,a,0); mdf(x,b,0);
Push(x,y);
}
else if(pa==2&&pb==2){
int pos=match(x,a,-1);
mdf(x,pos,2);
mdf(x,a,0); mdf(x,b,0);
Push(x,y);
}
else if(pa==1&&pb==2){
if(i==ln&&j==lm){
bool flag=true;
for(int k=1;k<=m+1&&flag;k++) if(k!=a&&k!=b&&get(x,k)) flag=false;
if(flag)
ans+=y;
}
}
else{//pa==2&&pb==1
mdf(x,a,0); mdf(x,b,0);
Push(x,y);
}
}
else{
int u=pa,v=pb;
if(i<n){
mdf(x,a,u+v); mdf(x,b,0);
Push(x,y);
}
if(j<m){
mdf(x,a,0); mdf(x,b,u+v);
Push(x,y);
}
}
}
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%s",str+1);
for(int j=1;j<=m;j++){
if(str[j]=='.') mp[i][j]=0;
else mp[i][j]=1;
if(mp[i][j]!=1) ln=i,lm=j;
}
}
hu=0; hv=1;
s[hu].insert(0,1);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++){
draw(i,j);
swap(hu,hv);
s[hv].reset();
}
printf("%lld
",ans);
return 0;
}