题解 P3954 【成绩】

题面

这题显然可以用到线段树区间求和

上代码:

#include<bits/stdc++.h>

using namespace std;

double a,b,c;
double num[1000010];

//定义线段树节点

struct SegmentTree    
{
    int l,r;
    double sum;
}t[4000010];//开4倍空间

void BuildTree(int,int,int);

double Query(int,int,int,int,int);

int main()
{
    scanf("%lf%lf%lf",&a,&b,&c);
    num[1]=a*0.2;
    num[2]=b*0.3;
    num[3]=c*0.5;
    BuildTree(1,1,3);//建树
    printf("%.0lf\n",Query(1,1,3,1,3));//求和
return 0;
}

void BuildTree(int id,int l,int r)
{
    int mid=(l+r)>>1;
    t[id].l=l;
    t[id].r=r;
    if(l==r)//最后的叶节点,记下该节点值后回溯
    {
        t[id].sum=num[l];
        return;
    }
    BuildTree(id*2,l,mid);//建左子树,根据二叉树的性质,左儿子的节点编号为父亲*2
    BuildTree(id*2+1,mid+1,r);//右子树
    t[id].sum=t[id*2].sum+t[id*2+1].sum;//上传区间和
}

double Query(int id,int l,int r,int x,int y)
{
//[l,r]为当前区间,[x,y]为要查询的区间
    int mid=(l+r)>>1;
    double sum=0;
    if(x<=l&&y>=r)return t[id].sum;//如果当前的区间在查询的区间内,直接返回区间和
    if(x<=mid)sum+=Query(id*2,l,mid,x,y);//如果查询的区间蔓延当前区间的左边,查询左子树
    if(y>mid)sum+=Query(id*2+1,mid+1,r,x,y);//同上
    return sum;
}

猜你喜欢

转载自www.cnblogs.com/Hdgs3-blog/p/10335710.html