题目描述
如果一个包含NNN个元素的数组aaa里面的元素的值是在1...K1...K1...K之间的整数,存在多少个不同的数组aaa,进行了如上扫描之后,maxmaxmax恰好进行了PPP次更新
把答案mod(109+7)mod (10^9+7)mod(109+7)输出
题目解析
DPDPDP
数组f[i][j][k]f[i][j][k]f[i][j][k]表示前iii位,最大值为jjj,并且已经更新了kkk次的方案数
那么f[i][j][k]=f[i−1][q][k−1]+f[i−1][j][k]∗jf[i][j][k]=f[i-1][q][k-1]+f[i-1][j][k]*jf[i][j][k]=f[i−1][q][k−1]+f[i−1][j][k]∗j,其中qqq为所有小于jjj数字
因为当我们在最后一位加入jjj时,所有第iii位之前最大的数比jjj小的方案都会更新一次,所以要加上所有的f[i−1][q][k−1]f[i-1][q][k-1]f[i−1][q][k−1],qqq要小于kkk
如果第iii位之前最大的数已经为jjj,则本次不会更新,但第iii位可以填入任何比jjj小的数,所以加上f[i−1][j][k]∗jf[i-1][j][k]*jf[i−1][j][k]∗j
最后输出f[n][k][p]f[n][k][p]f[n][k][p]即可
可用前缀和优化一下第二步
代码
#include<bits/stdc++.h>
using namespace std;
const int M=1e9+7;
int T,n=100,k=300,p=99,ans;
long long f[105][305][105],sum[105][305][105];
int main()
{
for(int i=1;i<=k;i++)
{
f[1][i][0]=1;
sum[1][i][0]=i%M;
}
for(int i=2;i<=n;i++)
for(int j=1;j<=k;j++)
{
f[i][j][0]=((f[i][j][0]%M)+(f[i-1][j][0]%M)*(j%M))%M;
sum[i][j][0]=((sum[i][j][0]%M)+(f[i][j][0]%M)+(sum[i][j-1][0]%M))%M;
for (int s=1;s<=p;s++)
{
f[i][j][s]=((f[i][j][s]%M)+(sum[i-1][j-1][s-1]%M)+(f[i-1][j][s]%M)*(j%M))%M;
sum[i][j][s]=((sum[i][j][s]%M)+(f[i][j][s]%M)+(sum[i][j-1][s]%M))%M;
}
}
cin>>T;
while(T--)
{
cin>>n>>k>>p;
cout<<sum[n][k][p]<<endl;
}
}