题目链接:https://www.luogu.com.cn/problem/CF1710C
求有多少对 0 ≤ a , b , c ≤ n 0\leq a,b,c\leq n 0≤a,b,c≤n满足 a x o r b , a x o r c , b x o r c a\ xor\ b,a\ xor\ c,b\ xor\ c a xor b,a xor c,b xor c作为边长时能构成一个非退化三角形。
n以二进制形式给出
1 ≤ n < 2 2 × 1 0 5 1\leq n< 2^{2\times 10^5} 1≤n<22×105
我们假设 x = a x o r b , y = a x o r c , z = b x o r c x=a\ xor\ b,y=a\ xor\ c,z=b\ xor\ c x=a xor b,y=a xor c,z=b xor c,会发现有 x x o r y = z x\ xor\ y=z x xor y=z。
我们先默认 m a x ( x , y ) ≤ z max(x,y)\leq z max(x,y)≤z,那么一个合法的状态有 x + y > z x+y>z x+y>z,也就是 x + y > x x o r y x+y>x\ xor\ y x+y>x xor y。
呃我们有 x + y ≥ x x o r y x+y\geq x\ xor\ y x+y≥x xor y,所以我们考虑减去不合法的状态,也就是 x + y = x x o r y x+y=x\ xor\ y x+y=x xor y。
也就是对于每一位来说 x x x和 y y y不能都是 1 1 1,我们对于每一位来说,每次暴力枚举 a , b , c a,b,c a,b,c的取值,设 f i , s f_{i,s} fi,s表示做到第 i i i位,目前 a , b , c a,b,c a,b,c中取到上界的状态为 s s s时的方案数即可。
时间复杂度: O ( log n ) O(\log n) O(logn)
#include
#include
#include
#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;
}