BZOJ 3329: Xorequ 数位dp 矩阵乘法

3329: Xorequ

Time Limit: 1 Sec  Memory Limit: 256 MB
Submit: 1292  Solved: 546
[Submit][Status][Discuss]

Description

Input

第一行一个正整数，表示数据组数据 ，接下来T行
每行一个正整数N

Output

2*T行
第2*i-1行表示第i个数据中问题一的解，

第2*i行表示第i个数据中问题二的解，

Sample Input

1
1

Sample Output

1
2

HINT

x=1与x=2都是原方程的根，注意第一个问题的解
不要mod 10^9+7

1<=N<=10^18 
1<=T<=1000

很显然 x 的二进制位上 没有连续的1

之后我们发现

<=2^n 的满足条件的方案数即为 fib[n+1]

所以第二问就直接矩阵快速幂

第一问就数位dp一下就好

注意下第一问不取模 /泪

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>
using namespace std;

typedef double db;
typedef long long ll;

inline ll read()
{
	ll x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
void print(ll x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=110,mod=int(1e9)+7;

ll fib[N];

ll solve_n(ll n)
{
	if(!n) return 0;
	register int i(0),B;
	ll res(0);
	while((1ll<<i)<=n) i++;
	B=i-1;
	fib[0]=fib[1]=1;
	for(i=2;i<=B+1;++i)
		fib[i]=fib[i-1]+fib[i-2];
	for(i=B;~i;i--)
		if(n&(1ll<<i))
		{
			res+=fib[i+1];
			if(n&(1ll<<i+1)){res--;break;}
		}
	return res;
}

struct matrix
{
	int a[2][2];
	matrix(){memset(a,0,sizeof(a));};
	
	friend matrix operator *(const matrix &x,const matrix &y)
	{
		matrix res;
		for(int i=0;i<2;++i)
			for(int j=0;j<2;++j)
				for(int k=0;k<2;++k)
					(res.a[i][j]+=1ll*x.a[i][k]*y.a[k][j]%mod)%=mod;
		return res;
	}
};

matrix qpow(matrix &x,ll y)
{
	matrix res;
	res.a[0][0]=res.a[1][1]=1;
	while(y)
	{
		if(y&1) res=res*x;
		x=x*x;
		y>>=1;
	}
	return res;
}

int solve_two(ll n)
{
	if(!n) return 1;
	matrix res;
	res.a[0][1]=res.a[1][0]=res.a[1][1]=1;
	res=qpow(res,n);
	return (res.a[1][0]+res.a[1][1])%mod;
}

int main()
{
	register int T=read();
	ll n;
	while(T--)
	{
		n=read();
		print(solve_n(n));putchar('\n');
		print(solve_two(n));putchar('\n');
	}
	return 0;
}