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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u013912596/article/details/26851745

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




没有更多推荐了,返回首页