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