bzoj 2734 [HNOI2012]集合选数 状压DP+预处理

这道题很神啊……

神爆了……

思路大家应该看别的博客已经知道了，但大部分用的插头DP。我加了预处理，没用插头DP，一行一行来，速度还挺快。

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <cmath>
#define N 100100
#define M 50
#define yu 1000000001
#define inf 1<<29
using namespace std;
int n;
long long ans=1;
long long f[2][N/M]={0};
int keneng[N/M],kenengnum;
int sushu[N],sushunum;

void make_sushu(int limit) // 这里不是筛素数，而是把不是2,3的倍数的筛出来 
{                          // 如果筛素数，25不是素数，但是5的矩阵不能包含25 
    int i,j,k;
    int vis[N]={0};
    sushunum=0;
    for (i=1;i<=limit;i++)
        if (i%2!=0&&i%3!=0)
            sushu[++sushunum]=i;
}

bool shifou(int now)    // 预处理的判断，判断是否有两个1相连 
{                         // 即是否取连续的两个数
    int i,j,k;
    i=now&1; now>>=1;
    while (now!=0)
    {
        j=now&1;
        if (i==j&&i&&j)
            return false;
        i=j;
        now>>=1;
    }
    return true;
}

void chuli(int limit)  //预处理，把可以选的状态存起来，省时间 
{
    int i,j,k;
    int maxn=1;
    kenengnum=0;
    for (i=1;i<=limit;i++)
      maxn*=2;
    for (i=0;i<maxn;i++)
    {
        if (shifou(i))
        {
            keneng[kenengnum++]=i;
        }
    }
}

int pd_long(int x) // 由于矩阵每一行长度不同，所以预处理某些不能用 
{                  // 这个判断每一行长度 
    int i=1,zanshi=0;
    while (i<=x)
    {
        zanshi++;
        i*=3;
    }
    return zanshi;
}

long long make_juzhen(int now)
{
    int i,j,k,x=1,y,z;
    int limit,lnow,lnowbefore=inf;
    limit=pd_long(n/now);
    chuli(limit);
    f[0][0]=1;
    for (i=1;x*now<=n;i++,x*=2)
    {
        memset(f[i%2],0,sizeof(f[i%2]));
        lnow=pd_long(n/now/x);  // 当前行的长度 
        for (j=0;j<kenengnum;j++)
        {
            if (keneng[j]>>lnow>0) //如果有选超过当前行长度的，不选 
            {
                f[i%2][j]=0;
                continue;
            }
            for (k=0;k<kenengnum;k++) // 和上一行比较，是否有相连的1选了 
            {
                if (keneng[k]>>lnowbefore>0) // 上一行状态不能超过上一行长度 
                    continue;
                if (keneng[j]&keneng[k])
                    continue;
                f[i%2][j]+=f[(i+1)%2][k];
                f[i%2][j]%=yu;
            }
        }
        lnowbefore=lnow;  // 上一行长度 
    }
    return (f[(i+1)%2][0]+f[(i+1)%2][1])%yu; //最后一行一定只有一个，答案只有这两种情况加起来，最后一个选或不选 
}

int main()
{
    int i,j,k,x,y,z;
    cin>>n;
    make_sushu(n);
    for (i=1;i<=sushunum;i++)
    {
           ans*=make_juzhen(sushu[i]);
           ans%=yu;
    }
    cout<<ans<<endl;
}