bzoj3211: 花神游历各国

链接

　　http://www.lydsy.com/JudgeOnline/problem.php?id=3211

题解

　　容易发现，$10^9$开6平方根就变成1了，所以可以每个线段树开一个$w$数组，$w_i$表示这个区间整体开$i$次根号的和是多少，$tag$表示这个区间被整体开根了多少次。
　　更新的时候直接枚举左右儿子的$w_i$然后加起来就好了，下放标记的时候让$w_i$整体滚动就行了。
　　这道题加了数据之后卡常数，它本意可能是想卡分块…但是线段树常数大的话也可能卡掉（我就被卡掉了）。在下放标记和用儿子更新自己的时候，如果发现$w_1=w_2$，那说明所有数都已经开根到极限了，这时候可以直接退出。
　　然后就过了。

代码

//线段树
#include <cstdio>
#include <algorithm>
#include <cmath>
#define maxn 100010
#define ll long long
using namespace std;
ll N, M, ndtot, a[maxn], building=1;
struct segtree
{
    ll l, r, w[7], tag;
    segtree *ch[2], *f;
}pool[maxn*4], *root;
inline ll read(ll x=0)
{
    char c=getchar();
    while(c<48 or c>57)c=getchar();
    while(c>=48 and c<=57)x=(x<<1)+(x<<3)+c-48,c=getchar();
    return x;
}
inline void pushdown(segtree *p)
{
    if(p->tag)
    {
        if(p->w[1]==p->w[2]){p->tag=0;return;}
        ll i;
        for(i=1;i<=6;i++)p->w[i]=p->w[min(6ll,i+p->tag)];
        if(p->ch[0])p->ch[0]->tag+=p->tag, p->ch[1]->tag+=p->tag;
        p->tag=0;
    }
}
inline void pushup(segtree *p)
{
    if(p->l==p->r)return;
    ll i;
    pushdown(p->ch[0]), pushdown(p->ch[1]);
    if(!building and p->w[1]==p->w[2])return;
    for(i=1;i<=6;i++)p->w[i]=p->ch[0]->w[i]+p->ch[1]->w[i];
}
void segtag(segtree *p, ll l, ll r)
{
    pushdown(p);
    ll mid=(p->l+p->r)>>1;
    if(l<=p->l and r>=p->r){p->tag++;return;}
    if(l<=mid)segtag(p->ch[0],l,r);
    if(r>mid)segtag(p->ch[1],l,r);
    pushup(p);
}
ll segsum(segtree *p, ll l, ll r)
{
    pushdown(p);
    ll mid=(p->l+p->r)>>1, ans=0;
    if(l<=p->l and r>=p->r)return p->w[1];
    if(l<=mid)ans+=segsum(p->ch[0],l,r);
    if(r>mid)ans+=segsum(p->ch[1],l,r);
    return ans;
}
void build(segtree *p, ll l, ll r)
{
    ll mid=(l+r)>>1, i;
    p->l=l, p->r=r;
    if(l==r)
    {
        p->w[1]=a[l];
        for(i=2;i<=6;i++)p->w[i]=sqrt(p->w[i-1]);
        return;
    }
    build(p->ch[0]=pool+ ++ndtot,l,mid);
    build(p->ch[1]=pool+ ++ndtot,mid+1,r);
    pushup(p);
}
int main()
{
    ll x, l, r, i;
    N=read();
    for(i=1;i<=N;i++)a[i]=read();
    build(root=pool+ ++ndtot,1,N);
    building=0;
    M=read();
    for(i=1;i<=M;i++)
    {
        x=read(), l=read(), r=read();
        if(x==1)printf("%lld\n",segsum(root,l,r));
        if(x==2)segtag(root,l,r);
    }
    return 0;
}