上海预选赛-hdu 4027 Can you answer these queries?

最新推荐文章于 2022-02-25 19:38:17 发布

acmj1991

最新推荐文章于 2022-02-25 19:38:17 发布

阅读量513

点赞数

分类专栏：线段树文章标签： tree query build struct 优化 c

本文链接：https://blog.csdn.net/acmj1991/article/details/6791072

版权

线段树专栏收录该内容

6 篇文章 0 订阅

订阅专栏

原文：http://blog.acmj1991.com/?p=732

题意：给你n个小于long long的数，后面再给你m个操作：a,b,c当a为0时表示对[b,c]区间所有的数进行开方操作。当a为1是表示对[b,c]区间进行求和操作

思路：由于开方是对每一个叶子节点进行操作所以直接对叶子节点操作会超时，这里有一个优化。Long long型数就多开方7次就变成1，所以经过大量开方后大部分的数都会变成1。所以当区间所有值为1时，就不需要进行更新和求和了

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<iostream>
using namespace std;

#define maxN 100010
#define LL(x) (x<<1)
#define RR(x) ((x<<1)+1)

struct node{
    int l,r;
    __int64 key,sum;
    int mi(){return (l+r)/2;}
}po[4*maxN];
__int64 key[maxN];

__int64 build_tree(int l,int r,int st)
{
    po[st].l=l,po[st].r=r,po[st].key=-1;
    if(l==r)
        return po[st].sum=key[l];
    int mid=po[st].mi();
    if(r<=mid)
        return po[st].sum=build_tree(l,r,LL(st));
    else if(l>mid)
        return po[st].sum=build_tree(l,r,RR(st));
    else
        return po[st].sum=build_tree(l,mid,LL(st))+build_tree(mid+1,r,RR(st));
}
__int64 update(int l,int r,int st)
{
    __int64 m;
    if(po[st].key==1)
        return 0;
    if(po[st].l==po[st].r){
        m=po[st].sum-(int)sqrt((double)po[st].sum);
        if(po[st].sum-m==1)po[st].key=1;
    }
    else
    {
        int mid=po[st].mi();
        if(r<=mid)
            m=update(l,r,LL(st));
        else if(l>mid)
            m=update(l,r,RR(st));
        else
            m=update(l,mid,LL(st))+update(mid+1,r,RR(st));
        if(po[LL(st)].key==1&&po[RR(st)].key==1)
            po[st].key=1;
    }
    po[st].sum-=m;
    return m;
}
__int64 query(int l,int r,int st)
{
    if(po[st].key==1)
        return r-l+1;
    if(po[st].l==l&&po[st].r==r)
        return po[st].sum;
    int mid=po[st].mi();
    if(r<=mid)
        return query(l,r,LL(st));
    else if(l>mid)
        return query(l,r,RR(st));
    else
        return query(l,mid,LL(st))+query(mid+1,r,RR(st));
}

int main()
{
    int t=1,n,m;
    while(~scanf("%d",&n)){
        printf("Case #%d:\n",t++);
        for(int i=1;i<=n;i++)
            scanf("%I64d",&key[i]);
        build_tree(1,n,1);
        scanf("%d",&m);
        while(m--){
            int l,r,f;
            scanf("%d%d%d",&f,&l,&r);
            if(l>r){
                int temp=r;
                r=l,l=temp;
            }
            if(!f)
                update(l,r,1);
            else
                printf("%I64d\n",query(l,r,1));
        }
        printf("\n");
    }
}