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);
}