hdu5921Binary Indexed Tree

最新推荐文章于 2021-11-15 15:45:56 发布

Fsss_7

最新推荐文章于 2021-11-15 15:45:56 发布

阅读量1.2k

点赞数 2

分类专栏： online judge Hdu 数学----------------------------

本文链接：https://blog.csdn.net/Fsss_7/article/details/52750340

版权

online judge Hdu 同时被 2 个专栏收录

129 篇文章 0 订阅

订阅专栏

数学----------------------------

92 篇文章 0 订阅

订阅专栏

链接：http://acm.hdu.edu.cn/showproblem.php?pid=5921

题意：给出多个n，求 $\sum_{i=1}^{n}\sum_{j=0}^{i-1}g(i,j)$ ， $g(i,j)$ 的意义是 $i$ 和 $j$ 在二进制意义下去掉公共前缀后的1的个数之和，例如11={1011}，13={1101}，去掉公共前缀{1100}后剩下{0011}和{0001}那么 $g(11,13)=3$ 。

分析：令 $f(i)$ 表示i的二进制下1的个数，那么 $ans=\sum_{i=1}^{n}\sum_{j=0}^{i-1}g(i,j)=\frac{1}{2}\sum_{i=0}^{n}\sum_{j=0}^{n}[g(i)+g(j)-2g(lcp(i,j))]=\sum_{i=0}^{n}\sum_{j=0}^{n}g(j)-\sum_{i=0}^{n}\sum_{j=0}^{n}g(lcp(i,j))$ 。那么这样我们就只需要求这两部分就行了，第一部分： $\sum_{i=0}^{n}g(i)$ ，第二部分 $\sum_{i=0}^{n}\sum_{j=0}^{n}g(lcp(i,j))$ ，这两部分我们都只需要在n的二进制意义下分别考虑每一位的贡献就行了。 $O(64*n)$

代码：

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<vector>
#include<string>
#include<stdio.h>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=400010;
const int M=50010;
const int mod=1000000007;
const int MOD1=1000000007;
const int MOD2=1000000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=1000000007;
const int INF=1000000010;
const ll MAX=1ll<<55;
const double eps=1e-5;
const double inf=~0u>>1;
const double pi=acos(-1.0);
typedef long double db;
typedef unsigned int uint;
typedef unsigned long long ull;
int a[100];
ll b[100],c[100],e[100];
ll get(int len) {
    ll ret=0;
    for (int i=0;i<len;i++) ret+=a[i];
    for (int i=len-1;i>=0;i--)
    if (a[i]) {
        if (i) (ret+=c[i-1])%MOD;
        (ret+=b[i+1]*e[i]%MOD)%=MOD;
    } else (ret+=b[i+1]*e[i]%MOD)%=MOD;
    return ret;
}
ll getlcp(int len) {
    ll ret=0;
    for (int i=len-1;i>=0;i--)
    if (a[i]) {
        if (i) (ret+=(c[i-1]+1)*(c[i-1]+1)%MOD)%=MOD;
        else (ret+=1ll)%=MOD;
        (ret+=b[i+1]*e[i]%MOD*e[i]%MOD)%=MOD;
    } else (ret+=b[i+1]*e[i]%MOD*e[i]%MOD)%=MOD;
    return ret;
}
int main()
{
    int i,ca,T,len;
    ll n,m,ans;
    scanf("%d", &T);
    for (i=0;i<63;i++) e[i]=(1ll<<i)%MOD;
    for (ca=1;ca<=T;ca++) {
        scanf("%I64d", &n);m=n;
        for (len=0;m;m>>=1) a[len++]=m%2;
        b[len]=0;c[0]=a[0];
        for (i=1;i<len;i++) c[i]=(c[i-1]+(a[i]*1ll<<i))%MOD;
        for (i=len-1;i>=0;i--) b[i]=((b[i+1]<<1)+a[i])%MOD;
        ans=((n+1)%MOD*get(len))%MOD;
        printf("Case #%d: %I64d\n", ca, ((ans-getlcp(len))%MOD+MOD)%MOD);
    }
    return 0;
}