GMOJ 6809. 【2020.10.29提高组模拟】不难题

GMOJ 6809. 【2020.10.29提高组模拟】不难题

题目描述

题解

这题真是恶心到我，交了整整15次才切。

这题的做法类似之前做过的求在一个平面直角坐标系中，不经过特定的点到达终点的方案数。

设一个(f[i])表示连续走了(k)次(i)，且(i)之前的的点都做完了的方案数。

暴力转移是(O(n^5))级别的，所以要优化。

• 第一个优化，发现固定左端点，右端点每次递增，很好维护，可以优化到(O(n^4))。
• 第二个优化，利用题目数据随机的条件。因为只有当(y)的每个点都在(x)前面的时候，(f[y])才可以转移到(f[x])，所以可以很容易发现，随着区间右端点(R)的增大，(x)能走到的点越来越少，所以只需要每次记录一下(x)可以走到点，就能大大降低时间复杂度。
• 但是这样还是会超时，看了LAZA的博客之后，我又发现了另外一种优化方法，可以发现如果(R)每次都从(L)开始枚举，是很慢的。所以我们不妨预处理的时候将(R=L)的情况处理了，直接从(L+1)开始枚举，这样就能过了。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 301
#define mo 1000000007
#define ll long long
#pragma GCC optimize(3)
using namespace std;
int n,k,i,j,l,r,a[N][N],f[N],x,y,p[2][N][N],wp[N][N],xp;
ll fac[N*N],inv[N*N],ans,sum[N][N],sum1[N][N],pl[2][N][N],pl1[2][N][N];
ll mi(ll x,ll n){
	if (n==1) return x;
	ll sum=mi(x*x%mo,n/2);
	if (n%2) sum=sum*x%mo;
	return sum;
}
int main(){
	freopen("nothard.in","r",stdin);
	freopen("nothard.out","w",stdout);
	scanf("%d%d",&k,&n);
	fac[0]=1;
	for (i=1;i<=n*k;i++) fac[i]=fac[i-1]*i%mo;
	inv[n*k]=mi(fac[n*k],mo-2);
	for (i=n*k-1;i>=1;i--) inv[i]=inv[i+1]*(i+1)%mo;
	inv[0]=1;
	for (i=1;i<=k;i++){
		for (j=1;j<=n;j++) scanf("%d",&a[i][j]),wp[a[i][j]][i]=j;
		a[i][n+1]=n+1;wp[n+1][i]=n+1;
	}
	for (l=1;l<=k;l++){
		xp=0;
		for (i=1;i<=n+1;i++)
			sum[i][l]=wp[i][l]-1,sum1[i][l]=inv[wp[i][l]-1],p[xp][i][0]=0;
		for (i=1;i<=n+1;i++)
			for (j=1;j<=n+1;j++)
				if (wp[j][l]<wp[i][l]){
					p[xp][i][++p[xp][i][0]]=j;
					pl[xp][i][p[xp][i][0]]=wp[i][l]-wp[j][l]-1;
					pl1[xp][i][p[xp][i][0]]=inv[wp[i][l]-wp[j][l]-1];
				}
		for (r=l+1;r<=k;r++){
			xp^=1;
			for (i=1;i<=n+1;i++){
				x=a[l][i];
				sum[x][r]=sum[x][r-1]+wp[x][r]-1;
				sum1[x][r]=sum1[x][r-1]*inv[wp[x][r]-1]%mo;
				f[x]=fac[sum[x][r]]*sum1[x][r]%mo*fac[r-l+1]%mo;
				p[xp][x][0]=0;
				for (j=1;j<=p[xp^1][x][0];j++)
					if (wp[p[xp^1][x][j]][r]<wp[x][r]){
						y=p[xp^1][x][j];
						p[xp][x][++p[xp][x][0]]=y;
						pl[xp][x][p[xp][x][0]]=pl[xp^1][x][j]+wp[x][r]-wp[y][r]-1;
						pl1[xp][x][p[xp][x][0]]=pl1[xp^1][x][j]*inv[wp[x][r]-wp[y][r]-1]%mo;
						f[x]=(f[x]+mo-f[y]*fac[pl[xp][x][p[xp][x][0]]]%mo*pl1[xp][x][p[xp][x][0]]%mo*fac[r-l+1]%mo)%mo;
					}
			}
			ans=(ans+f[n+1]*inv[r-l+1])%mo;
		}
	}
	printf("%lld
",ans);
	return 0;
}