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
Source
By Wcmg
打表可得符合条件的x的二进制位没有相邻的1。
打表可得第二问答案是fib
第一问就是数位DP了…
设dp[i][0/1]表示二进制长度为i最高位是0/1的个数…
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
const int SZ = 100;
const int INF = 1000000010;
const LL mod = 1000000007;
LL dp[SZ][2],base[SZ];
void init()
{
base[1] = 1;
for(int i = 2;i <= 63;i ++)
base[i] = base[i - 1] << 1;
dp[1][0] = 1; dp[1][1] = 1;
for(int i = 2;i <= 63;i ++)
{
dp[i][0] = dp[i - 1][0] + dp[i - 1][1];
dp[i][1] = dp[i - 1][0];
}
}
LL work1(LL n)
{
LL ans = 0;
int len = 63;
while(n < base[len]) len --;
for(int i = 1;i < len;i ++)
ans += dp[i][1];
int cur = n / base[len];
int pre = cur;
n %= base[len];
for(int i = len - 1;i >= 1;i --)
{
cur = n / base[i];
if(cur == 1)
ans += dp[i][0];
if(cur && pre) break;
pre = cur;
n %= base[i];
}
return ans;
}
struct matrix{
int n,m;
LL num[10][10];
matrix(int a,int b) : n(a),m(b) {memset(num,0,sizeof(num));}
};
matrix operator *(const matrix &a,const matrix &b)
{
matrix ans(a.n,b.m);
for(int i = 1;i <= ans.n;i ++)
for(int j = 1;j <= ans.m;j ++)
for(int k = 1;k <= a.m;k ++)
ans.num[i][j] = (ans.num[i][j] + (LL)a.num[i][k] * b.num[k][j] % mod) % mod;
return ans;
}
matrix ksm(matrix a,LL b)
{
matrix ans(a.n,a.m);
for(int i = 1;i <= ans.n;i ++)
ans.num[i][i] = 1;
while(b)
{
if(b & 1) ans = ans * a;
a = a * a;
b >>= 1;
}
return ans;
}
LL work2(LL n) //fib
{
matrix fib(2,2),ans(1,2);
fib.num[1][1] = 1; fib.num[1][2] = 1;
fib.num[2][1] = 1; fib.num[2][2] = 0;
ans.num[1][1] = 1; ans.num[1][2] = 1;
return (ans * ksm(fib,n)).num[1][1];
}
void print(LL x)
{
for(int i = 10;i >= 0;i --)
printf("%d",x >> i & 1);
printf(" ");
}
int main()
{
init();
int T;
scanf("%d",&T);
while(T --)
{
LL n;
scanf("%lld",&n);
printf("%lld\n%lld\n",work1(n + 1),work2(n));
}
return 0;
}