经典题目,永远的状压第一题,状压思路从这里启蒙
状压,就是将每一种状态以一个二进制数来表示,与朴素方法的区别就是能通过位运算进行优化,把要处理的问题降维。比如说一个二维的图能通过将一行的所有状态储存起来,通过遍历每一行的状态,能将整个图遍历。这在判断状态与状态之间的影响时帮助巨大。
题目很短:在N×N的棋盘里面放K个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上左下右上右下八个方向上附近的各一个格子,共8个格子。
输入:N,K ( 1 <=N <=9, 0 <= K <= N * N)
输出:方案数
看到这一题,没有上一篇(炮兵阵地)那么毒瘤的地形设定,直接开始初始化
初始化: 找到一行里面能出现的所有状态,记录下来状态本身和其对应的放置数
long long dp[10][15000][82],tmp[100005],s[100005];
// dp [行][状态][已有数量] tmp[i]状态 s[i]是i对应的状态放置数
long long n,k,num;
//num总状态数
int ghs(int h){//统计当前状态的放置数
int ss=0;
while(h!=0){
if(h&1)ss++;
h=h>>1;//和h/=2一样,这个快点
}
return ss;
}
void start(){
for(int i=1;i<(1<<n);i++){
if(i&(i>>1)||i&(i<<1))continue;
tmp[++num]=i;
s[num]=ghs(i);
}
}
判断递推条件:从题意中我们能得到的限制条件是k,所以将k作为dp的维度,且因为只需要考虑上一行的状态,所以我们只需要记录本行的状态,可以得到递推式
dp[i][j][s[j]+sum]+=dp[i-1][p1][sum];//dp [行][状态][已有数量]
通过状态合法的条件和递推式进行遍历:
for(int i=1;i<=num;i++){
if(s[i]<=k)dp[1][i][s[i]]=1;
}
for(int i=2;i<=n;i++){//行数
for(int j=1;j<=num;j++){//枚举状态
for(int p1=1;p1<=num;p1++){//上一行状态
if(tmp[j]&tmp[p1]||(tmp[j]>>1)&tmp[p1]||(tmp[j]<<1)&tmp[p1])continue;//判断
for(int sum=1;sum<=k;sum++){//遍历数量
if(sum+s[j]<=k){
dp[i][j][s[j]+sum]+=dp[i-1][p1][sum];
}
}
}
}
}
最后给出ac代码:
#include<bits/stdc++.h>
using namespace std;
long long dp[10][15000][82],tmp[100005],s[100005];
long long n,k,num;
// dp [行][状态][已有数量] tmp[i]状态 s[i]是tmp中的状态放置数
int ghs(int h){
int ss=0;
while(h!=0){
if(h&1)ss++;
h=h>>1;
}
return ss;
}
void start(){
for(int i=1;i<(1<<n);i++){
if(i&(i>>1)||i&(i<<1))continue;
tmp[++num]=i;
s[num]=ghs(i);
}
}
int main(){
tmp[++num]=0;
cin>>n>>k;
start();
for(int i=1;i<=num;i++){
if(s[i]<=k)dp[1][i][s[i]]=1;
}
//judge tmp[i]&tmp[j]==0 tmp[i]&(tmp[j]>>1)==0 tmp[i]&(tmp[j]<<1)==0
for(int i=2;i<=n;i++){//行数
for(int j=1;j<=num;j++){//枚举状态
for(int p1=1;p1<=num;p1++){//上一行状态
if(tmp[j]&tmp[p1]||(tmp[j]>>1)&tmp[p1]||(tmp[j]<<1)&tmp[p1])continue;//judge
for(int sum=1;sum<=k;sum++){
if(sum+s[j]<=k){
dp[i][j][s[j]+sum]+=dp[i-1][p1][sum];
}
}
}
}
}
long long ans=0;
for(int i=1;i<=n;i++){
for(int j=1;j<=num;j++){
ans+=dp[i][j][k];
}
}
cout<<ans;
}
一些小细节:
1、进行状压前先估算状态数,把数组开够(wa了一发)
2、别忘了long long(wa了两发)
3、枚举当前状态(j)和上一行状态(p1)的顺序可以互换
4、sum那组遍历从前到后和从后到前没区别(和背包的概念一定要区分开,别混淆)
写在最后:
入门dp切勿胆怯,理清思路豁然开朗
然后孩子想放国庆555555