foj2200 n个人的环取k人且任意两个人的距离不能为2的方法数

最新推荐文章于 2020-12-22 19:48:59 发布

ccyy-

最新推荐文章于 2020-12-22 19:48:59 发布

阅读量595

点赞数

分类专栏： dp 困难

本文链接：https://blog.csdn.net/cacyth/article/details/48938095

版权

困难同时被 2 个专栏收录

109 篇文章 0 订阅

订阅专栏

47 篇文章 0 订阅

订阅专栏

写了一晚上，思路越来越像正确的方向靠近，只不过又犯了老错误——更改思路将第三维由2改成4后居然忘了改定义数组的部分！搞得我看了半天都不明白为什么本该加成0的变量会变成1。。。吐血ing

枚举前两个数取或不取的情况，相应的对第三位开始dp，dp[i][j][k]表示已经判别到第i位时共取了j个合法的位置且i-1位和i位选取状态由k的二进制表示。注意不预处理会超时。

#include<iostream>
#include<algorithm>
#include<math.h>
#include<stdio.h>
#define mod 1000000007
#define ll long long
using namespace std;
ll dp[1003][1003][4];
ll s[1003][1003];
int main(){
	int n,k;
	n=1000,k=1000;
	memset(dp,0,sizeof(dp));
	dp[2][0][0]=1;
	for(int i=3;i<=n;++i){
		for(int j=0;j<=min(k,i);++j){
			dp[i][j][0]=(dp[i][j][0]+dp[i-1][j][0]+dp[i-1][j][2])%mod;
	if(j>0) dp[i][j][1]=(dp[i][j][1]+dp[i-1][j-1][0])%mod;
			dp[i][j][2]=(dp[i][j][2]+dp[i-1][j][1]+dp[i-1][j][3])%mod;
	if(j>0) dp[i][j][3]=(dp[i][j][3]+dp[i-1][j-1][1])%mod;	
			s[i][j]=(s[i][j]+dp[i][j][0]+dp[i][j][1]+dp[i][j][2]+dp[i][j][3])%mod;
		} 
	}
	///
	memset(dp,0,sizeof(dp));
	dp[2][1][1]=1;
	for(int i=3;i<=n;++i){
		for(int j=0;j<=min(k,i);++j){
			dp[i][j][0]=(dp[i][j][0]+dp[i-1][j][0]+dp[i-1][j][2])%mod;
	if(j>0) dp[i][j][1]=(dp[i][j][1]+dp[i-1][j-1][0])%mod;
			dp[i][j][2]=(dp[i][j][2]+dp[i-1][j][1]+dp[i-1][j][3])%mod;
	if(j>0) dp[i][j][3]=(dp[i][j][3]+dp[i-1][j-1][1])%mod;
			s[i][j]=(s[i][j]+dp[i][j][0]+dp[i][j][2])%mod;
		} 
	}
	///
	memset(dp,0,sizeof(dp));
	dp[2][1][2]=1;
	for(int i=3;i<=n;++i){
		for(int j=0;j<=min(k,i);++j){
			dp[i][j][0]=(dp[i][j][0]+dp[i-1][j][0]+dp[i-1][j][2])%mod;
	if(j>0) dp[i][j][1]=(dp[i][j][1]+dp[i-1][j-1][0])%mod;
			dp[i][j][2]=(dp[i][j][2]+dp[i-1][j][1]+dp[i-1][j][3])%mod;
	if(j>0) dp[i][j][3]=(dp[i][j][3]+dp[i-1][j-1][1])%mod;
			s[i][j]=(s[i][j]+dp[i][j][0]+dp[i][j][1])%mod;
		} 
	}
	///
	memset(dp,0,sizeof(dp));
	dp[2][2][3]=1;
	for(int i=3;i<=n;++i){
		for(int j=0;j<=min(k,i);++j){
			dp[i][j][0]=(dp[i][j][0]+dp[i-1][j][0]+dp[i-1][j][2])%mod;
	if(j>0) dp[i][j][1]=(dp[i][j][1]+dp[i-1][j-1][0])%mod;
			dp[i][j][2]=(dp[i][j][2]+dp[i-1][j][1]+dp[i-1][j][3])%mod;
	if(j>0) dp[i][j][3]=(dp[i][j][3]+dp[i-1][j-1][1])%mod;
			s[i][j]=(s[i][j]+dp[i][j][0])%mod;
		} 
	}
	int t;
	scanf("%d",&t); 
	while(t--){
		scanf("%d%d",&n,&k);
		printf("%I64d\n",s[n][k]);
	}
	return 0;
}