[2022CCPC广州] XOR Sum

合集 - 个人记录(16)

1.CF1209G2 Into Blocks (hard version)02-102.「POJ 3744」Scout YYF I02-103.记录trick02-104.洛谷P10936 导弹防御塔02-105.CF1635E Cars02-106.P872602-107.CF633G02-108.CF1436E02-109.[BJOI2017] 喷式水战改02-1110.「CF407E」k-d-sequence02-1311.「雅礼集训 2017 Day5」珠宝02-1312.AT_jsc2019_final_h Distinct Integers02-1613.CF1430G Yet Another DAG Problem02-1714.[2022CCPC广州] B Ayano and sequences02-2015.[2022CCPC广州] Infection02-20

16.[2022CCPC广州] XOR Sum02-20

收起

这个肯定要分二进制位来做了，所以考虑数位dp，先看要放哪些东西进dp状态：现在处理到第几位了，有多少个数现在是顶着上界的，还差多少值和才能到 $n$。现在看怎么转移。$k\le 18$ ，可以直接枚举这一位上有多少个 $1$ ，当第 $t$ 位有 $x$ 个 $1$ 时，对和的贡献是 $2^t\times x\times (k-x)$ ，这是显然的。对于 $m$ 的第 $t$ 位是否为 $1$,要进行分类讨论，如果是，就可以从顶着上界的数里面选 $1$ ，然后被选为$1$ 的继续顶着上界。如果$m$ 的第 $t$ 位不为为 $1$，则只能从没有顶着上界的数里面选 $1$。两种情况的答案都要乘组合数。但是直接这样写是要 TLE 的，加入剪枝：如果后面所有位都按照最优的情况来选（即为 $1$ 和 $0$ 各一半）和还是小于 $n$ 就直接返回 $0$ ，就是这两行：

	int s1=k/2,s0=k-s0;
	if(((1ll<<t+1)-1)*s0*s1<tot)return 0;

时间复杂度证明可以看这里

完整代码：

#include <bits/stdc++.h>
#define int long long
#define ll long long
#define ull unsigned long long
#pragma GCC optimeze(3)
#pragma GCC optimeze(2)
#define PII pair<int, int>
#define pb push_back
#define fi first
#define se second
#define lowbit(x) (x & (-x))
#define inv(x) (qpow(x,mod-2))
#define lwz lower_bound
#define blong(i) ((i+K-1)/K)
using namespace std;
const int N=2e3+5;
const int M=3e2+5;
const int mod=1e9+7;
double eps=1e-6;
inline int read(){
	char ch=getchar();bool f=0;int x=0;
	for(;!isdigit(ch);ch=getchar())if(ch=='-')f=1;
	for(;isdigit(ch);ch=getchar())x=(x<<1)+(x<<3)+(ch^48);
	if(f==1)x=-x;return x;
}
ll qpow(ll a,ll b){
	ll ans=1;
	while(b){
		if(b&1)ans*=a,ans%=mod;
		a*=a,a%=mod,b>>=1;
	}
	return ans;
}
int gcd(int a,int b){return b==0? a:gcd(b,a%b);}
void add(int&a,int b){a+=b;if(a>=mod)a-=mod;}
void minus(int&a,int b){a-=b;if(a<0)a+=mod;}
int n,k,m,a[40],c[40][40];
unordered_map<int,int>dp[40][40];
int dfs(int t,int lim,int tot){
	if(t<0)return tot==0;
	if(tot<0)return 0;
	int s1=k/2,s0=k-s0;
	if(((1ll<<t+1)-1)*s0*s1<tot)return 0;
	if(dp[t][lim].count(tot))return dp[t][lim][tot];
	int ret=0;
	if(a[t]==1){
		for(int i=0;i<=lim;i++){
			for(int j=0;j<=k-lim;j++){
				int val=(1ll<<t)*(i+j)*(k-i-j);
				add(ret,dfs(t-1,i,tot-val)*c[lim][i]%mod*c[k-lim][j]%mod);
			}
		}
	}
	else{
		for(int i=0;i<=k-lim;i++){
			int val=(1ll<<t)*i*(k-i);
			add(ret,dfs(t-1,lim,tot-val)*c[k-lim][i]%mod);
		}
	}
	dp[t][lim][tot]=ret;
	return ret;
}
signed main(){ 
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n>>m>>k;
	for(int i=0;i<=39;i++){
		a[i]=((m>>i)&1);
	}
	for(int i=0;i<=k;i++){
		c[i][0]=1;
		for(int j=1;j<=i;j++){
			c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
		}
	}
	cout<<dfs(39,k,n);
	return 0;
}