您的位置:首页 > 理论基础 > 数据结构算法

Codeforces Round #248 (Div. 2) B称号 【数据结构:树状数组】

2015-08-23 20:07 666 查看
主题链接:http://codeforces.com/contest/433/problem/B

题目大意:给n(1 ≤ n ≤ 105)个数据(1 ≤ vi ≤ 109),当中有m(1 ≤ m ≤ 105)个问题,分两种。第一种:给出l,r,让你求出v[l],v[r]之间的全部数据和;另外一种:先将原数据升序排序得到vv数组,给出l,r,让你求出vv[l],vv[r]之间的全部数据和;

此题假设用暴力求解,因为数据太大,会TLE,所以利用树状数组,高速求解区间和的问题。

假设不懂树状数组,能够去看看关于数据结构的知识

代码:

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;

int v[100011];
int vv[100011];
long long Cv[100011];// 每一个C数组代表v[i-lowbit(i)+1]到v[i]之间的和
long long Cvv[100011];

int lowbit(int x)
{
return x&(x^(x-1));
}
long long Sumv(int i)   //求出数组v[1]到v[i]的和
{
long long result=0;
while (i>=1)
{
result+=Cv[i];
i-=lowbit(i);
}
return result;
}
long long Sumvv(int i)   //求出数组vv[1]到vv[i]的和
{
long long result=0;
while (i>=1)
{
result+=Cvv[i];
i-=lowbit(i);
}
return result;
}
int main()
{
int n,m,t,l,r;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&v[i]);
vv[i]=v[i];
}
sort(vv,vv+n+1);
for(int i=1;i<=n;i++)
{
for(int j=i-lowbit(i)+1;j<=i;j++)
Cv[i]+=v[j];
for(int j=i-lowbit(i)+1;j<=i;j++)
Cvv[i]+=vv[j];
}
scanf("%d",&m);
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&t,&l,&r);
if(t==1)
{
printf("%I64d\n",Sumv(r)-Sumv(l-1));
}
else
{
printf("%I64d\n",Sumvv(r)-Sumvv(l-1));
}
}
return 0;
}


此题也能够不用树状数组,能够用s[i]数组代表v[1]到v[i]的全部数据和,要求v[l]到v[r]之间的数据和,仅仅须要用s[r]-s[l-1]即可了,两种问题都一样,大家能够自己试试。


内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: