hdu 4027(線段樹區間開方求和)

本來很早就做了這道題的,但是今天突然發現了.
傳送門:http://acm.hdu.edu.cn/showproblem.php?pid=4027
解法:首先這道題的數據就告訴我們只能用線段樹,但區間操作似乎很蛋疼,我當時想的是用lazy標記記錄這段區間要開方几次,等query時在統一進行單點更新,然後…….我就T了,如果update一次query一次,我的lazy標記便並卵了,所以我們需要想更好方法來解這道題.
每個數的範圍是int,且向下取整,這代表在少量操作後,它就會爲一,所以我們只需要看這段區間的值是否爲r-l+1,如果是,就continue,否則就單點更新,這樣的時間複雜度大約在nlogn左右.問題完美解決

#include <cstdio>
#include <cmath>
#include <iostream>
#include <cstring>
#include <algorithm>
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
using namespace std;
int sum[400005],n,m,x,l,r;
void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int rt,int l,int r)
{
    if (l==r)
    {
        scanf("%d",&sum[rt]);
        return;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}
void update(int rt,int l,int r,int x,int y)
{
    if (x<=l&&y>=r&&l==r) 
    {
         sum[rt]=sqrt(sum[rt]);
         return;
    }
    if (sum[rt]==r-l+1)
    {
        return;
    }
    int mid=(l+r)>>1;
    if (l<=mid) update(lson,x,y);
    if (r>mid) update(rson,x,y);
    pushup(rt);
}
int query(int rt,int l,int r,int x,int y)
{
    int ans=0;
    if(l==x&&r==y) 
    {
        return sum[rt];
    }
    int mid=(l+r)>>1;
    if(y<=mid) ans=query(lson,x,y);
    else if(x>mid) ans=query(rson,x,y);
    else 
    {
        ans=query(lson,x,mid)+query(rson,mid+1,y);
    }
    pushup(rt);
}
void readdata()
{
    scanf("%d",&n);
    build(1,1,n);
    cin>>m;
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&l,&r);
        {
            if (x==0)
            {
                update(1,1,n,l,r);
            }
            if (x==1)
            {
                printf("%d",query(1,1,n,l,r));
            }
        }
    }
}
void o_o()
{
}
void T_T()
{
}
int main()
{
    readdata();
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章