Codeforces Round #228 Div.1 D 数位dp+数学

题意

设S为非负整数集。若对任意$a,b\in S$有$a\oplus b\in S$，则称S是好的。其中$a\oplus b$表示a异或b。问有多少个集合S满足S是好的且S中的最大值不超过n。
n<=10^9，答案模1e9+7输出。

分析

14年集训队论文《浅谈线性相关》的论文题。
根据定义不难得到一个好的集合必然是一个线性空间。那么我们就可以用一组基来表示这个集合。现在问题在于如何使得每个线性空间找到唯一一个基来对应并求出满足条件的基的数量。
这个可以用数位dp来做。
设f[i,j,0/1]表示从高到低做到第i位，已经加入了j个向量，且当前线性空间的最大值是否等于n的方案。
有两种决策：一种是加入一个新向量，且该向量的最高位1为第i-1位。这时第i-1位就可以被我们自由决策，于是其他向量的第i-1位为0。也就是转移到f[i-1,j+1]
第二种是不加入新向量，这时就考虑已经加入的j个向量的第i-1位放1还是0。不难发现有$2^{j-1}$种方法使得线性空间最大值的第i-1位为0，其余$2^{j-1}$种方法使得最大值第i-1位为1。直接转移即可。
注意这里线性空间的最大值实际就是所有向量的异或和。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int MOD=1000000007;

int n,bin[35],f[35][35][2];

void updata(int &x,int y)
{
    x+=y;x-=x>=MOD?MOD:0;
}

int main()
{
    scanf("%d",&n);
    bin[0]=1;
    for (int i=1;i<=30;i++) bin[i]=bin[i-1]*2;
    for (int i=30;i>=0;i--)
        if (n&bin[i]) {f[i][1][1]=f[i][0][0]=1;break;}
    for (int i=30;i>=1;i--)
        for (int j=0;j<=30;j++)
        {
            if (f[i][j][0])
            {
                updata(f[i-1][j+1][0],f[i][j][0]);
                updata(f[i-1][j][0],(LL)f[i][j][0]*bin[j]%MOD);
            }
            if (f[i][j][1])
            {
                if (n&bin[i-1])
                {
                    updata(f[i-1][j+1][1],f[i][j][1]);
                    updata(f[i-1][j][0],(LL)f[i][j][1]*bin[j-1]%MOD);
                    updata(f[i-1][j][1],(LL)f[i][j][1]*bin[j-1]%MOD);
                }
                else updata(f[i-1][j][1],(LL)f[i][j][1]*bin[j-1]%MOD);
            }
        }
    int ans=0;
    if (!n) ans++;
    for (int i=0;i<=30;i++) updata(ans,f[0][i][0]),updata(ans,f[0][i][1]);
    printf("%d",ans);
    return 0;
}