HDU 6304 Chiaki Sequence Revisited（找规律）

题目链接：Chiaki Sequence Revisited

题意

定义一个序列

$$a_n=\begin{cases} 1&n=1,2\\ a_{n-a_{n-1}}+a_{n-1-a_{n-2}}&n\geq3 \end{cases}$$ 求 $(\sum_{i=1}^na_i)\%(10^9+7)$ 的值。

输入

第一行为一个整数 $T~(1\leq T\leq10^5)$，表示有 $T$ 组数据，每组数据包含一个整数 $n~(1\leq n\leq10^{18})$。

输出

每组数据输出 $(\sum_{i=1}^na_i)\%(10^9+7)$。

样例

输入
10 1 2 3 4 5 6 7 8 9 10
输出
1 2 4 6 9 13 17 21 26 32

题解

先打出前 $40$ 个 $a_n$ 的表：

$$1~1~2~2~3~4~4~4~5~6~6~7~8~8~8~8~9~10~10~11~12~12~12~13~14~14~15~16~16~16~16~16~17~18~18~19~20~20~20~21$$ 首先对于任意两个相邻且不相等的数字，它们的差都为 $1$。我将除了第 $1$ 个数字和第 $40$ 个数字的连续相同的数字统计个数以及个数的前缀和打表：

${\bf a}$	$1$	$2$	$3$	$4$	$5$	$6$	$7$	$8$	$9$	$10$	$11$	$12$	$13$	$14$	$15$	$16$	$17$	$18$	$19$	$20$
${\bf cnt}$	$1$	$2$	$1$	$3$	$1$	$2$	$1$	$4$	$1$	$2$	$1$	$3$	$1$	$2$	$1$	$5$	$1$	$2$	$1$	$3$
${\bf sum}$	$1$	$3$	$4$	$7$	$8$	$10$	$11$	$15$	$16$	$18$	$19$	$22$	$23$	$25$	$26$	$31$	$32$	$34$	$35$	$38$
${\bf mark}$	$\Delta$	$\Delta$		$\Delta$				$\Delta$								$\Delta$

把 $cnt$ 中每个数字第一次出现的位置打上标记，可以发现这是一个 $2^k$ 的循环，对于每一个数字 $n$，我们首先可以找到小于等于它的最大的 $2^k-1$ 的 $sum$ 值，$2^k-1$ 个数字其对应的 $a$ 的值只增加了 $2^{k-1}$ 个数字，将 $a_n$ 加上 $2^{k-1}$，$n$ 减去 $2^k-1$，不断查找下去，就可以在 $O(\log n\log(\log n))$ 次内确定 $a_n$ 的值。而从 $a_2$ 到 $a_n$ 的所有数字的前缀和，通过观察可以发现，如果 $a_n$ 是某段连续相同数字的最后一项，可以先计算 $2^0$ 的等差数列求和，再计算 $2^1$ 等差数列求和……每次都计算一个首项为 $2^k$，末项为 $\lfloor\frac{n}{2^k}\rfloor\times2^k$，项数为 $\lfloor\frac{n}{2^k}\rfloor$，的一个等差数列，直到 $2^k>a_n$，停止计算，就能得到 $a_2$ 到 $a_n$ 的所有数字的前缀和；而如果 $a_n$ 不是某段连续相同数字的最后一项，就要先计算出 $a_n$ 的前一个连续的相同数字的最后一项，按照上述方法计算后，加上 $a_n$ 这段连续的数字中，$a_n$ 前面的数字个数。由于前面的规律是在去掉第 $1$ 个数字找到的，所以计算最后答案的时候要 $+1$。
最后注意答案会乘爆 long long，所以在乘法之前应该先取模一次。

过题代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <cfloat>
#include <cstring>
#include <string>
#include <vector>
#include <list>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <algorithm>
using namespace std;

#define LL long long
const LL MOD = 1000000007;
const LL inv = 500000004;
int T;
LL n;
LL two[64], two_1[64];
LL *it;

LL get_a(LL n) {
    if(n == 1) {
        return 0;
    }
    --n;
    LL ret = 0;
    while(n != 0) {
        it = upper_bound(two_1, two_1 + 62, n);
        --it;
        ret += two[it - two_1 - 1];
        n -= *it;
    }
    return ret;
}

LL solve(LL d, LL n) {
    return (((d + (n / d * d % MOD)) % MOD) * ((n / d) % MOD) % MOD) * inv % MOD;
}

int main() {
    #ifdef Dmaxiya
    freopen("test.txt", "r", stdin);
    #endif // Dmaxiya
    ios::sync_with_stdio(false);

    two[0] = 1;
    for(int i = 1; i < 62; ++i) {
        two[i] = two[i - 1] * 2;
    }
    for(int i = 0; i < 62; ++i) {
        two_1[i] = two[i] - 1;
    }
    cin >> T;
    while(T--) {
        cin >> n;
        if(n == 1) {
            cout << 1 << endl;
            continue;
        }
        LL a_n = get_a(n);
        LL n_1 = n;
        LL a_n_1;
        do {
            --n_1;
            a_n_1 = get_a(n_1);
        } while(a_n_1 == a_n);
        LL ans = 0;
        for(int i = 0; two[i] <= a_n_1; ++i) {
            ans += solve(two[i], a_n_1);
            ans %= MOD;
        }
        ans = (ans + (a_n % MOD) * (n - n_1) % MOD) % MOD;
        ans = (ans + 1) % MOD;
        cout << ans << endl;
    }

    return 0;
}