POJ 2034 Anti-prime Sequences(素数预处理+DFS回溯)

POJ 2034 Anti-prime Sequences(素数预处理+DFS回溯)

http://poj.org/problem?id=2034

题意:

        给你一个n,n+1,n+2,n+3,…m的数字序列,要你重新安排该序列的顺序,使得任意连续i个(2<=i<=d)个数的和是合数.且输出字典序最小的该序列即可.

分析:

        首先用筛选法预处理出1000*10以内的所有素数,然后再用dfs回溯构造出字典序最小的合法序列.

        其中程序中用ans保存了所求序列，用sum保存了连续和,即sum[0]=0表示空序列的连续和，sum[i]表示当前所选的数a1(即ans[0])+a2+a3…+ai 的和。

        在dfs中我们首先要判断我们上一步添加完当前ans的最后一个数之后,序列是否满足任意2到d的连续个数和不是合数.然后才是回溯构造下一个可行的数.

        注意：假设当前构造了第cnt个数，那么我们如何判断[0,cnt-1]范围内的连续数和都是合数呢？因为我们经过上一步DFS已经知道了[0,cnt-2]内的所有连续数和肯定符合要求，新添加了一个数a[cnt-1]做结尾，所以我们只需要判断[0, cnt-1]的所有后缀的和是否是合数即可。(想想是不是)

        注意代码实现细节.我用的是vector,如果用数组表示ans和sum的话应该速度更快.

AC代码:

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn=10000+5;
int prime[maxn];//prime[i]==0表示 i为素数
vector<int> ans;
vector<int> sum;
int n,m,d;
bool vis[1000+5];
bool DFS(int cnt,int all)//当前已经构造了cnt个数,且这cnt个数的和等于all
{
    if(cnt>=2)//cnt>=2时,才需要判断连续i个值的和是否为合数. 且这里我们只需要判断后缀即可
    {
        for(int len=2;len<=min(cnt,d);len++)
        {
            int pre_sum = sum[cnt-len];          //求出前cnt-len个数的和,即sum[cnt-len]
            if(!prime[all-pre_sum]) return false;//长度为len的后缀和是素数,非法
        }
    }
    if(cnt==m-n+1) return true;
    for(int i=n;i<=m;i++)if(!vis[i])
    {
        vis[i]=1;
        ans.push_back(i);
        int temp_sum= all+i;
        sum.push_back(temp_sum);
        if(DFS(cnt+1,all+i)) return true;
        else
        {
            vis[i]=0;
            ans.pop_back();
            sum.pop_back();
        }
    }
    return false;
}

int main()
{
    for(int i=2;i<maxn;i++)if(prime[i]==0)
    {
        for(int j=i*i; j<maxn; j+=i)
            prime[j]=1;
    }
    while(scanf("%d%d%d",&n,&m,&d)==3&&n)
    {
        memset(vis,0,sizeof(vis));
        ans.clear();
        sum.clear();
        sum.push_back(0);
        if(DFS(0,0))
        {
            printf("%d",ans[0]);
            for(int i=1;i<ans.size();i++)
                printf(",%d",ans[i]);
            printf("\n");
        }
        else printf("No anti-prime sequence exists.\n");
    }
    return 0;
}