【CF 338D】Fox and Perfect Sets（线性基）（数位DP）

传送门

---

题解：

显然所有合法集合可以直接用线性基表示出来。

考虑怎么唯一表示，显然是将线性基高斯消元搞成下三角矩阵。

当这一位上有基的时候，不允许线性基中其他位置出现这一位上的$1$，否则上面的位置可以任意选择。

从高位向低位数位dp，设状态$f[i][j][0/1]$，表示处理了前$i$个高位，线性基中已经有$j$个元素，当前的最大xor和是否卡$k$的上界。

然后就是xjb转移就行了。

---

代码：

#include<bits/stdc++.h>
#define re register

const int mod=1e9+7;
int k,len;
int lim[35];
long long dp[35][35][2],ans;

signed main(){
	scanf("%d",&k);
	if(k==0)return !printf("1\n");
	while(k){
		lim[++len]=k&1;
		k>>=1;
	}std::reverse(lim+1,lim+len+1);
	dp[0][0][1]=1;
	for(int re i=0;i<len;++i){
		for(int re j=0;j<=i;++j)
		for(int re k=0;k<2;++k)if(dp[i][j][k]){
			dp[i+1][j][k&(lim[i+1]^1)]+=dp[i][j][k]<<(j?j-1:0);
			if(!k||lim[i+1]){
				if(j)dp[i+1][j][k]+=dp[i][j][k]<<j>>1;
				dp[i+1][j+1][k]+=dp[i][j][k];
			}
		}
		for(int re j=0;j<=i+1;++j)
		for(int re k=0;k<2;++k)dp[i+1][j][k]%=mod;
	}
	for(int re i=0;i<=len;++i)ans+=dp[len][i][0]+dp[len][i][1];
	printf("%lld",ans%mod);
	return 0;
}