BZOJ 4798: [Ceoi2015]Calvinball championship

Description

将所有人分组，每个组的编号是这个组中名字字典序最小的人的排名，将所有人名字按字典序排名后，求这是第几大的序列。

Solution

DP.

\(f[i][j][0/1]\)表示当前第\(i\)位，最高位为\(j\)，是否顶格...

转移跟数位DP差不多...

因为我一开始把极限写成了\(mx\)...WA...

没滚数组...MLE...

取模太多被卡常...TLE...

新人求助，本地AC提交TLE...我发现本地\(0.2s 10^8\)

我也很绝望啊...

Code

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
const int N = 10050;
const int p = 1000007;

inline void Add(int &x,LL y) { x=(x+y)%p; }

int n,mx,cur;
int a[N];
int f[2][N][2];

int main() {
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	cur=mx=f[1][1][1]=1;
	for(int i=2;i<=n;i++) {
		cur^=1;
		for(int j=1;j<=i;j++) {
			f[cur][j][0]=f[cur][j][1]=0;
			Add(f[cur][j][0],(LL)f[cur^1][j][0]*j%p+f[cur^1][j-1][0]);
			if(j==mx) Add(f[cur][j][0],f[cur^1][j][1]*(a[i]-1));
		}mx=max(mx,a[i]),f[cur][mx][1]=1;
//		for(int j=1;j<=n;j++) printf("%d,%d%c",f[cur][j][0],f[cur][j][1]," \n"[j==n]);
	}
	int ans=0;
	for(int i=1;i<=n;i++) Add(ans,f[cur][i][0]);
	printf("%d\n",(ans+1)%p);
	return 0;
}