正题
题目链接:https://www.luogu.com.cn/problem/CF1710C
题目大意
求有多少对0≤a,b,c≤n0\leq a,b,c\leq n0≤a,b,c≤n满足axorb,axorc,bxorca\ xor\ b,a\ xor\ c,b\ xor\ ca xor b,a xor c,b xor c作为边长时能构成一个非退化三角形。
n以二进制形式给出
1≤n<22×1051\leq n< 2^{2\times 10^5}1≤n<22×105
解题思路
我们假设x=axorb,y=axorc,z=bxorcx=a\ xor\ b,y=a\ xor\ c,z=b\ xor\ cx=a xor b,y=a xor c,z=b xor c,会发现有xxory=zx\ xor\ y=zx xor y=z。
我们先默认max(x,y)≤zmax(x,y)\leq zmax(x,y)≤z,那么一个合法的状态有x+y>zx+y>zx+y>z,也就是x+y>xxoryx+y>x\ xor\ yx+y>x xor y。
呃我们有x+y≥xxoryx+y\geq x\ xor\ yx+y≥x xor y,所以我们考虑减去不合法的状态,也就是x+y=xxoryx+y=x\ xor\ yx+y=x xor y。
也就是对于每一位来说xxx和yyy不能都是111,我们对于每一位来说,每次暴力枚举a,b,ca,b,ca,b,c的取值,设fi,sf_{i,s}fi,s表示做到第iii位,目前a,b,ca,b,ca,b,c中取到上界的状态为sss时的方案数即可。
时间复杂度:O(logn)O(\log n)O(logn)
code
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=2e5+10,P=998244353;
ll n,f[N][8],ans;
char s[N];
signed main()
{scanf("%s",s+1);n=strlen(s+1);f[0][7]=1;for(ll i=1;i<=n;i++){ans=(ans*2ll+(s[i]-'0'))%P;for(ll j=0;j<8;j++){for(ll k=0;k<8;k++){ll a=k&1,b=(k>>1)&1,c=(k>>2)&1;if(s[i]=='0'&&!(j&k)){if((a^b)&&(b^c)&&(a^c^1)||(a^b)&&(b^c^1)&&(a^c)||(a^b^1)&&(b^c^1)&&(a^c^1))(f[i][j]+=f[i-1][j])%=P;}else if(s[i]=='1'){if((a^b)&&(b^c)&&(a^c^1)||(a^b)&&(b^c^1)&&(a^c)||(a^b^1)&&(b^c^1)&&(a^c^1))(f[i][j&k]+=f[i-1][j])%=P;}}}}ans++;ans=(ans*ans%P*ans%P+2ll*ans+3*ans*(ans-1)%P)%P;for(ll i=0;i<8;i++)(ans+=P-f[n][i]*3ll%P)%=P;printf("%lld\n",ans);return 0;
}