2022“杭电杯”中国大学生算法设计超级联赛（9）

1007 Matryoshka Doll
思路：

𝑑𝑝(𝑥,𝑦)表示将前𝑥个套娃分成𝑦组的方案数。
转移时考虑第x个套娃放在前面的某一组（x-1,y），或者单独一组(x-1,y-1)。 即𝑑𝑝(𝑥,𝑦) = 𝑑𝑝(𝑥 − 1,𝑦 − 1) + 𝑑𝑝(𝑥 − 1,𝑦) · max{0,𝑦 − 𝑓 (𝑥)}， 𝑓(𝑥)表示满足1 ≤ 𝑧 <𝑥且𝑎𝑥 − 𝑟 < 𝑎𝑧 ≤ 𝑎𝑥的𝑧的个数, 即y组中不能放入x进行转移的组的数量。
只要知道有几个组可以转移就行了，不需要知道前面哪些组可以转移，也不需要维护前面每一组的信息，只要知道能转移的组的个数就行，不能转移的个数可以维护前面最后一个a[p]+r<=a[i]的数的位置p得到。
分成y组不需要知道每组的最大值，因为反正dp是考虑了所有情况的，肯定会包含那些不能进行转移的情况，只需要y把这些情况减掉就行了

//1007
#include<bits/stdc++.h>
typedef long long LL;
using namespace std;
const int maxn = 5050, mod = 998244353;
int a[maxn], dp[maxn][maxn];                    //前i个数分成j组的方案数
int main(){
    int T;  cin>>T;
    while(T--){
        memset(dp,0,sizeof(dp));
        int n, k, r;  cin>>n>>k>>r;
        for(int i = 1; i <= n; i++)cin>>a[i];
        int p = 0;  dp[0][0] = 1;
        for(int i = 1; i <= n; i++){
            while(p+1<i && a[p+1]+r<=a[i])p++;  //p维护前面最后一个a[p]+r<=a[i]的数的位置
            int s = i-p-1;                      //前面的不能转移的数的个数
            for(int j = s+1; j <= i; j++){      //单独成组，或放在前面的某一组
                dp[i][j] = (dp[i-1][j-1]+1ll*dp[i-1][j]*(j-s))%mod;
            }
        }
        cout<<dp[n][k]<<"\n";
    }
    return 0;
}

。