今天我们在夏令营主要学习了状压DP的知识。
状压DP是指把DP状态从多维压缩到少维,通常使用二进制的01表示是否操作。
应用基本上就是预处理排去不合适的和用dfs搜索各种状态,其中注意dfs中注意被之前的阶段的影响。
一些例题:
第一题:种玉米poj3254
就是玉米不能种在某一些位置且不能相邻,问有多少种种法?
这一题就是去除所有横向相邻的所有01状态,离散化之后枚举行数和上一行状态(只有上一行影响这一行),判断一下上下相邻和不能种的位置就可以传下去了。
#include <iostream>
#include <fstream>
#include <algorithm>
using namespace std;
//ifstream fin("A.in");
//#define cin fin
int n,m;
int c[15][15];
int vnum,v[(1<<12)+1];
int f[15][(1<<12)+1];
void getvnum()
{
for(int i=0;i<(1<<m);i++)
if(!(i&(i<<1)))
v[vnum++]=i;
}
bool pd(int row,int x)
{
for(int i=m-1;i>=0;i--)
{
if(!c[row][i] && x&1) return false;
x>>=1;
}
return true;
}
int main()
{
cin>>n>>m;
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
cin>>c[i][j];
getvnum();
for(int i=0;i<n;i++)
{
for(int j=0;j<vnum;j++)
if(pd(i,v[j]))
{
// cout<<v[j]<<" ";
if(i==0) f[i][j]++;
else for(int k=0;k<vnum;k++)
if(pd(i-1,v[k]) && !(v[j]&v[k]))
f[i][j]+=f[i-1][k];
}
// cout<<endl;
}
int sum=0;
for(int i=0;i<vnum;i++)
sum=(sum+f[n-1][i])%100000000;
/* for(int i=0;i<n;i++)
{
for(int j=0;j<vnum;j++)
cout<<f[i][j]<<" ";
cout<<endl;
}
*/ cout<<sum<<endl;
return 0;
}
第二题:poj2411
把一个n×m的用1×2刚好铺满(只能横铺或竖铺),问有多少种方法?
这题就是用dfs+状压DP。枚举行数与上一行的状态,然后dfs这一行的状态。dfs时,可以选择放和不放(01)0就是下一行时竖放,1就是横放。然后把状态传下去。
#include <iostream>
#include <fstream>
#include <algorithm>
#include <cstring>
using namespace std;
int n,m;
long long f[15][1<<11];
long long add;
int getf(int p)
{
for(int i=0;i<m;i++)
if(!(p&(1<<(m-i-1)))) return i;
// else cout<<(1<<(m-i-1))<<" ";
return m;
}
void dfs(int r,int p,int k)
{
// cout<<r<<" "<<p<<" "<<k<<" "<<f[r][p]<<endl;
// cout<<(1<<(m-k-1))<<" "<<(1<<(m-k-2))<<endl;
if(k>=m-1 || p==((1<<m)-1)/**/)
{
f[r][p]+=add;
// cout<<"#"<<p<<" "<<f[r][p]<<endl;
return ;
}
dfs(r,p,k+1);
if(k<=m-2 && !(p&(1<<(m-k-1))) && !(p&(1<<(m-k-2))))
dfs(r,p|(1<<(m-k-1))|(1<<(m-k-2)),k+2);
}
void work()
{
add=1;
dfs(0,0,0);
// cout<<getf(12)<<endl;
// return ;
/* for(int j=0;j<(1<<m);j++)
cout<<f[0][j]<<" ";
cout<<endl;
*/ for(int i=1;i<n;i++)
{
for(int j=0;j<(1<<m);j++)
{
if(f[i-1][j]) add=f[i-1][j];
else continue;
// cout<<"j:"<<j<<endl;
int k;
k=getf(~j&((1<<m)-1));
dfs(i,~j&((1<<m)-1),k);
/* for(int j=0;j<(1<<m);j++)
cout<<f[i][j]<<" ";
cout<<endl;
*/ }
// cout<<endl;
}
// cout<<f[1][3]<<" ";
int sum=0;
for(int i=0;i<(1<<m);i++)
sum+=f[n-1][i];
cout<<f[n-1][(1<<m)-1]<<endl;
}
int main()
{
while(1)
{
cin>>n>>m;
if((n*m)%2)
{
cout<<0<<endl;
continue;
}
if(!n || !m) break;
memset(f,0,sizeof(f));
work();
}
return 0;
}
第三题:炮兵阵地
一片区域内有山地与平地,只可以在平地上放炮兵。炮兵会向上上左左下下右右8个平地和山地放炮。给出一片一片区域问最多放多少支炮兵?
就是把第一题的预处理与第二题的dfs结合起来。先枚举行数,上上行、上行的状态,然后有dfs搜索,搜到后dp加上状态炮兵个数。
#include <iostream>
#include <fstream>
#include <algorithm>
#include <cstring>
using namespace std;
//ifstream fin("C.in");
//ofstream fout("C.out");
//#define cin fin
//#define cout fout
int n,m;
char c[101][11];
int vnum,v[1<<10];
int f[101][101][101];
int b[1<<10];
int h[101];
int Max;
void getv()
{
for(int i=0;i<(1<<m);i++)
if(!(i&(i<<1)) && !(i&(i<<2)))
{
b[i]=vnum,v[vnum]=i;
int tmp=i;
while(tmp>0)
{
if(tmp&1) h[vnum]++;
tmp>>=1;
}
vnum++;
}
}
bool pd(int r,int x)
{
for(int i=m;i>=1;i--,x>>=1)
if(c[r][i]=='H' && x&1) return false;
return true;
}
void dfs(int r,int x1,int x2,int x,int k)
{
// cout<<"#:"<<r<<" "<<x1<<" "<<x2<<" "<<x<<" "<<k<<endl;
// cout<<((x1|x2)>>(m-k))<<endl;
if(k>m)
{
f[r][b[x2]][b[x]]=max(f[r-1][b[x1]][b[x2]]+h[b[x]],\
f[r][b[x2]][b[x]]);
Max=max(Max,f[r][b[x2]][b[x]]);
// cout<<x1<<" "<<x2<<" "<<x<<" "<<f[r][b[x2]][b[x]]<<endl;
return ;
}
dfs(r,x1,x2,x,k+1);
if(pd(r,x|(1<<(m-k))) && !(((x1|x2)>>(m-k))&1) && \
!(((x&(-x))>>1)&(1<<(m-k))) && !(((x&(-x))>>2)&(1<<(m-k))))
dfs(r,x1,x2,x|(1<<(m-k)),k+1);
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
cin>>c[i][j];
getv();
// for(int i=0;i<vnum;i++)
// cout<<v[i]<<" "<<i<<": "<<h[i]<<endl;
memset(f,-1,sizeof(f));
f[0][0][0]=0;
dfs(1,0,0,0,1);
// return 0;
// dfs(2,0,9,0,1);
// return 0;
for(int i=2;i<=n;i++)
for(int s1=0;s1<vnum;s1++)
{
for(int s2=0;s2<vnum;s2++)
if(f[i-1][s1][s2]!=-1)
// cout<<i<<": "<<v[s1]<<" "<<v[s2]<<": \n",
dfs(i,v[s1],v[s2],0,1);
if(i==2) break;
}
cout<<Max<<endl;
return 0;
}