YBTOJ3的倍数（数位dp）

YBTOJ3的倍数（数位dp）

这应该是全网唯一一篇这道题的题解了吧……
ybt上的题解写的真的不是人话……
首先明确几个概念：

1. 按位或：两数对应位中有一个是1结果就为1

2. 一个数满足是3的倍数，当且仅当其奇数位上的1与其偶数位上的1在模3的意义下相等

3. 容斥原理
   
   因为看到3的倍数要考虑二进制下1的位置，所以我们先要把t二进制拆分一下
   并统计出其奇位1和偶位1的个数
   然后我们考虑容斥，枚举某些位的或结果本该为1（即t二进制下为1的位数），我们强制让它或的结果为0（即枚举时枚举0的位数），并统计方案数、
   我们枚举容斥掉的位数中，奇位和偶位分别有a和b个1（即对应把t二进制拆分后奇位和偶位1的个数），然后dp：
   $f_{i,j}$表示有i个奇位1，j个偶位1被强制变为0的合法方案数
   这样我们就可以推出$f$的转移方程:
   $f_{i,j}={\Sigma }_{0,i}{\Sigma }_{0,j}{C}_{i,k}\ast Ck,j$

   其中C为从i里选出k个1的组合数
   我们再枚举容斥掉的位中，有i个奇位1和j个偶位1又变回了1。
   可以算出这些位强制变成0之后，每个x有$f_{a-i,b-j}$种方案，有n个x，所以将$f_{i,j}$n次幂乘上（-1）^（i+j）（根据容斥原理），再乘两个组合数，即选出i个奇位和j个偶位1的组合数，计入答案即可。
   以上是本题做法，接下来看讲讲原因。
   看完上面的做法你肯定要问了
   为什么要用容斥原理？
   为什么要把1强制变成0？
   怎么保证保证按位或的结果就是t？
   。 。 。 。 。 。
   首先要分解乘二进制数，肯定涉及到1的奇偶位问题
   但是我们又不好处理0的位置，
   那我们索性就把所有的1都变成0，然后用容斥原理来解决1的问题
   这样问题就迎刃而解了
   AC代码：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int mod=998244353;
int n,t,m,a,b,ans,f[100][100],c[100][100];
inline int read()
{
	int ret=0;char ch=' ',c=getchar();
	while(!(c<='9'&&c>='0')) ch=c,c=getchar();
	while(c<='9'&&c>='0') ret=(ret<<1)+(ret<<3)+c-'0',c=getchar();
	return ch=='-'?-ret:ret;
}
int qpow(int a,int b)
{
	int ret=1;
	while(b)
	{
		if(b&1) ret=ret*a%mod;
		a=a*a%mod;
		b>>=1;
	}
	return ret;
}
signed main()
{
	n=read(),t=read();
	while(1ll<<m+1<=t) m++;
	for(int i=0;i<=m;i++)
	{
		if(t>>i&1)
		{
			if(i&1) a++;
			else b++;
		}
	}
	for(int i=0;i<=m+2;i++)
	{
		for(int j=0;j<=i;j++)
		{
			if(j==0) c[i][j]=1;
			else c[i][j]=(c[i-1][j]%mod+c[i-1][j-1]%mod);
		}
	}
	for(int i=0;i<=a;i++)
	{
		for(int j=0;j<=b;j++)
		{
			for(int k=0;k<=i;k++)
				for(int h=0;h<=j;h++)
					if(k%3==h%3) f[i][j]=(f[i][j]%mod+c[i][k]*c[j][h]%mod)%mod;
			f[i][j]=qpow(f[i][j],n);
		}
	}
	for(int i=0;i<=a;i++)
	{
		for(int j=0;j<=b;j++)
		{
			int res=c[a][i]*c[b][j]%mod*f[a-i][b-j]%mod;
			if(i+j&1) ans=((ans-res)%mod+mod)%mod;
			else ans=(ans+res)%mod;
		}
	}
	printf("%lld",ans);
	return 0;
}