[FJOI2016] 建筑师

题目描述

小 Z 是一个很有名的建筑师，有一天他接到了一个很奇怪的任务：在数轴上建

小 Z 有很严重的强迫症，他不喜欢有两个建筑的高度相同。另外小 Z 觉得如果从最左边（所有建筑都在右边）看能看到

如果建筑

输入输出格式

输入格式：

第一行一个整数

输出格式：

对于每组数据输出一行答案 $\text{mod } 10^9+7mod 109+7 。$

输入输出样例

输入样例#1：

2
3 2 2
3 1 2

输出样例#1：

2
1

说明

对于 $\%10% 的数据： 1 \leq n \leq 101≤n≤10 。$

对于 $\%20% 的数据： 1 \leq n \leq 1001≤n≤100 。$

对于 $\%40% 的数据： 1 \leq n \leq 50000, \ 1 \leq T \leq 51≤n≤50000, 1≤T≤5 。$

对于 $\%100% 的数据： 1 \leq n \leq 50000, \ 1 \leq A, B \leq 100, \ 1 \leq T \leq 2000001≤n≤50000, 1≤A,B≤100, 1≤T≤200000 。$

/*
    f[i][j][k] -> f[i+1][j+1][k]
    f[i][j][k] -> f[i+1][j][k+1]
    f[i][j][k] * (i-1) -> f[i+1][j][k]
*/
#include<bits/stdc++.h>
#define ll long long
using namespace std;
#define pb push_back
const int ha=1e9+7;
inline void add(int &x,int y){ x+=y; if(x>=ha) x-=ha;}
struct node{ int num,A,B;};
vector<node> g[200005];
int f[2][105][105],T,n,X;
int Y,ans[200005],M,a,b;

inline void solve(){
	f[0][1][1]=1;
	for(int i=2,now=1,pre=0;i<=M;pre=now,now^=1,i++){
		memset(f[now],0,sizeof(f[now]));
		
		for(int j=0;j<=a;j++)
		    for(int k=0;k<=b;k++) if(f[pre][j][k]){
		    	add(f[now][j][k],f[pre][j][k]*(ll)(i-2)%ha);
		    	add(f[now][j+1][k],f[pre][j][k]);
		    	add(f[now][j][k+1],f[pre][j][k]);
			}
		
		node x;
		
		for(int j=g[i].size()-1;j>=0;j--){
			x=g[i][j];
			ans[x.num]=f[now][x.A][x.B];
		}
	}
}

int main(){
	scanf("%d",&T);
	for(int i=1;i<=T;i++){
		scanf("%d%d%d",&n,&X,&Y);
		g[n].pb((node){i,X,Y});
		M=max(M,n),a=max(a,X),b=max(b,Y);
	}
	
	solve();
	
	for(int i=1;i<=T;i++) printf("%d\n",ans[i]);
	return 0;
}

　　但显然正解只能允许 50000 * 100的啦。。。。

我们考虑一下dp方程的转移，对于后两维，每次要么都不变(然后都乘上一个系数)，或者两维之一+1。

所以怎么快速计算这个呢？？我们考虑把后两位合并，最后再考虑两维+1顺序的问题。。。

也就是设 f[i][j] 为已经放了 i个桩子，左+右一共能看见j个的方案数(无序的)。

最后询问 A,B的时候再乘上C（A+B-2，A-1） [枚举两维+1的顺序] 就行啦。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int ha=1e9+7;
inline int add(int x,int y){ x+=y; return x>=ha?x-ha:x;}
inline void ADD(int &x,int y){ x+=y; if(x>=ha) x-=ha;}
int f[50005][233],C[233][233],T,n,A,B;

inline void init(){
	C[0][0]=1;
	for(int i=1;i<=200;i++){
		C[i][0]=1;
		for(int j=1;j<=i;j++) C[i][j]=add(C[i-1][j-1],C[i-1][j]);
	}
	
	f[1][2]=1;
	for(int i=1;i<50000;i++)
	    for(int j=2;j<=200;j++) if(f[i][j]){
	    	ADD(f[i+1][j],f[i][j]*(ll)(i-1)%ha);
	    	ADD(f[i+1][j+1],f[i][j]);
		}
}

int main(){
	init(),scanf("%d",&T);
	
	while(T--){
		scanf("%d%d%d",&n,&A,&B);
		printf("%d\n",f[n][A+B]*(ll)C[A+B-2][A-1]%ha);
	}
	
	return 0;
}

题目描述

输入输出格式

输入输出样例

说明

猜你喜欢