3038: 上帝造题的七分钟2 线段树+剪枝

来源：互联网发布：淘宝大连博哲贸易假货编辑：程序博客网时间：2024/04/25 09:52

一维的区间修改+区间查询，显然可以用线段树来做。
裸的线段树会超时，因为每次开方都要递归到底……（开始将开方的和写成了和的开方QAQ）
剪枝就是对于一个数如果是1或0，那么之后无需再对其进行操作，因为开方不会影响这个数值。

#include<bits/stdc++.h>#define N 400005#define ll long long using namespace std;int n,m;int l[N],r[N];ll sum[N],a[N>>2];bool flag[N];inline ll read(){    ll a=0,f=1; char c=getchar();    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}    return a*f;}inline void pushup(int k){    sum[k]=sum[k<<1]+sum[k<<1|1];    flag[k]=flag[k<<1]&flag[k<<1|1];}void build(int k,int x,int y){    l[k]=x; r[k]=y;    if (l[k]==r[k])     {        sum[k]=a[l[k]];        if (sum[k]==0||sum[k]==1) flag[k]=1;         return;    }    int mid=l[k]+r[k]>>1;    build(k<<1,x,mid); build(k<<1|1,mid+1,y);    pushup(k);}void change(int k,int x,int y){    if (flag[k]) return;    if (l[k]==r[k])    {        sum[k]=(ll)sqrt(sum[k]);        if (sum[k]==0||sum[k]==1) flag[k]=1;        return;    }    int mid=l[k]+r[k]>>1;    if (y<=mid) change(k<<1,x,y);    else if (x>mid) change(k<<1|1,x,y);    else change(k<<1,x,mid),change(k<<1|1,mid+1,y);    pushup(k);}ll query(int k,int x,int y){    if (x<=l[k]&&r[k]<=y) return sum[k];    int mid=l[k]+r[k]>>1;    if (y<=mid) return query(k<<1,x,y);    else if (x>mid) return query(k<<1|1,x,y);    else return query(k<<1,x,mid)+query(k<<1|1,mid+1,y);}int main(){    n=read();     for (int i=1;i<=n;i++) a[i]=read();    build(1,1,n);    m=read();    for (int i=1;i<=m;i++)    {        int opt=read(),l=read(),r=read();        if (l>r) swap(l,r);        if (opt==0) change(1,l,r);        else printf("%lld\n",query(1,l,r));    }    return 0;}

0 0