蓝桥杯试题 算法训练 Sereja and Equality （已AC）

蓝桥杯试题 算法训练 Sereja and Equality

资源限制
时间限制：1.0s 内存限制：512.0MB
问题描述
　　(注：这是codechef上的官方翻译)
　　佳佳称两个长度为n的数组A,B相似，如果对于所有i(1≤i≤n)，满足C(A,Ai)=C(B,Bi)。其中C(X,x)等于满足X[j]<x(1≤j≤n)的j的数目。
　　对于两个排列P1,P2，佳佳定义函数F(P1,P2)等于满足P1[l…r]相似于P2 [l…r] (1≤l≤r≤n)并且P1[l…r]包含不超过E个逆序对的数对(l,r)的数目。
　　现在佳佳对下面这个问题发生了兴趣：对P1,P2取遍所有n个元素的排列F(P1,P2)的总和是多少。
输入格式
　　输入数据的第一行包含一个整数T——测试数据的组数。
　　对于每组测试数据，仅包含一行两个整数n,E。
输出格式
　　对于每组测试数据，输出一行表示结果。答案对109+7取模。
样例输入
4
2 2
2 1
2 0
1 1
样例输出
10
10
9
1

题解
当比较区间长度定下来之后方案数就跟[ l , r ] 这个区间的位置无关了。
于是我们可以直接枚举区间的长度来计算贡献。
但是光这样是不够的。
我们需要递推出1~i 的所有排列中逆序对数为j jj的排列种类数。
这个貌似就是记录一个前缀和就行了。
f [ i ] [ j ] = ∑ f [ i − 1 ] [ k ] 且j − i + 1 ≤ k ≤ j
证明就是考虑已经处理好了1~i − 1的排列。
在所有位置插入i ii的贡献。
这个思想参见[HAOI2009]逆序对数列。
剩下的递推就是简单组合数学了。

代码

#include<bits/stdc++.h>
#define N 505
#define mod 1000000007
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
int C[N][N],f[N][N*N],T,n,k,ans,fac[N];
inline void init(){
	fac[0]=1;
	for(int i=1;i<=500;++i)fac[i]=1ll*fac[i-1]*i%mod;
	for(int i=1;i<=500;++i)C[i][0]=C[i][i]=1,C[i][1]=i;
	for(int i=2;i<=500;++i)for(int j=1;j<=i;++j){
		C[i][j]=C[i-1][j]+C[i-1][j-1];
		if(C[i][j]>=mod)C[i][j]-=mod;
	}
	int sum=0;
	f[1][0]=1;
	for(int i=2;i<=500;++i){
		sum=0;
		for(int j=0;j<=(i-1)*i/2;++j){
			sum+=f[i-1][j];
			if(sum>=mod)sum-=mod;
			f[i][j]=sum;
			if(j+1-i>=0)sum=(sum-f[i-1][j+1-i]+mod)%mod;
		}
	}
	for(int i=2;i<=500;++i)for(int j=1;j<=(i-1)*i/2;++j){
		f[i][j]+=f[i][j-1];
		if(f[i][j]>=mod)f[i][j]-=mod;
	}
}
int main(){
	init(),T=read();
	while(T--){
		n=read(),k=read(),ans=0;
		for(int i=1;i<=n;++i){
			int tmp=1ll*C[n][i]*C[n][i]%mod*f[i][min(i*(i-1)/2,k)]%mod*fac[n-i]%mod*fac[n-i]%mod*(n-i+1)%mod;
			ans+=tmp;
			if(ans>=mod)ans-=mod;
		}
		printf("%d\n",ans);
	}
	return 0;
}