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

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

2014-05-24 18:18 381 查看
题目链接: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]就行了,两种问题都一样,大家可以自己试试。


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