diverta2019-2 E - Balanced Piles

diverta2019-2 E - Balanced Piles

对于一个序列$a_1,a_2,...,a_n$，同时给它们分配一个$1,2,...,n$的排列$b_1,b_2,..,b_n$，满足：
如果$a_i<a_j(i\ne j)$，则$b_i<b_j$需满足。
$b$序列的性质：$a_i$相同的数的$b$序列一定是连续的一段。如果$b_x=1$，则$a_x$一定是序列中的最小值。
序列$b$用来表示序列$a$中数的操作顺序。
如$a=[1,3,2,2,3,1],b=[2,6,3,4,5,1]$先操作第$6$个位置，再第$1$个位置、第$3$个位置、第$4$个位置、第$5$个位置、第$2$个位置。

当$a_1=a_2=...=a_n=0$时，有$n!$种合法的排列$b$。

随后修改一下问题：
初始有一个序列$a_1=a_2=...=a_n=0$，首先需要给这个序列分配一个$1,2,...,n$的排列$b_1,b_2,...,b_n$，然后执行下述操作若干次：
设序列中的最大值为$M$，选择一个整数$x\in [M+1,M+D]$。

1. 如果$x\ne H$，则再选择若干个数$1\le i_1<i_2<...<i_t\le n$满足$b_{i_1},b_{i_2},...,b_{i_t}$排序后的结果是$1,2,...,t$（确定了$t$即可确定$1\le i_1<i_2<...<i_t\le n$），随后令$a_{i_1}=a_{i_2}=...=a_{i_t}=x$，对于 j ∉ { i 1 , i 2 , . . . , i t } j\notin \{i_1,i_2,...,i_t\} j∈/​{i1​,i2​,...,it​}，令$b_j=b_j-t$，再给$b_{i_1},b_{i_2},...,b_{i_t}$分配一个$n-t+1,...,n-1,n$的排列（有$t!$种可能的排列）。
2. 如果$x=H$，则令$a_1=a_2=...=a_n=H$且$b_1=1,b_2=2,...,b_n=n$。

问有多少中不同的操作方案可以得到$a=[H,H,...,H],b=[1,2,...,n]$。

设$d{p}_i$表示序列$a$中最大值为$i$时序列对$(a,b)$的数量，有
$$\begin{gathered} d{p}_0=n! \\ d{p}_i=\sum _{j=max(0,i-D)}^{i-1}d{p}_j(1!+2!+...+n!)(0<i<H) \\ d{p}_H=\sum _{j=max(0,H-D)}^{H-1}d{p}_j \end{gathered}$$

keyidea：在序列内决定好相同数的操作顺序，每次将最小值修改为$[M,M+D]$中的某个数改成每次修改多个最小的数到$[M+1,M+D]$中的某个数。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
// const int mod=998244353;
const int mod=1e9+7;
#define inf 0x3f3f3f3f
// #define inf 0x3f3f3f3f3f3f3f3fll
// #define DEBUG
inline void add(int &x,int y){x+=y;if(x>=mod)x-=mod;}
void ast(ll x,ll l,ll r){assert(x>=l&&x<=r);}
int main()
{
    #ifdef DEBUG
    freopen("1.in","r",stdin);
    freopen("1.out","w",stdout);
    #endif
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int n,h,d;
    cin>>n>>h>>d;
    vector<int>f(n+1);
    f[0]=1;
    for(int i=1;i<=n;++i) f[i]=1ll*f[i-1]*i%mod;
    vector<int>dp(h+1);
    int sum=0;
    dp[0]=f[n];
    for(int i=1;i<=n;++i) sum=(sum+f[i])%mod;
    int s=dp[0];
    for(int i=1;i<=h;++i)
    {
        if(i<h) dp[i]=1ll*s*sum%mod;
        else dp[i]=s;
        s=(s+dp[i])%mod;
        if(i-d>=0) s=(s+mod-dp[i-d])%mod;
    }
    cout<<dp[h]<<'\n';
}