Blocks(组合数学——生成函数)

Blocks

https://cn.vjudge.net/problem/POJ-3734

题意：

一排有n块砖，每块砖可以染成A,B,C,D四种颜色的其中一种，现在问你A颜色砖块有偶数个，B颜色砖块有偶数个的染色方法有几种？

题解：

这题虽然可以用矩阵快速幂的方法做，但也存在一种能秒杀这题的方法———生成函数!。
要求解这个问题世界上是{0,1,0,1…}和{1,1,1,1…}这两个序列相乘的平方(因为各有两个)，然后第n项系数就是问题的解了。但在计算这个的过程中用普通生成函数是比较难化简出系数的，因此我在这里用指数型生成函数。
{0,1,0,1...}=ex+e−x2\lbrace0,1,0,1...\rbrace=\frac{e^x+e^{-x}}{2}{0,1,0,1...}=2ex+e−x​
{1,1,1,1...}=ex\lbrace1,1,1,1...\rbrace=e^x{1,1,1,1...}=ex
因此生成函数就是
(ex+e−x2∗ex)2=e4x+2∗e2x+14(\frac{e^x+e^{-x}}{2}*e^x)^2=\frac{e^{4x}+2*e^{2x}+1}{4}(2ex+e−x​∗ex)2=4e4x+2∗e2x+1​
因此每一项的系数就是
4nn!∗4+2∗2nn!∗4=4n−1+xn−1n!\frac{4^{n}}{n!*4}+\frac{2*2^{n}}{n!*4}=\frac{4^{n-1}+x^{n-1}}{n!}n!∗44n​+n!∗42∗2n​=n!4n−1+xn−1​
所以答案就是4n−1+2n−14^{n-1}+2^{n-1}4n−1+2n−1，用快速幂求解即可

#include<iostream>
using namespace std;
const int maxn=1e5+5;
const int mod=1e4+7;
long long qpow(long long a,long long b)
{
	long long ans=1;
	while(b>0)
	{
		if(b&1) ans=ans*a%mod;
		a=a*a%mod;
		b>>=1;
	}
	return ans;
}
int main()
{
	int T,n,m,i,j,k;
	cin>>T;
	while(T--)
	{
		cin>>k;
		cout<<(qpow(4,k-1)+qpow(2,k-1))%mod<<endl;	
	 } 
}

转载于:https://www.cnblogs.com/na7-TRZNDP-Z/p/9782453.html