First One HDU - 5358(尺取)

最新推荐文章于 2020-09-08 12:54:42 发布

LBJHan

最新推荐文章于 2020-09-08 12:54:42 发布

阅读量638

点赞数

分类专栏：尺取

本文链接：https://blog.csdn.net/Sirius_han/article/details/81109006

版权

尺取专栏收录该内容

6 篇文章 0 订阅

订阅专栏

First One

题目链接：HDU - 5358

题意：数列a中S[i, j]表示 $\sum_{i}^{j}ai$ ，求出下面式子的值：

$\sum_{i=1}^{n}\sum_{j=1}^{n}(((\log_2 S(i, j))+1) * (i+j))$ ,其中的 $((\log _2 S(i, j))+1)$ （二式）是向下取整；

根据题目数据，S(i, j)最大不超过2^34，所以二式的范围是[1, 34]令k=二式；

我们可以枚举k的值，尺取法选择区间[l, r]满足k的;

固定i，找j满足的区间[l, r],则在区间中的(i+j)=i*(r-l+1)+(r-l+1)*(r+l)/2;最后再乘k；

#include <bits/stdc++.h>
using namespace std;
long long sum[100010];
int main(){
	int T;
	scanf("%d", &T);
	while(T--){
		sum[0]=0;
		int n;
		long long x;
		scanf("%d", &n);
		for(int i=1; i<=n; i++){
			scanf("%lld", &x);
			sum[i]=sum[i-1]+x;
		}
		long long ans=0;
		for(int k=1; k<=34; k++){
			long long L=1LL << (k-1), R=(1LL << k) - 1;
			if(k==1) L=0;
			long long l=1, r=0;
			for(int i=1; i<=n; i++){
				l=max((long long)i, l);
				while(sum[l]-sum[i-1]<L&&l<=n) l++;
				r=max(r, l-1);
				while(sum[r+1]-sum[i-1]>=L&&sum[r+1]-sum[i-1]<=R&&r+1<=n) r++;
				if(l>r) continue;
				ans+=(i*(r-l+1)+(r+l)*(r-l+1)/2)*k;
			}
		}
		printf("%lld\n", ans);
	}
	return 0;
}