【排列】(笛卡尔树上 dp?)

比赛在这

B.排列

前言:

笛卡尔树上 dp?这名字很妙啊,但其实不需要笛卡尔树,只不过利用了笛卡尔树的定义

一个性质:我们设一个区间 \([l,r]\) 中的最大值的位置为 \(pos\),发现可以把该区间分为 \([l,pos]\)\([pos,r]\) 两个子区间,并且这两部分互不影响,就是别管我左边怎么放数,不会对我的右边造成影响

注:如果你了解过笛卡尔树的话,那么下文所说区间都是指笛卡尔树上的区间;
如果你不知道笛卡尔树的话,我需要说:下文所说区间指一个数去向两边拓展,直到边上的数大于这个数

如下图整个区间分成了两个子区间

定义:

\(f_{i,j,0/1}\) 表示长度为 \(i\) 的区间用 \(j\) 次全部删完的方案数

1 表示该区间两边都挨着一个比该区间更大的数;0 表示一边挨着一个比区间内最大数更大的数,另一边是 \([1,n]\) 整个区间的边界

根据定义我们发现,显然我们没把 \([1,n]\) 整个区间囊括在 \(f\) 里,并且整个区间也不可能会有所有数全部删完的情况

所以最后整个区间的答案是要由它的两个子区间的的 dp 值单独计算的

考虑 dp 转移:

  • 对于 \(f_{i,j,1}\) 的转移:我们枚举最大值在该区间第几个数的位置 \(k\),对于每个 \(k\) 发现有 \(\tbinom{i-1}{k-1}\) 种方案(\(k-1\) 为最大值左侧区间数的个数)

    于是有转移:

    • 区间内最大值最后一个被删,那么左区间和右区间都是 \(j-1\) 次删完:

    \[f_{i,j,1} \leftarrow \sum_{k=1}^i \tbinom{i-1}{k-1}\times (f_{k-1,j-1,1}\times f_{n-k,j-1,1}) \]

    • 右区间中的最大值最后一个被删,则大区间的最大值是被左区间一边的更大值在第 \(j\) 次之前删掉:

    \[f_{i,j,1} \leftarrow \sum_{k=1}^i \tbinom{i-1}{k-1}\times (f_{n-k,j,1}\times \sum_{l=1}^{j-1} f_{k-1,l,1}) \]

    • 左区间中的最大值最后一个被删,同理:

    \[f_{i,j,1} \leftarrow \sum_{k=1}^i \tbinom{i-1}{k-1}\times (f_{k-1,j,1}\times \sum_{l=1}^{j-1} f_{n-k,l,1}) \]

    后两种情况式子里的枚举求和可以直接前缀和优化\(s\) 数组表示前缀合

    汇总一下转移就是:

    \[f_{i,j,1} \leftarrow \sum_{k=1}^i \tbinom{i-1}{k-1}\times (f_{k-1,j-1,1}\times f_{n-k,j-1,1}+f_{n-k,j,1}\times s_{k-1,l,1}+f_{k-1,j,1}\times s_{n-k,l,1}) \]

  • 对于 \(f_{i,j,0}\) 的转移:同样枚举最大值的位置,并且我们为避免重复,显然只应该计算一边为边界的情况,于是钦定左边为边界(

    因为左区间的左边没有更大值,所以区间最大值只能被右区间的右边的更大值删掉,于是右区间里的数一定是在 \(j\) 次以内被删完的

    有转移:

    • 左区间在大区间最大值删掉之前被删:

    \[f_{i,j,0} \leftarrow \sum_{k=1}^i \tbinom{i-1}{k-1}\times (\sum_{l=1}^{j-1} f_{k-1,l,0} \times f_{n-k,j-1,1}) \]

    • 左区间的最大值最后被删,于是中间最大值在 \(j\) 次以内被删,右区间所有数在 \(j-1\) 次以内被删完:

    \[f_{i,j,0} \leftarrow \sum_{k=1}^i \tbinom{i-1}{k-1}\times (f_{k-1,j,0}\times \sum_{l=1}^{j-2} f_{n-k,l,1}) \]

    同样前缀和优化( ,汇总一下:

    \[f_{i,j,0} \leftarrow \sum_{k=1}^i \tbinom{i-1}{k-1}\times (s_{k-1,l,0}\times f_{n-k,j-1,1}+f_{k-1,j,0}\times s_{n-k,l,1}) \]

时间复杂度

发现每次操作至少删掉区间中一半的数量,所以一个长为 \(i\) 的区间最多 \(\log i\) 次被删完

dp 过程中枚举区间长度、删除次数、最大值位置,复杂度为 \(O(n^2 \log)\)

答案计算:

有两种情况:

  1. 两个子区间都是在 \(K\) 次被删完了

  2. 一个子区间在 \(K\) 次被删完,而另一个在 \(K\) 次以内被删

code:

#include<bits/stdc++.h>
#define Aqrfre(x, y) freopen(#x ".in", "r", stdin),freopen(#y ".out", "w", stdout)
#define mp make_pair
#define Type ll
#define qr(x) x=read()
typedef __int128 INT;
typedef long long ll;
using namespace std;

inline ll read(){
    char c=getchar(); ll x=0, f=1;
    while(!isdigit(c)) (c=='-'?f=-1:f=1), c=getchar();
    while(isdigit(c)) x=(x<<1)+(x<<3)+(c^48), c=getchar();
    return x*f;
}

const int N = 1005; 

int T, n, K, p, C[N][N];
int f[N][22][2], s[N][22][2];

inline void AqrPre(){
    C[0][0] = 1;
    for(int i=1; i<=n; i++){
        C[i][0] = 1;
        for(int j=1; j<=i; j++){
            C[i][j] = (ll)(C[i-1][j] + C[i-1][j-1]) % p;
        }
    }
}

inline int qpow(int a, int b){
    int res = 1;
    while(b){
        if(b & 1) res = 1ll * res * a % p;
        a = 1ll * a * a % p;
        b >>= 1;
    }
    return res;
}

signed main(){ // a
    // Aqrfre(a, a);

    qr(n); qr(K); qr(p);
    
    if(K > 10){
        puts("0"); return 0;
    }
    if(K == 1){
        cout<<qpow(2, n - 1); return 0;
    }
    AqrPre();

    f[0][0][0] = f[0][0][1] = 1;
    for(int i=0; i<=K; i++) s[0][i][0] = s[0][i][1] = 1;

    for(int i=1; i<n; i++){
        for(int j=1; j<=K; j++){
            for(int k=1; k<=i; k++){
                f[i][j][1] = (f[i][j][1] + 1ll * C[i-1][k-1] * \
                        ( 1ll * f[k-1][j-1][1] * f[i-k][j-1][1] % p
                        + 1ll * f[k-1][j][1] * s[i-k][j-1][1] % p
                        + 1ll * s[k-1][j-1][1] * f[i-k][j][1] % p) % p) % p;

                f[i][j][0] = (f[i][j][0] + 1ll * C[i-1][k-1] * \
                        ( 1ll * s[k-1][j][0] * f[i-k][j-1][1] % p
                        + (j >= 2 ? 1ll * f[k-1][j][0] * s[i-k][j-2][1] : 0) % p) % p) % p;
            }

            s[i][j][1] = (s[i][j-1][1] + f[i][j][1]) % p;
            s[i][j][0] = (s[i][j-1][0] + f[i][j][0]) % p;
        }
    }
    
    int ans = 0;
    for(int i=1; i<=n; i++){
        ans = (ll)(ans + 1ll * C[n-1][i-1] *
        (1ll * f[i-1][K][0] * f[n-i][K][0] % p
       + 1ll * f[i-1][K][0] * s[n-i][K-1][0] % p
	   + 1ll * s[i-1][K-1][0] * f[n-i][K][0] % p) % p) % p;
    }
    cout<<ans<<"\n";
    

    return 0;
}
posted @ 2024-11-05 15:55  Aqr_Rn  阅读(63)  评论(9编辑  收藏  举报