Description
你真的认为选课是那么容易的事吗?HYSBZ的ZY同志告诉你,原来选课也会让人产生一种想要回到火星的感觉。假设你的一周有n天,那么ZY编写的选课系统就会给你n堂课。但是该系统不允许在星期i和星期i+1的时候选第i堂课,也不允许你在星期n和星期一的时候选第n堂课。然后连你自己也搞不清哪种选课方案合法,哪种选课不合法了。你只想知道,你到底有多少种合法的选课方案。
Input
有多组数据,请读到文件末结束。
对于每组数据仅一行,1个正整数 n。
Output
对于每组输出只有一行,1个非负整数,为选课方案数量 mod (10^9+7).
Sample Input
2
4
Sample Output
0
2
【样例解释】
对于样例二:
周一上第二堂课,周二上第三堂课,周三上第四堂课,周四上第一堂课;
周一上第三堂课,周二上第四堂课,周三上第一堂课,周四上第二堂课。
共2种选课方案。
Data Constraint
对于第i组数据,n<=10*i, 1<=i<=3
对于100%的数据:1<=n<=100000
思路
容斥原理,假设有 1 堂课所在的天数是错误的情况数是 W(1),则每个错误的课程 i 可以放在 i 或 i+1 天,剩下的选课方法数是(n-1)!。
所以 w(1)=C(n,1)2(n-1)!,这一步很容易想到,有点坑的是 2 个以上的情况,假设有 k 堂课所在的天数是错误的情况数是 W(K)
总共 n 堂课分别记为 1,2,…n,它们可放的天数可以表示为(1,2)(2,3)(3,4)…(n,1)
现在我们把括号去掉即得到一个数列 1,2,2,3,3,4,….n,1
现在我们从里面取出 K 个数,分别表示 k 堂课所在的天数,现在只要求出满足这个条件的取法数就可以了。
定理:假定数n个顶点沿一圆周排列,则从其中选取k个不相邻顶点的方法数是n/k*C(n-k-1,k-1)
证明:对于任意一个顶点A,先取A,然后再从不和A相邻的n-3个其他顶点中取k-1个不相邻顶点,显然可得到符合定理要求的组合,这个组合的个数为C((n-3)-(k-1)+1,k-1)=C(n-k-1,k-1)。一共有n个顶点,而且在每个组合中有k个元素,即可完成证明。
那么W(K)=(n-k)!*C(2n-k-1,k-1)*2n/k
W(K)确定以后就是最裸的容斥原理了。
代码
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=300077,mod=1e9+7;
long long fac[N],unfac[N],ny[N],ass,n;
long long power(long long x,int y)
{
long long s=1;
while(y)
{
if(y&1) s=(s*x)%mod;
x=x*x%mod;
y>>=1;
}
return s;
}
void init()
{
fac[0]=unfac[0]=1;
fac[1]=1; unfac[1]=1; ny[1]=1;
for(int i=2; i<=N-50; i++) fac[i]=fac[i-1]*i%mod,unfac[i]=power(fac[i],mod-2)%mod,ny[i]=power(i,mod-2)%mod;
}
long long C(int x,int y)
{
return fac[y]*unfac[x]%mod*unfac[y-x]%mod;
}
int main()
{
init();
while(scanf("%lld",&n)!=EOF)
{
ass=0;
if(n<=2)
{
printf("0\n"); continue;
}else
{
for(int i=0; i<=n; i++)
{
if(i&1) ass=(ass-2*n*ny[2*n-i]%mod*C(i,2*n-i)%mod*fac[n-i])%mod; else
ass=(ass+2*n*ny[2*n-i]%mod*C(i,2*n-i)%mod*fac[n-i])%mod;
}
printf("%lld\n",(ass+mod)%mod);
}
}
}