hdu5358

来源：互联网发布：医疗优化站不收录编辑：程序博客网时间：2024/04/29 04:31

题目名称：First One

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

题意：计算原题中的式子

1.思路：先用sum数组记录a[i] 的前缀和，顺便计算 1*( i + j ) 部分，即 i 算了(n-i+1)次再加上 i +(i+1)+...+n 。因为log2S(i,j) 是向下取整，我们可以看成s(i,j)在[2^(k),2^(k+1))之间时， log2（s(i,j)）向下取整的值就是k。假如从 i 开始，每次找出和 i 相差 k 和 k+1 的左右端点，然后把这区间的和的全部值乘以k 就是求log2S(i,j)*(i+j)的值了

代码如下：

#include<cstdio>#include<iostream>#include<cstring>#include<vector>#include<set>#include<queue>#include<cmath>using namespace std;typedef long long ll;const int MAXN=100000+10;ll Scan(){ll res = 0, ch, flag = 0;if((ch = getchar()) == '-')//判断正负flag = 1;else if(ch >= '0' && ch <= '9')//得到完整的数res = ch - '0';while((ch = getchar()) >= '0' && ch <= '9' )res = res * 10 + ch - '0';return flag ? -res : res;}ll a[MAXN],sum[MAXN],power[40];int main(){    ll t,n;    for(int i=0;i<40;i++)    {        power[i]=1LL<<i;    }    t=Scan();    while(t--)    {        n=Scan();        ll ans=0;        sum[0]=0;        for(int i=1;i<=n;i++)        {            a[i]=Scan();            sum[i]=sum[i-1]+a[i];            ans+=1LL*(n+i)*(n-i+1)/2+1LL*i*(n-i+1);//计算式子中的1*(i+j)        }        for(int i=1;i<40;i++)        {            if(power[i]>sum[n])                break;            ll ss=0;            ll L=1,R=0;            for(int j=1;j<=n;j++)            {                while(L<=n&&sum[L]-sum[j-1]<power[i])       //寻找log2的值在i和i+1之间的aj + aj+1 +...ax的最小X为 L和最大X为R                    L++;                while(R+1<=n&&sum[R+1]-sum[j-1]<power[i+1])                    R++;                if(L<=R)                    ss+=(R+L)*(R-L+1)/2+j*(R-L+1);   //计算 (j+L)+(j+L+1)+...+(j+R)            }            ans+=ss*i;              //计算式子中的(i+j)*log2S(i,j) （式子中 i=log2(i,j)）        }        printf("%lld\n",ans);    }    return 0;}

2.思路：先用sum数组保存下i的后缀和，顺便把 1 * ( i + j ) 部分给算了，就是 i 算了(n-i+1)次加上每次从i+(i+1)+...+n。因为log2S(i,j) 是向下取整，我们可以看成s(i,j)在[2^(k),2^(k+1))之间时， log2（s(i,j)）向下取整的值就是k。计算log2S(i,j)*(i+j)，假设k=1,求出在该区间的左端点为 i，右端点为 j，即对于在j后面的数x，总数都加上( i + x ) ；当k=2时，求出 i ，j 因为在k=1时已经加上一次( i + x ) 了，所以只需再加上一次 (本来要*2的)......以此类推。画一下图就可以理解了，比如i -> j1是区间为[2^1,2^2)，i -> j2是区间为[2^2,2^3)，我们在计算i -> j1 时因为是加上j后面的数，也就是说把i -> j2 的那部分也加上了，所以我们计算时只需再算一次。。。

代码如下：

#include<cstdio>#include<iostream>#include<cstring>#include<string>#include<algorithm>using namespace std;typedef long long LL;const int MAXN=100000+10;LL sum[MAXN],power[40];int a[MAXN],n;int main(){    int T;    for(int i=0;i<40;i++)    {        power[i]=1LL<<i;    }    scanf("%d",&T);    for (int ss=1;ss<=T;ss++)    {        scanf("%d",&n);        for(int i=0;i<n;++i)            scanf("%d",&a[i]);        LL ans=0;        sum[n]=0;        for(int i=n-1;i>=0;i--)        {            sum[i]=sum[i+1]+i+1;         //算后缀和            ans+=LL(i+1)*(n-i)+sum[i];   //算(i+j)*1        }        for(int k=1;k<=35;k++)        {            LL s=0;            LL j=0;            for(int i=0;i<n;i++)            {                while(j<n&&s<power[k])                {                    s+=a[j];                    j++;                }                if(s>=power[k])                    ans+=1LL*(i+1)*(n-j+1)+sum[j-1];  //算log2(i,j)*(i+j)                s-=a[i];            }        }        printf("%lld\n",ans);    }    return 0;}

0 0