[Codeforces Global Round 14] F. Phoenix and Earthquake (组合数学+打表逆元+dp)

题目链接：F. Phoenix and Earthquake

题解

我们考虑手动开灯情况，很明显就是一段段连续的1中间最多隔一个0。
每个区间内灯被点亮的方案数为$2^{len-1}$个,(len为区间长度)。

定义：$设dp[i][j]:前i个灯有j个手动开,最后一个是手动打开的方案数。$
初始化：$dp[i][i]=2^{i-1}$

状态转移：$dp[i+k+1][j+k]=dp[i][j]\ast 2^{k-1}\ast C_{j+k}^k(k就是下一个连续手动开灯的区间长度)$

答案：${\sum }_{j=0}^{n}dp[n][j]$

由于要取模，所以求组合数时要求逆元，直接求会超时，所以逆元线性打表把1～n的逆元表打出来，这样可以O(n)递推求解阶乘逆元。

代码

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<bitset>
#include<cassert>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<deque>
#include<iomanip>
#include<list>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<vector>
#include<unordered_set>
#include<unordered_map>
using namespace std;
//extern "C"{void *__dso_handle=0;}
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define pii pair<int,int>
#define lowbit(x) x&-x

const double PI=acos(-1.0);
const double eps=1e-6;
const ll mod=1e9+7;
const int inf=0x3f3f3f3f;
const ll llinf=1e18;
const int maxn=1e5+10;
const int maxm=400+10;
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);

int n,m;
ll d[maxm][maxm];
ll f[maxm],f2[maxm],inv[maxm],invF[maxm];
void init()
{
	f[1]=1,f2[0]=1,f2[1]=2;
	inv[1]=invF[1]=1;
	for(int i=2;i<=n;i++)
	{
		f[i]=i*f[i-1]%m;
		inv[i]=((m-m/i)*(inv[m%i])+m)%m;
		invF[i]=invF[i-1]*inv[i]%m;
		f2[i]=2*f2[i-1]%m;
	}
}
inline ll c(ll a,ll b) {
	if(a<b || a<0 || b<0) return 0;
	return f[a]*invF[b]%m*invF[a-b]%m;
}
int main()
{
	scanf("%d%d",&n,&m);
	init();
//	前i个灯有j个手动开,最后一个是手动打开的方案数
//	dp[i+k+1][j+k]=dp[i][j]*qpow(2,k-1)*c(k,j+k);
	for(int i=1;i<=n;i++) d[i][i]=f2[i-1];
	for(int i=1;i<=n;i++)
		for(int j=0;j<=i;j++)
		{
			for(int k=1;i+k+1<=n;k++)
				d[i+k+1][j+k]=(d[i+k+1][j+k]+d[i][j]*f2[k-1]%m*c(j+k,k)%m)%m;
		}
	ll ans=0;
	for(int j=0;j<=n;j++) ans=(ans+d[n][j])%m;
	printf("%lld\n",ans);
}