pku_2034_Anti-prime Sequences

一道不錯的搜索入門題.

題目大意:

給三個數,n,m,d,求 在n - m 之間,連續 d個數的合不是素數.如果有多種情況選最小的.

解題思想:

完全的暴力DFS,先打一個素數表,然後DFS對每個要加進來的數判斷這個數與其前面d個數的合是否有素數即可.

源代碼:

#include <myhead.h>

const int M=10011;
const int N=1001;
int n,m,d,num;
bool _hash[N],prim_hash[M];
int result[N];

void prim()
{//素數表
	memset(prim_hash,0,sizeof(prim_hash));
	prim_hash[0]=prim_hash[1]=true;
	for(int i=2;i<=100;++i) {
		if(!prim_hash[i]) {
			for(int j=i+i;j<M;j+=i)
				prim_hash[j]=true;
		}
	}
}

inline bool rec(int index,int value)
{//判斷其前d個數的合是否有素數
	if(index==0)
		return true;
	int left=index-d+1;
	left=max(left,0);
	for(int i=index-1;i>=left;--i) {
		value+=result[i];
		if(!prim_hash[value])
			return false;
	}
	return true;
}

bool dfs(int index)
{//回溯法搜索
	if(index==num)
		return true;
	for(int i=n;i<=m;++i) {
		if(!_hash[i]&&rec(index,i)) {
			_hash[i]=true;
			result[index]=i;
			if(dfs(index+1)) 
				return true;
			_hash[i]=false;
		}
	}
	return false;
}

int main()
{
	prim();
	while(~scanf("%d%d%d",&n,&m,&d),n) {
		num=m-n+1;
		memset(result,0,sizeof(result));
		memset(_hash,0,sizeof(_hash));
		if(dfs(0)) {
			printf("%d",result[0]);
			for(int i=1;i<num;++i)
				printf(",%d",result[i]);
			puts("");
		}else
			puts("No anti-prime sequence exists.");
	}
	return 0;
}