最大值 （补档） dp+前缀和优化

题目描述

找到一个数组的最大值的一种方法是从数组开头从前到后对数组进行扫描,令max=a[0]（数组下表从0..N-1）,如果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

输出格式

输出文件findmax.out包括T行,每行一个答案。

分析

据——老伙计所说
f[i][j][k]表示前i项最大值为j且更新了k次的方案数，

$$f[i][j][k]=f[i][j][k]+f[i−1][j][k]∗j+f[i][v][k−1](v<j)$$

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<string>
#include<algorithm>

using namespace std;

long long f[201][401][121];
long long sum[201][401][121];
int n,m,p;
int main()
{
    freopen("findmax.in", "r", stdin);
    freopen("findmax.out", "w", stdout);    

    int g=1000000007;
    n=100;
    m=300;
    p=100;

    for (int i=1;i<=m;i++)
    {
        f[1][i][0]=1;sum[1][i][0]=i;
    }

    for (int i=2;i<=n;i++)
        for (int j=1;j<=m;j++)
        {
            f[i][j][0]=((f[i-1][j][0]%g)*(j%g))%g;
            sum[i][j][0]=((f[i][j][0]%g)+(sum[i][j-1][0]%g))%g;
            for (int k=1;k<=p;k++)
            {
                f[i][j][k]=((sum[i-1][j-1][k-1]%g)+(f[i-1][j][k]%g)*j)%g;
                sum[i][j][k]=((f[i][j][k]%g)+(sum[i][j-1][k]%g))%g;
            }
        }

    int t;
    scanf("%d",&t);
    for (int i=1;i<=t;i++)
    {
        scanf("%d%d%d",&n,&m,&p);
        printf("%lld\n",sum[n][m][p]);
    }
    return 0;
}