Loj #6069. 「2017 山东一轮集训 Day4」塔

题目描述

现在有一条 $ [1, l] $ 的数轴，要在上面造 $ n $ 座塔，每座塔的坐标要两两不同，且为整点。
塔有编号，且每座塔都有高度，对于编号为 $ i $ 座塔，其高度为 $ i $。对于一座塔，需要满足它与前面以及后面的塔的距离大于等于自身高度（不存在则没有限制）。问有多少建造方案。答案对 $ m $ 取模。

塔不要求按编号为顺序建造。

输入格式

一行三个整数 $ n, l, m $。

输出格式

输出一个整数，代表答案对 $ m $ 取模的值。

样例

样例输入

3 9 17

样例输出

15

数据范围与提示

对于 $ 10% $ 的数据，$ n \leq 10; l \leq 25 $；
对于 $ 30% $ 的数据，$ n \leq 20 $；
对于 $ 50% $ 的数据，$ n \leq 50 $；
对于 $ 70% $ 的数据，$ l \leq 105 $；
对于 $ 100% $ 的数据，$ n \leq 100; 1 \leq l \leq 10 ^ 9; 1 \leq m \leq 10 ^ 9 $。

首先我们得到一个排列\(P\),设\(S=\sum max\{P_{i-1},P_i\}\)，\(S+1\)就是这个排列紧密地排在一起时的长度。还剩下了\(l-(S+1)\)个格子， 我们就将这些格子放在相邻的元素之间，方案数为\(\binom{l-S-1+n}{n}\)。

所以我们要先求出\(S=k\)的排列个数。

设\(f_{i,j,k}\)表示放了前\(i\)个数，整个排列分为了\(j\)个联通块，排列的总长度为\(k\)的方案数。因为我们设从小到大放数，所以如果一个数\(i\)，它的左侧是空的或者会有另一个数，则不会对\(S\)有贡献，右侧同理。

转移的时候就考虑第\(i\)个数与多少个联通块相连（最多两个），就可以知道第\(i\)个数对\(S\)的贡献。

接下来考虑处理组合数。因为\(n\)比较小，所以对于每个\(L\),我们只保留\(\binom{L}{0..n}\)。用矩阵快速幂处理即可。

代码：

#include<bits/stdc++.h>
#define ll long long
#define N 105using namespace std;
inline int Get() {int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9') {if(ch=='-') f=-1;ch=getchar();}while('0'<=ch&&ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}return x*f;}int n,l;
ll mod;
int f[N][N*N],g[N][N*N];
int per[N*N];struct matrix {int a[105][105];void Init() {memset(a,0,sizeof(a));}
}F,G;int Mod(int a) {return a<mod?a:a-mod;}
matrix operator *(const matrix &x,const matrix &y) {static matrix tem;tem.Init();for(int i=0;i<=n;i++)for(int j=0;j<=n;j++)for(int k=0;k<=n;k++)tem.a[i][j]=Mod( tem.a[i][j]+1ll*x.a[i][k]*y.a[k][j]%mod);return tem;
}matrix ksm(matrix g,int x) {matrix ans;ans.Init();for(int i=0;i<=n;i++) ans.a[i][i]=1;for(;x;x>>=1,g=g*g)if(x&1) ans=ans*g;return ans;
}int main() {n=Get(),l=Get(),mod=Get();int sum=0;f[1][0]=1;sum=2;for(int i=2;i<=n;i++) {for(int j=1;j<i;j++) memset(g[j],0,sizeof(g[j]));for(int j=1;j<=i;j++) {for(int k=0;k<=sum;k++) {if(!f[j][k]) continue ;g[j+1][k]=Mod(g[j+1][k]+1ll*f[j][k]*(j+1)%mod);g[j][k+i]=Mod(g[j][k+i]+1ll*f[j][k]*j*2%mod);if(j>1) g[j-1][k+2*i]=Mod(g[j-1][k+2*i]+1ll*f[j][k]*(j-1)%mod);}}memcpy(f,g,sizeof(f));sum+=2*i;}for(int i=0;i<sum;i++) per[i+1]=f[1][i];int mx=0;for(int i=0;i<sum;i++) if(per[i]) mx=i;mx=min(mx,l+n);F.a[0][0]=1;for(int i=0;i<=n;i++) {G.a[i][i]=1;if(i) G.a[i-1][i]=1;}F=F*ksm(G,l-mx+n);ll ans=0;for(int i=mx;i>=0;i--) {(ans+=1ll*per[i]*F.a[0][n])%=mod;for(int j=n;j>0;j--) {(F.a[0][j]+=F.a[0][j-1])%=mod;}}cout<<ans;return 0;
}