标签:
as small as possible!
. Output a blank line after every test case.2 5 3 6 2 2 4 2 1 4 0 2 2 7 7 2 0 1 1 1
Case #1: 6 4 Case #2: 0 0
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 100050
using namespace std;
int sorted[N]; //排序完的数组
int toleft[30][N]; //toleft[i][j]表示第i层从1到k有多少个数分入左边
int tree[30][N]; //表示每层每个位置的值
int n;//输入的数的个数
long long sum[N];//保存包括当前位置及之前的所有元素之和
long long lsum[30][N];//在deep层 第i个元素在的区间中 在i前面被划分到左子树的元素之和
long long tsum;//表示小于中位数ave部分的和
void building(int l,int r,int dep)
{
if(l==r) return;
int mid = (l+r)>>1;
int temp = sorted[mid];
int i,sum_same=mid-l+1;//表示等于中间值而且被分入左边的个数
for(i=l; i<=r; i++)
if(tree[dep][i]<temp)
sum_same--;
int leftpos = l;
int rightpos = mid+1;
for(i=l; i<=r; i++)
{
if(tree[dep][i]<temp)//比中间的数小,分入左边
{
tree[dep+1][leftpos++]=tree[dep][i];
lsum[dep][i] = lsum[dep][i-1] + tree[dep][i];//记录被划分到左边的和是多少
}
else if(tree[dep][i]==temp&&sum_same>0)//等于中间的数值,分入左边,直到sum==0后分到右边
{
tree[dep+1][leftpos++]=tree[dep][i];
lsum[dep][i] = lsum[dep][i-1] + tree[dep][i];
sum_same--;
}
else //右边
{
tree[dep+1][rightpos++]=tree[dep][i];
lsum[dep][i] = lsum[dep][i-1];
}
toleft[dep][i] = toleft[dep][l-1] + leftpos - l; //从1到i放左边的个数
}
building(l,mid,dep+1);
building(mid+1,r,dep+1);
}
//查询区间第k大的数,[L,R]是大区间,[l,r]是要查询的小区间
int query(int L,int R,int l,int r,int dep,int k)
{
if(l==r) return tree[dep][l];
int mid = (L+R)>>1;
int cnt = toleft[dep][r] - toleft[dep][l-1]; //[l,r]中位于左边的个数
if(cnt>=k)
{
int newl = L + toleft[dep][l-1] - toleft[dep][L-1]; //L+要查询的区间前被放在左边的个数
int newr = newl + cnt - 1; //左端点加上查询区间会被放在左边的个数
return query(L,mid,newl,newr,dep+1,k);
}
else
{
int newr = r + (toleft[dep][R] - toleft[dep][r]);
int newl = newr - (r-l-cnt);
tsum+=lsum[dep][r]-lsum[dep][l-1];//区间
}
}
int main()
{
int t,cas=1;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
int i;
for(i=1; i<=n; i++)
{
scanf("%lld",&sorted[i]);
sum[i] = sum[i-1]+sorted[i];//保存包括当前位置及之前的所有元素之和
tree[0][i]=sorted[i] ;
}
sort(sorted+1,sorted+1+n);
building(1,n,0);
int l,r;
int m;//查询次数
scanf("%d",&m);
printf("Case #%d:\n",cas++);
while(m--)
{
scanf("%d%d",&l,&r);
l++,r++;
tsum=0;//表示小于中位数ave部分的和
int k=(r-l)/2+1;//k是区间ab的中位数
long long ave= query(1,n,l,r,0,k);//ave是中位数的值,r-l+1-k是左边数的个数
long long ans=sum[r]-sum[l-1]-ave*(r-l+1-k)-tsum-ave;//中位数右边和,大于ave的部分
ans+=(k-1)*ave-tsum;//中位数左边和,小于ave的部分
printf("%lld\n",ans);
}
printf("\n");
}
return 0;
}
HDU 3473-Minimum Sum(划分树-求区间sigma最小值)
标签:
原文地址:http://blog.csdn.net/mikasa3/article/details/51345004