【线段树】HDU - 4027 Can you answer these queries?

【题目】http://acm.hdu.edu.cn/showproblem.php?pid=4027

【题意】

【思路】

【代码】

#include<bits/stdc++.h>
using namespace std;
long long n;
long long arr[100005];
long long Tree[800005];


void create(long long root,long long head,long long tail)
{
    if(head==tail)Tree[root]=arr[head];
    else
    {
        long long mid=(head+tail)/2;
        create(root*2,head,mid);
        create(root*2+1,mid+1,tail);
        Tree[root]=Tree[root*2]+Tree[root*2+1];
    }
}


long long quest(long long root,long long ql,
                long long qr,long long head,long long tail)
{
    if(ql>tail||qr<head)return 0;
    if(ql<=head&&qr>=tail)return Tree[root];
    long long mid=(head+tail)/2;
    return quest(root*2,ql,qr,head,mid)+quest(root*2+1,ql,qr,mid+1,tail);
}

void qu()
{
    long long l,r;
    scanf("%I64d%I64d",&l,&r);
    if(l>r)swap(l,r);
    printf("%I64d\n",quest(1,l,r,1,n));
}

void flash(long long root,long long cl,long long cr,long long head,long long tail)
{
    if(tail-head+1==Tree[root])return;
    if(cl>tail||cr<head)return;
    if(head==tail&&(head>=cl&&tail<=cr))
        Tree[root]=sqrt(Tree[root]);
    else
    {
        long long mid=(head+tail)/2;
        flash(root*2,cl,cr,head,mid);
        flash(root*2+1,cl,cr,mid+1,tail);
        Tree[root]=Tree[root*2]+Tree[root*2+1];
    }
}


void update()
{
    long long l,r;
    scanf("%I64d%I64d",&l,&r);
    if(l>r)swap(l,r);
    flash(1,l,r,1,n);
}
int main()
{
    int t=1;
    while(scanf("%I64d",&n)!=EOF)
    {
        memset(Tree,0,sizeof(Tree));
        for(long long i=1;i<=n;i++)
        {
            scanf("%I64d",&arr[i]);
        }
        create(1,1,n);
        printf("Case #%d:\n",t++);
      //  for(long long i=0;i<100;i++)printf("%I64d ",Tree[i]);
        long long m;
        scanf("%I64d",&m);
        for(long long i=0;i<m;i++)
        {
            long long d0;
            scanf("%I64d",&d0);
            if(d0==1)qu();
            else update();
        }
        printf("\n");
    }
}

猜你喜欢

转载自blog.csdn.net/qq_32259423/article/details/81320510