JZOJ 4806. 打工

Problem

Description

Input

第一行，一个整数N表示参赛人数。
第二行，N个整数，表示询问的分队方式的序列。

Output

一行，一个整数表示这种方式会在第几天被采用。答案对1,000,007取模。

Sample Input

3
1 2 2

Sample Output

4

Data Constraint

对于100%的数据，N ≤ 10000 ， 数据保证询问的数列是一个有效的序列。

Hint

比赛各天的分队情况如下：
第一天：1,1,1
第二天：1,1,2
第三天：1,2,1
第四天：1,2,2
第五天：1,2,3

Solution

果断DP。设$F_{i,j}$表示做到第i个人，最大的队伍序号为j的个数，且枚举的序列在输入序列之前。那么我们分两种情况讨论：
①之前有j队人，它可以选1-j队任一一队。
②它新开一队。
那么转移方程为

$$F_{i,j}=F_{i-1,j}*j+F_{i-1,j-1}$$
不要忘记加上前面与输入序列严格相同，尾数为1..a[i]-1的那些序列！

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 10010
#define mo 1000007
#define LL long long
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
LL f[2][N],a[N];
LL i,j,p,n,mx,ans;
int main()
{
    scanf("%lld",&n);
    fo(i,1,n) scanf("%lld",&a[i]);
    p=1;mx=a[1];
    fo(i,1,n)
    {
        memset(f[p],0,sizeof(f[p]));
        fo(j,1,i)
            f[p][j]=(f[1-p][j]*j+f[1-p][j-1])%mo;
        f[p][mx]=(f[p][mx]+a[i]-1)%mo; 
        mx=max(mx,a[i]);
        p=1-p;
    }
    fo(i,1,n) ans=(ans+f[1-p][i])%mo;
    ans=(ans+1)%mo;
    printf("%lld",ans);
}