花神游历各国（线段树，区间开平方）

题目描述

原题来自：BZOJ 3211

花神喜欢步行游历各国，顺便虐爆各地竞赛。花神有一条游览路线，它是线型的，也就是说，所有游历国家呈一条线的形状排列，花神对每个国家都有一个喜欢程度（当然花神并不一定喜欢所有国家）。

每一次旅行中，花神会选择一条旅游路线，它在那一串国家中是连续的一段，这次旅行带来的开心值是这些国家的喜欢度的总和，当然花神对这些国家的喜欢程序并不是恒定的，有时会突然对某些国家产生反感，使他对这些国家的喜欢度 x 变为 sqrt(x)（可能是花神虐爆了那些国家的 OI，从而感到乏味）。

现在给出花神每次的旅行路线，以及开心度的变化，请求出花神每次旅行的开心值。

输入格式

第一行是一个整数 n，表示有 n个国家；

第二行有 n 个空格隔开的整数，表示每个国家的初始喜欢度xi ；

第三行是一个整数m ，表示有 m条信息要处理；

输出格式

每次 x=1 时，每行一个整数。表示这次旅行的开心度。

样例

样例输入

4
1 100 5 5
5
1 1 2
2 1 2
1 1 2
2 2 3
1 1 4

样例输出

101
11
11

 

题目链接：https://loj.ac/problem/10128 

思路：根据数据范围我们知道，对于一个数来说，不会开很多次根号。再来考虑 sqrt(0)=0  sqrt(1)=1

所有对于连续 0 或 1 的数据根本不需要处理的。所以我们用一个标记记录区间是否需要操作，继续跑线段树就好了。

AC代码如下：

#include<bits/stdc++.h>
#define ll long long
#define N 400010
using namespace std;
ll Sum[N];
bool book[N];
void build(int l,int r,int o)
{
    if(l==r)
    {
        scanf("%lld",&Sum[o]);
        book[o]=((!Sum[o])||(Sum[o]==1));
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,o<<1);
    build(mid+1,r,o<<1|1);
    Sum[o]=Sum[o<<1]+Sum[o<<1|1];
    book[o]=book[o<<1]&&book[o<<1|1];
    return;
}
void update(int x,int y,int l,int r,int o)
{
    if(book[o])return;
    if(l==r)
    {
        Sum[o]=(ll)sqrt(Sum[o]+0.0);
        book[o]=((!Sum[o])||(Sum[o]==1));
        return;
    }
    int mid=(l+r)>>1;
    if(x<=mid)update(x,y,l,mid,o<<1);
    if(y>mid)update(x,y,mid+1,r,o<<1|1);
    Sum[o]=Sum[o<<1]+Sum[o<<1|1];
    book[o]=book[o<<1]&&book[o<<1|1];
}
ll query(int x,int y,int l,int r,int o)
{
    ll sum=0;
    if(l>=x&&r<=y)return Sum[o];
    int mid=(l+r)>>1;
    if(x<=mid)sum+=query(x,y,l,mid,o<<1);
    if(y>mid)sum+=query(x,y,mid+1,r,o<<1|1);
    return sum;
}
int main()
{
    int n;
    scanf("%d",&n);
    build(1,n,1);
    int m;
    scanf("%d",&m);
    while(m--)
    {
        int op,l,r;
        scanf("%d%d%d",&op,&l,&r);
        if(op==1)printf("%lld\n",query(l,r,1,n,1));
        else update(l,r,1,n,1);
    }
    return 0;
}