题目
有nnn张卡片,第iii张卡片上写着aia_iai,可以选择扔掉第iii张卡片并向前走aia_iai步,当走到厄运数字时,game over,问有多少种方法可以扔掉所有卡片
分析
这是一道状压dp的题目,dp[i]dp[i]dp[i]表示扔掉集合iii所有卡片的方案,初始dp[0]=1dp[0]=1dp[0]=1,dis[i]dis[i]dis[i]表示集合iii扔掉所有卡片所在的位置,那么dis[i]=dis[i异或lowbit(i)]+dis[lowbit(i)]dis[i]=dis[i异或lowbit(i)]+dis[lowbit(i)]dis[i]=dis[i异或lowbit(i)]+dis[lowbit(i)],判断该位置是否能走,再进行dp,那么状态转移方程是什么dp[i]=dp[i]+dp[去掉二进制下i的某一位]dp[i]=dp[i]+dp[去掉二进制下i的某一位]dp[i]=dp[i]+dp[去掉二进制下i的某一位](记得开O2)
代码
#include <cstdio>
#include <cctype>
#define rr register
using namespace std;
const int mod=1000000007,N=1<<24;
int n,dp[N],dis[N],lim0=-1,lim1=-1;
inline signed iut(){
rr int ans=0; rr char c=getchar();
while (!isdigit(c)) c=getchar();
while (isdigit(c)) ans=(ans<<3)+(ans<<1)+(c^48),c=getchar();
return ans;
}
inline signed modd(int x){return x>=mod?x-=mod:x;}
inline void answ(int x){
for (rr int t=x,j;t;t^=j) j=-t&t,dp[x]=modd(dp[x]+dp[x^j]);
}
signed main(){
n=iut(); dp[0]=1; rr int all=1<<n;
for (rr int i=0;i<n;++i) dis[1<<i]=iut();
rr int m=iut(); if (m>0) lim0=iut(); if (m>1) lim1=iut();
for (rr int i=1;i<all;++i){
rr int j=-i&i;
dis[i]=dis[i^j]+dis[j];
if (dis[i]==lim0||dis[i]==lim1) continue;
answ(i);
}
return !printf("%d",dp[all-1]);
}