uoj196,DP好题

传送门
orz梁大
一开始的想法是设f[i][j][k]表示i次操作后第j个数为k的概率，
但不同的f[i][j][k]好像之间并不是独立的，反正没写出来。
然后就去抄梁大的题解，似乎很巧妙。

#include<cstdio> 
#include<cstring>
#include<algorithm>
using namespace std;
const int mo=1000000007,N=405;
int n,a[N],c[N],i,j,k,l,r,q,m,d[N],ff[2][N][N],*f[N],*g[N],*t,h[N][N],ans,s[N][N];
pair<int,int> b[N];
inline void swAp(){
    for(int i=0;i<N;++i){
        t=f[i];
        f[i]=g[i];
        g[i]=t;
    }
}
inline void solve(int l,int r,int p){
    int i,j,k,x;
    for(i=0;i<N;++i)f[i]=ff[0][i],g[i]=ff[1][i];
    for(i=0;i<N;++i)for(j=0;j<N;++j)ff[0][i][j]=ff[1][i][j]=0;
    g[l][r]=1;
    for(i=1;i<=q;++i){
        for(j=l;j<=r;++j)
            for(k=r,x=0;k>=j;--k){
                f[j][k]=x;
                x=(x+1ll*(n-k)*g[j][k])%mo;
            }
        x=0;
        for(j=r;j>=l;--j)
            for(k=l,x=0;k<=j;++k){
                f[k][j]=(f[k][j]+x)%mo;
                x=(x+1ll*(k-1)*g[k][j])%mo;
            }
        for(j=l;j<=r;++j)
            for(k=j;k<=r;++k)
                f[j][k]=(f[j][k]+1ll*g[j][k]*s[j][k])%mo;
        swAp();
    }
    for(i=l;i<=r;++i)
        for(j=i;j<=r;++j)
            for(k=i;k<=j;++k)
                h[k][c[p]]=(h[k][c[p]]+g[i][j])%mo;
}
inline int calc(int x){
    return x*(x+1)>>1;
}
int main(){
    //freopen("ex_segment3.in","r",stdin);
    //freopen("2.txt","w",stdout);
    scanf("%d%d",&n,&q);
    for(i=1;i<=n;++i)scanf("%d",a+i),b[i]=make_pair(a[i],i);
    for(i=1;i<=n;++i)
        for(j=i;j<=n;++j)
            s[i][j]=calc(i-1)+calc(n-j)+calc(j-i+1);
    sort(b+1,b+n+1);
    for(i=1;i<=n;++i){
        if(i==1 || b[i-1].first<b[i].first)d[++m]=b[i].first;
        c[b[i].second]=m;
    }
    for(i=1;i<=n;++i){
        for(l=i;l && c[l]<=c[i];--l);
        for(r=i;r<=n && c[r]<=c[i];++r);
        solve(++l,--r,i);
    }
    for(i=1;i<=n;++i){
        ans=l=0;
        for(j=1;j<=n;++j)if(h[i][j])ans=(ans+1ll*(h[i][j]-l+mo)%mo*d[j])%mo,l=h[i][j];
        printf("%d ",ans);
    }
    return 0;
}