首页 > 代码库 > BZOJ 3329 Xorequ 数位DP+矩阵乘法
BZOJ 3329 Xorequ 数位DP+矩阵乘法
题目大意:给定n,求[1,n]以内以及[1,2^n]以内有多少x满足x^3x=2x
x^3x=2x等价于x^2x = 3x
而3x=x+2x 且2x=x<<1
故x满足条件当且仅当x&(x<<1)=0
故x的二进制拆分中任意两个1不相邻
令f[i]为i位数中最高位为0的满足条件的数的数量
g[i]为i位数中最高位为1的满足条件的数的数量
则显然有
f[i+1]=f[i]+g[i]
g[i+1]=f[i]
于是第一问数位DP 第二问矩阵乘法即可
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #define MOD 1000000007 using namespace std; typedef long long ll; struct Matrix{ ll xx[2][2]; Matrix(ll _,ll __,ll ___,ll ____) { xx[0][0]=_; xx[0][1]=__; xx[1][0]=___; xx[1][1]=____; } ll* operator [] (int x) { return xx[x]; } }; ll f[70],g[70]; void operator *= (Matrix &x,Matrix &y) { int i,j,k; Matrix z(0,0,0,0); for(i=0;i<2;i++) for(j=0;j<2;j++) for(k=0;k<2;k++) z[i][j]+=x[i][k]*y[k][j],z[i][j]%=MOD; x=z; } ll Digital_DP(ll x) { int i,temp=0; long long re=0; for(i=0;1ll<<i<=x;i++); for(;i;i--) { if( x&(1ll<<i-1) ) { re+=f[i]; if(temp) return re-1; temp=1; } else temp=0; } return re-1; } ll Matrix_Mutiplication(ll y) { Matrix a(1,0,0,1),x(0,1,1,1); while(y) { if(y&1) a*=x; x*=x; y>>=1; } return (a[0][1]+a[1][1])%MOD; } int main() { int T,i;ll x; f[0]=1; for(i=1;i<=63;i++) f[i]=f[i-1]+g[i-1],g[i]=f[i-1]; for(cin>>T;T;T--) { scanf("%lld",&x); printf("%lld\n", Digital_DP(x+1) ); printf("%lld\n", Matrix_Mutiplication(x) ); } }
BZOJ 3329 Xorequ 数位DP+矩阵乘法
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。