bzoj3329 Xorequ

Xorequ

Time Limit: 1 Sec Memory Limit: 256 MB

Description

Input

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

Output

2T行
第2i-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

我就服这个简单的数位dp。。。。 大概就是洗澡的时候瞎脑补了个算法。。。反正我很菜就是了。。。 矩阵快速幂没清零的我是不是很强？？？ ```c++

include<bits/stdc++.h>

using namespace std;
const int mod = 1e9 + 7;
struct lpl{
int n, m;
long long a[3][3];
}lin;
long long n;
long long f[66][2], pw[66];
int num[66];

lpl operator * (const lpl &A, const lpl &B)
{
lpl ret; ret.n = A.n; ret.m = B.m;
memset(ret.a, 0, sizeof(ret.a));
for(int i = 1; i <= ret.n; ++i)
for(int j = 1; j <= ret.m; ++j)
for(int k = 1; k <= A.m; ++k)
ret.a[i][j] = (ret.a[i][j] + A.a[i][k] * B.a[k][j] % mod) % mod;
return ret;
}

inline void prepare()
{
f[1][0] = 1; f[1][1] = 1; pw[0] = 1; pw[1] = 2;
for(int i = 2; i <= 63; ++i){
pw[i] = (pw[i - 1] << 1);
f[i][0] = f[i - 1][0] + f[i - 1][1];
f[i][1] = f[i - 1][0];
}
}

inline long long workk1(long long t)
{
long long ret = 0;
int len = 62;
while(pw[len] > t) len--;
for(int i = 1; i <= len + 1; ++i){
num[i] = (t & 1); t >>= 1;
}
num[len + 2] = 0;
for(int i = len + 1; i >= 1; --i){
if(num[i]){
ret += f[i][0];
if(num[i + 1]){
ret--; break;
}
}
}
return ret;
}

inline long long workk2(long long t)
{
lpl ret;
lin.a[1][1] = lin.a[1][2] = lin.a[2][1] = 1; lin.a[2][2] = 0; lin.n = lin.m = 2;
ret.a[1][1] = ret.a[2][2] = 1; ret.a[1][2] = ret.a[2][1] = 0; ret.n = ret.m = 2;
while(t){
if(t & 1)
ret = ret * lin;
t >>= 1; lin = lin * lin;
}
lpl ans; ans.n = 2; ans.m = 1; ans.a[1][1] = ans.a[2][1] = 1; ans.a[1][2] = ans.a[2][2] = 0;
ans = ret * ans;
return ans.a[1][1];
}

int main()
{
prepare();
int T; scanf("%d", &T);
while(T--){
scanf("%lld", &n);
printf("%lld\n%lld\n", workk1(n), workk2(n));
}
return 0;
}