HDU 5358 多校第6场 First One

First One

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 131072/131072 K (Java/Others)

Total Submission(s): 672    Accepted Submission(s): 193


Problem Description

soda has an integer array a1,a2,…,an.
Let S(i,j) be
the sum of ai,ai+1,…,aj.
Now soda wants to know the value below:

∑i=1n∑j=in(⌊log2S(i,j)⌋+1)×(i+j)

Note: In this problem, you can consider log20 as
0.

 

Input

There are multiple test cases. The first line of input contains an integer T,
indicating the number of test cases. For each test case:

The first line contains an integer n (1≤n≤105),
the number of integers in the array.

The next line contains n integers a1,a2,…,an (0≤ai≤105).

 

Output

For each test case, output the value.

 

Sample Input

1
2
1 1

 

Sample Output

12

 

Source

2015 Multi-University Training Contest 6

首先题意不是啥问题，就是给你个数组，让你求那个式子，s表示i到j的和。

比赛的时候想了个带二分的，就是统计log2可能出现的可能，一共有34种，然后求一个前缀和，前缀和是递增的，可以二分，就写了

复杂度没算好，以为33*n*log(n)能过，结果跪了一下午，昨天就这么愉快的跪了

赛后题解上说用俩指针扫一遍，不得不吐槽一下，MD题解都是英文的。

联想到一次bestcoder 有一个题是求一共序列中两个数的和模上一个数最大，我曾经的做法就是排序二分，而正解就是俩指针扫一遍，然而我没有记住

这个题就卡这个log（n）然而我就是过不去，诶，还是思维被限制住了

今天上午搞了个33 * n的

思路：

把数组处理成 前缀和，把log分类，有35种情况，题中吧log(0) 规为0，需要特判，题中给的最大的sum值为10^10,所以有35种情况，

那么就分情况讨论，一层循环，里面就是枚举n。然后找到两个值一个是sum[i-1] + 2^j的位置，还有一个是sum[i-1]+2^(j+1)的位置，这段区间内的log值就全为j，用等差数列求和公式就能瞬间算出来这段区间的值，而且只需要两个标记扫一遍，为何可以只扫一遍呢，sum数组的值是递增的，需要找的值也是递增的，所以下次找的时候只需要在上次寻找的后面继续接上就OK

/****************************************
** 2015 Multi-University Training Contest 6
** 1006 First One
** HDU 5358
** by calamity_coming
**************************************/

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

const int MAX_LONG = 1E5;
ll a[MAX_LONG + 10];
ll sum[MAX_LONG + 10];
ll cf2[40];

void init()
{
ll k = 1;
for(int i=0; i<=34; ++i)
{
cf2[i] = (k<<(i));
}
}
int main()
{
init();
int t;
scanf("%d",&t);
while(t--)
{
int n;
scanf("%d",&n);
sum[0] = 0;
for(int i=1; i<=n; ++i)
{
scanf("%I64d",&a[i]);
sum[i] = sum[i-1] + a[i];//前缀和处理
}
sum[n+1] = 1E17;
ll ans = 0;

//0
for(int i=1; i<=n; ++i)//0比较特殊，要特殊的干
{
int p = i;
while(sum[p]==sum[i-1] && p<=n+1)
{
++p;
}
--p;
if(p>=i)
{
ans += (ll)(i*3+p)*(ll)(p-i+1)/2;
}
}

for(int j=0; j<=33; ++j)//这是每个情况
{
int p1 = 0,p2 = 0;//用两个指针扫一遍
for(int i=1; i<=n; ++i)//当i递增时，新的p1,p2一定在后面
{
ll adc = sum[i-1] + cf2[j];
while(sum[p1]<adc && p1<=n)
{
++p1;
}
adc = sum[i-1] +cf2[j+1];
while(sum[p2]<adc && p2<=n+1)
{
++p2;
}
--p2;
if(p2>=p1 && p2)
{
ans += (j+1)*(ll)(i*2+p1+p2)*(ll)(p2-p1+1)/2;
}
}
}
printf("%I64d\n",ans);
}
return 0;
}