题目描述:
找到一个数组的最大值的一种方法是从数组开头从前到后对数组进行扫描,令max=a0,如果a[i]>max,就更新max,这样就可以在O(N)的时间里找到一个数组的最大值。
这个问题是相当简单的,但是想到了另一个问题,如果一个包含N个元素的数组a里面的元素的值是在1…K之间的整数,存在多少个不同的数组a,进行了如上扫描之后,max恰好进行了P次更新?
下面是N = 4,K = 3,P = 2时所有情况
1) {1,1,2,3}
2) {1,2,1,3}
3) {1,2,2,3}
4) {1,2,3,1}
5) {1,2,3,2}
6) {1,2,3,3}
共有6种情况
由于答案可能很大,所以你仅仅需要把答案mod (10^9+7)输出。
样例输入
输入文件findmax.in的第一行T,本题有T组数据。
接下来T行,每行三个整数N,K,P
3
4 3 2
2 3 1
3 4 1
样例输出
输出文件findmax.out包括T行,每行一个答案。
6
3
30
数据范围:
30%数据 T=1;1 <= n <= 10;1 <= K <= 2;0 <= P < n
60%数据 T=1;1 <= n <= 50;1 <= K <= 10;0 <= P < n
100%数据1 <= T <= 100;1 <= n <= 100;1 <= K <= 300;0 <= P < n
分析:
很明显,这道题要用dp。设f[i][j][k]表示当前序列长度为i,最大值为j,更新了k次时的方案数。
很容易就可以得到方程:
f[i][j][k]=f[i][j][k]+f[i-1][1~j-1][k-1]+(f[i-1][j][k]*j)
为什么要乘j呢?
因为有j个数小于等于j。
CODE:
#include<cstdio>
#include<cstring>
using namespace std;
const int mo=1000000007;
long long f[101][301][101];
long long zl[101][301][101];
long long sum;
void work()
{
int n,m,p,i,j,k;
scanf("%d%d%d",&n,&m,&p);
if (zl[n][m][p]!=0)
{
printf("%d\n",zl[n][m][p]);
return;
}
long long ans=0;
for (i=1;i<=m;i++)
{
ans=(ans+f[n][i][p]);
}
zl[n][m][p]=ans%mo;
printf("%d\n",ans%mo);
}
int main()
{
int t;
scanf("%d",&t);
int i,j,k;
for (i=1;i<=300;i++)
{
f[1][i][0]=1;
}
for (i=2;i<=100;i++)
{
for (k=0;k<=100;k++)
{
sum=0;
for (j=1;j<=300;j++)
{
f[i][j][k]=(f[i-1][j][k]*j+sum)%mo;
sum=(sum+f[i-1][j][k-1])%mo;
}
}
}
for (int i=1;i<=t;i++)
{
work();
}
return 0;
}