HDU 6304 Chiaki Sequence Revisited（二分+dp）

Description

序列$a_n$满足$a_1=a_2=1,a_n=a_{n-a_{n-1}}+a_{n-1-a_{n-2}},n\ge 3$，给出一整数$n$，求$\sum\limits_{i=1}^na_i$

Input

第一行一整数$T$表示用例组数，每组用例输入一整数$n(1\le T\le 10^5,1\le n\le 10^{18})$

Output

输出$\sum\limits_{i=1}^na_i$

Sample Input

10
1
2
3
4
5
6
7
8
9
10

Sample Output

1
2
4
6
9
13
17
21
26
32

Solution

可以发现$a$序列有两个规律，首先$a_n$的值较$a_{n-1}$每次增加$0$或$1$，其次，假设$a_n=2^m\cdot k,k$为奇数，那么$a_n$在序列中出现$m+1$次

以此规律，令$num[x]$表示前$x$个自然数在该序列中出现的次数，$sum[x]$表示在该序列中出现的前$x$个自然数的和

对于$1$~$x$中的奇数，在序列中只出现一次，对于$1$~$x$中的偶数，把每个数字除以二则转化为求$num[\lfloor\frac{x}{2}\rfloor]$和$sum[\lfloor\frac{x}{2}\rfloor]$，由于$2k$出现次数比$k$多一次，故有转移：

$num[x]=num[\lfloor\frac{x}{2}\rfloor]+x,sum[x]=2\cdot sum[\lfloor\frac{x}{2}\rfloor]+\frac{x(x+1)}{2}$

二分找到最大的$x$满足$num[x]\le n$，答案即为$sum[x]+(n-num[x])\cdot (x+1)$，时间复杂度$O(Tlog^2n)$

Code

#include<cstdio>
using namespace std;
typedef long long ll;
#define mod 1000000007
#define inv2 500000004
ll count(ll n)
{
    if(n==1)return 2;
    return n+count(n/2);
}
ll Solve(ll n)
{
    if(n==1)return 1;
    return (n%mod*((n+1)%mod)%mod*inv2%mod+2ll*Solve(n/2)%mod)%mod;
}
int main()
{
    int T;
    ll n;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%I64d",&n);
        if(n<=2)printf("%I64d\n",n);
        else
        {
            ll l=1,r=1e18,pos,mid;
            while(l<=r)
            {
                mid=(l+r)/2;
                if(count(mid)<=n)pos=mid,l=mid+1;
                else r=mid-1;
            }   
            ll ans=(Solve(pos)+1)%mod;
            ans=(ans+(n-count(pos))*(pos+1)%mod)%mod;
            printf("%I64d\n",ans);
        }
    }
    return 0;
}