ACM-ICPC 2018 徐州赛区网络预赛 A-dp

题目链接:点击这里

 

解题思路:

好像这题有个O(1)公式,但是看题目的数据范围标程应该也跟我一样是dp.

和原本的相邻位置不能染相同颜色不同的是它是相邻不能位互补.所以要有三个状态:

f[i]表示i个人相邻位置值不互补且头尾值互补的的方案数

F[i]表示i个人相邻位置值不互补且头尾值不互补且头尾值不相同的方案数

G[i]表示i个人相邻位置值不互补且头尾值不互补且头尾值相同的方案数(其实头尾值相同头尾就肯定不是互补的)

头尾就是n个人组成圈的头尾,所以头尾也不能互补.

 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mx = 1e6 + 10;
int n,top,k;
const int mod = 1e9 + 7;
ll fac[mx],f[mx],F[mx],G[mx];
int main()
{
	fac[0] = 1;
	for(int i=1;i<mx;i++) fac[i] = 2*fac[i-1]%mod;
	int t;
	scanf("%d",&t);
	while(t--){
		scanf("%d%d",&n,&k);
		ll inv = fac[k];
		//f是互补,F不互补不相用，G不互补相同 
		f[1] = 0,F[1] = fac[k],G[1] = 0;
		//f[2] = 0,F[2] = fac[k]*(fac[k]-1)%mod;
		for(int i=2;i<=n;i++){
			if(i>2) f[i] = (f[i-1]+F[i-1])%mod;
			F[i] = (f[i-1]*(inv-2)%mod+G[i-1]*(inv-2)%mod)%mod;
			if(k>1){
				if(i!=2) F[i] = (F[i]+F[i-1]*(inv-3)%mod)%mod;
				else F[i] = (F[i]+F[i-1]*(inv-2)%mod)%mod;
			}
			G[i] = (F[i-1]+G[i-1])%mod;
			//cout << f[i] << " " << F[i] << " " << G[i] << endl;
 		}
		printf("%lld\n",(F[n]+G[n]+2*mod)%mod);
	}
	return 0;
}