【Nodecoder】 暑期多校day4 Chiaki Sequence Reloaded （二进制树）

题目大意

求$f(n)$，共有$10^5$组询问：

$$a_n=\begin{cases}1,&\text{$n=1$}\\\\a_ {\lfloor \frac{n}{2} \rfloor} + (-1)^{\frac{n*(n-1)}{2}},&\text{$n \geq 2$}\end{cases} \quad \quad f(n)=\sum_{i=1}^{n} \vert a_i\vert$$

<script type="math/tex; mode=display" id="MathJax-Element-6168"></script>

链接：https://www.nowcoder.com/acm/contest/142/C

---

解题思路

比赛的时候没有想通预处理的思路，之后多亏dalao的博客才开窍：
http://www.cnblogs.com/zhouzhendong/p/NowCoder-2018-Summer-Round4-C.html

题设中的数据范围暗示该题的时间复杂度Tlogn，考虑用二进制的相关知识解决该问题。

由于$a_ {\lfloor \frac{n}{2} \rfloor}=a_{n<<1}$，通过观察从 $a_{n<<1}$ 到 $a_n$ 的转移可知，对于一个二级制数，每在最后一位的后面加上一位与最后一位相同的数，所得到的 $a_i$ 值为 $a_{i>>1}+1$，反之则为 $a_{i>>1}-1$。例如，对于二进制数 $1010_2$ 对应的 $a_i$ 值为-3，那么它转移得到的 $10101_2$ 对应的 $a_i$ 值就为 $-4 (-3 -1)$，$10100_2$ 对应的 $a_i$ 值为$-2 (-3 +1)$。

将上述规律对应到01字典树：如果该点是由上一个结点走“1”边转移过来，那么这个结点走“1”边转移时，子节点val=该节点val+1，反之则是子节点val=该节点val-1。

但是询问的是绝对值的前缀和，这就比较难办。考虑预处理出g[i][j]，表示当前结点的贡献为 i ，还应该在01字典树上走 j 步时，所到达的所有结点的贡献之和。这部分预处理可以用组合数来做，枚举接下来走的 j 步中有 k 步使val加1，共有 $C_j^k$ 种走法，则可知$g[i][j]=\sum_{j=1}^{log n} \sum_{k=1}^{j} C_j^k*(i-j+k+k)$。这部分用$log^3n$的复杂度预处理出来。

每次询问统计答案时，首先把所有深度小于$log_2n$的树的贡献加入答案；然后用n在01字典树上由高位向低位转移，不过要注意，每走一条“1”边时，要把左子树的所有贡献也计入答案；最后把目的结点的贡献计入答案。时间复杂度为$log^3n$。

---

代码

#include <bits/stdc++.h>
using namespace std;

const int maxn=int(233), base=70, moder=int(1e9)+7;
inline int add(int a,int b) {return (a+b>=moder)?a+b-moder:a+b;}
inline int mul(int a,int b) {return 1ll*a*b%moder;}

int T;
int c[maxn][maxn], g[maxn][maxn];

void init() {
    register int i,j,k;
    for(i=0;i<maxn;++i) c[i][0]=1;
    for(i=0;i<maxn;++i)
        for(j=1;j<=i;++j)
            c[i][j]=add(c[i-1][j-1],c[i-1][j]);
    for(i=-65;i<=65;++i)
        for(j=0;j<=65;++j)
            for(k=0;k<=j;++k)
                g[i+base][j]=add(g[i+base][j],mul(c[j][k],abs(i-j+2*k)));
    return;
}

long long n;
int sta[maxn], top=0;

void work() {
    scanf("%lld",&n);
    top=0;
    for(;n;n>>=1) sta[++top]=n&1;

    int res=0, cur=0, tmp=0;
    register int i;
    for(i=top-1;i>=1;--i,cur+=tmp) {
        tmp=(sta[i]==sta[i+1])?1:-1;
        if(sta[i]) res=add(res,g[cur-tmp+base][i-1]);
        res=add(res,g[0+base][i-1]);
    }
    printf("%d\n",add(res,abs(cur)));
    return;
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("input.txt","r",stdin);
    freopen("output.txt","w",stdout);
#endif
    init();
    for(scanf("%d",&T);T;T--)
        work();

    return 0;
}