【NOIP2016提高A组五校联考3】打工 题解

Description

Input

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

Output

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

Sample Input

3
1 2 2

Sample Output

4

Data Constraint

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

Solution

有一个很显然的性质：第i个组的第一个必须在第i-1个组的第一个后面，才能保证字典序最好
那么DP
设$F[i,j]$表示长度为i，最大的组别为j，并且比输入的数小的方案数
再设$G[i,j]$表示长度为i，最大的组别为j，并且等于输入的数的方案数，显然只可能为0或1
自己推方程

Code

#include<cstdio>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define N 20000
#define mo 1000007ll
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
using namespace std;
long long ans,n,a[N],f[2][N],b[N],g[2][N];
int main()
{
    scanf("%d",&n);
    fo(i,1,n) scanf("%lld",&a[i]),b[i]=max(b[i-1],a[i]);
    int q=1;
    g[0][0]=1;
    fo(i,1,n)
    {
        if(i>=3) g[1-q][b[i-3]]=0;g[1-q][b[i-1]]=1;
        fo(j,1,i)
        {
            f[q][j]=g[q][j]=0;
            f[q][j]=(f[1-q][j]*j+g[1-q][j]*min(j,a[i]-1)+f[1-q][j-1])%mo;
            if(a[i]>j) f[q][j]=(f[q][j]+g[1-q][j-1])%mo;
        }
        q=1-q;
    }
    int ans=0;
    fo(i,1,n) ans=(ans+f[1-q][i])%mo;
    printf("%lld",(ans+1ll)%mo);
}