Newcoder 142 C.Chiaki Sequence Reloaded（数位DP）

Description

定义

$$a(n)=\left\{ \begin{array}{ccl} 0&&n=1\\ a(\lfloor\frac{n}{2}\rfloor)+(-1)^{\frac{n(n+1)}{2}}&&n\ge 2 \end{array} \right.$$
输出 $\sum\limits_{i=1}^n|a_i|$

Input

第一行一整数$T$表示用例组数，每组用例输入一整数$n$

$(1\le T\le 10^5,1\le n\le 10^{18})$

Output

输出答案，结果模$10^9+7$

Sample Input

10
1
2
3
4
5
6
7
8
9
10

Sample Output

0
1
2
2
4
4
6
7
8
11

Solution

根据$(-1)^{\frac{n(n+1)}{2}}$的符号可以得到以下等式

$$a(n)=\left\{ \begin{array}{ccl} a(\lfloor\frac{n}{2}\rfloor)+1&&n\%4=0,3\\ a(\lfloor\frac{n}{2}\rfloor)-1&&n\%4=1,2 \end{array} \right.$$
进而可知 $a(n)$的值为其二进制表示相邻位相同的个数减去相邻位不同的个数，用数位 $DP$维护该绝对值的和，以 $dp(pos,num,pre,zero)$表示前 $pos$位已经确定、二进制表示相邻位相同的个数减去相邻位不同的个数为 $num$、上一位为 $pre$时的绝对值之和，其中 $zero$表示当前位之前是否全为 $0$，时间复杂度$O(log^2n)$

Code

#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long ll;
#define mod 1000000007
int add(int x,int y)
{
    x+=y;
    if(x>=mod)x-=mod;
    return x;
}
int T,m=65,a[65],dp[65][130][2][2];
ll n;
int dfs(int pos,int num,int pre,int zero,int fp)
{
    if(pos<0)return abs(num);
    if(!fp&&dp[pos][num+m][pre][zero]!=-1)return dp[pos][num+m][pre][zero];
    int ans=0,fpmax=fp?a[pos]:1,temp;
    for(int i=0;i<=fpmax;i++)
    {
        if(zero)temp=(i==pre?1:-1);
        else temp=0;
        ans=add(ans,dfs(pos-1,num+temp,i,zero|i,fp&&i==fpmax));
    }
    if(!fp)dp[pos][num+m][pre][zero]=ans;
    return ans;
}
int main()
{
    memset(dp,-1,sizeof(dp));
    scanf("%d",&T);
    while(T--)
    {
        scanf("%lld",&n);
        int len=0;
        while(n)a[len++]=n%2,n/=2;
        printf("%d\n",dfs(len-1,0,0,0,1));
    }
    return 0;
}