hdu5358 First One

最新推荐文章于 2019-07-05 08:48:08 发布

Kirito_Acmer

最新推荐文章于 2019-07-05 08:48:08 发布

阅读量773

点赞数

CC 4.0 BY-SA版权

分类专栏： hdu 文章标签： hdu 多校

本文链接：https://blog.youkuaiyun.com/Kirito_Acmer/article/details/47377305

hdu 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种针对二维数组的求和问题及其优化算法。该算法通过预处理数组元素和使用前缀和技巧，有效地减少了计算复杂度，并利用35个指针进行区间查找，实现了快速计算指定公式的目标值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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:

\sum i = 1 n \sum j = i n (⌊ log 2 S (i, j) ⌋ + 1) \times (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

(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

这题题意容易懂，就是求和，其中(⌊log2S(i,j)⌋+1)的意思就是S(i,j)化成二进制后的比特位个数，因为S(i,j)不超过10^10，所以比特位不会超过35个。我们可以先初始化b[],
记录比特位为i的所有数中的最后一个数2^i-1，用sum[i]把从1到i的总和记录下来，然后用35个指针pt[i]记录以i为起点的最大下标k满足sum[k]-sum[i-1]<=b[j]。
最后注意要用G++交，C++会超时。。
#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<string>
#include<algorithm>
using namespace std;
#define ll long long
#define maxn 100060
ll b[50],sum[maxn];//b[1]=2^i-1
ll a[maxn];
int pt[44];//指针
void init()
{
	int i,j;
	b[0]=-1;
	b[1]=1;
	for(i=2;i<=35;i++){
		b[i]=(1LL<<i)-1; //也可以是b[i]=((ll)1<<i)-1;,但不加的话会爆int 
	}
}
int main()
{
	int n,m,i,j,T,len;
	ll ans;
	init();
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		sum[0]=0;ans=0;
		for(i=1;i<=n;i++){
			scanf("%lld",&a[i]);
			sum[i]=sum[i-1]+a[i];
		}
		for(i=1;i<=35;i++)pt[i]=0;
		for(i=1;i<=n;i++){
			pt[0]=i-1;
			for(j=1;j<=34;j++){
				while(sum[pt[j]+1]-sum[i-1]<=b[j] && pt[j]<n){//如果a>b，那么pt[a]一定大于等于pt[b] 
					pt[j]++;
				}
				//if(sum[pt[j]]-sum[i-1]>b[j-1] && sum[pt[j]]-sum[i-1]<=b[j] && pt[j]>=i ){ 这一句可以不用写 
					len=(pt[j]-pt[j-1]);
					ans+=(ll)j*len*i;
					ans+=(ll)j*len*(pt[j-1]+1+pt[j])/2;
				//}
			}
		}
		printf("%lld\n",ans);
	}
	return 0;
}