uoj196,DP好题

来源:互联网 发布:苏州爱知科技怎么样 编辑:程序博客网 时间:2024/05/21 02:51

传送门
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;}