CodeChef Little Elephant and Movies [DP 排列]

https://www.codechef.com/FEB14/problems/LEMOVIE

题意：

对于一个序列，定义其“激动值”为序列中严格大于前面所有数的元素的个数。
给定n个数p1;,p2... pn，求这n个数的所有排列中，激动值不超过k的个数。$1 k \le n \le 200，1 \le pi \le 200$

---

这种题有一个很神的想法：

把排列按某种顺序往里插入，使得后不会影响前

对于本题，先离散化去重后，从大到小插入，后插入的元素不会影响已经插入的元素严格大于前面所有数

$f[i][j]$表示插入了前$i$大的数，激动值为$j$的方案数

激动值改变只有可能是当前要插入的数中有一个放在了最前面

转移时分类讨论有没有数插在最前面，需要用到隔板法，

设已经插入$x$个数，第$i$大的有$y$个数

$f(i,j)=f(i-1,j)*\binom{x+y-1}{x-1}*{y!}\ +\ f(i-1,j-1)*\binom{x+y-1}{x}*{y!}$

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N=,INF=1e9+,P=1e9+;
inline int read(){
    char c=getchar();int x=,f=;
    while(c<''||c>''){if(c=='-')f=-;c=getchar();}
    while(c>=''&&c<=''){x=x*+c-'';c=getchar();}
    return x*f;
}
int n,k,a[N];
int m,d[N],s[N];
inline bool cmp(int a,int b){return a>b;}
ll c[N][N],fac[N];
inline void mod(ll &x){if(x>=P) x-=P;}
void ini(){
    c[][]=;fac[]=;
    for(int i=;i<=;i++){
        c[i][]=;
        for(int j=;j<=i;j++) mod(c[i][j]+=c[i-][j]+c[i-][j-]);
        fac[i]=fac[i-]*i%P;
    }
}
ll f[N][N];
void dp(){
    f[][]=;
    for(int i=;i<=m;i++)
        for(int j=;j<=k;j++){
            int x=s[i-],y=d[i];
            f[i][j]=f[i-][j]*c[x+y-][x-]%P*fac[y]%P;
            mod(f[i][j]+=f[i-][j-]*c[x+y-][x]%P*fac[y]%P);
            //printf("f %d %d %lld\n",i,j,f[i][j]);
        }
    ll ans=;
    for(int i=;i<=k;i++) mod(ans+=f[m][i]);
    printf("%lld\n",ans);
}
int main(){
    freopen("in","r",stdin);
    ini();
    int T=read();
    while(T--){
        memset(d,,sizeof(d));
        n=read();k=read();
        for(int i=;i<=n;i++) a[i]=read();
        sort(a+,a++n,cmp);
        m=;
        d[++m]=;
        for(int i=;i<=n;i++){
            if(a[i]==a[i-]) d[m]++;
            else d[++m]=;
        }
        for(int i=;i<=m;i++) s[i]=s[i-]+d[i];
        //for(int i=1;i<=m;i++) printf("%d ",d[i]);puts("");
        //for(int i=1;i<=m;i++) printf("%d ",s[i]);puts("");
        dp();
    }
}