上帝造题的七分钟 2 / 花神游历各国（线段树）

green qwq

于 2024-05-03 21:42:35 发布

阅读量905

点赞数 30

分类专栏：洛谷算法题合集文章标签：算法

本文链接：https://blog.csdn.net/m0_60738889/article/details/138425332

版权

洛谷算法题合集专栏收录该内容

142 篇文章 0 订阅

订阅专栏

这篇文章讲述了如何使用线段树数据结构解决一个包含区间修改（开平方）和区间查询的编程问题，模拟《上帝造题的七分钟》第二部中的数学题设置，要求处理数据范围、时间限制和整数运算限制。

摘要由CSDN通过智能技术生成

上帝造题的七分钟 2 / 花神游历各国

题目背景

XLk 觉得《上帝造题的七分钟》不太过瘾，于是有了第二部。

题目描述

"第一分钟，X 说，要有数列，于是便给定了一个正整数数列。

第二分钟，L 说，要能修改，于是便有了对一段数中每个数都开平方(下取整)的操作。

第三分钟，k 说，要能查询，于是便有了求一段数的和的操作。

第四分钟，彩虹喵说，要是 noip 难度，于是便有了数据范围。

第五分钟，诗人说，要有韵律，于是便有了时间限制和内存限制。

第六分钟，和雪说，要省点事，于是便有了保证运算过程中及最终结果均不超过 $64$ 位有符号整数类型的表示范围的限制。

第七分钟，这道题终于造完了，然而，造题的神牛们再也不想写这道题的程序了。"

——《上帝造题的七分钟·第二部》

所以这个神圣的任务就交给你了。

输入格式

第一行一个整数 $n$ ，代表数列中数的个数。

第二行 $n$ 个正整数，表示初始状态下数列中的数。

第三行一个整数 $m$ ，表示有 $m$ 次操作。

接下来 $m$ 行每行三个整数 k l r。

$k = 0$ 表示给 $[l, r]$ 中的每个数开平方（下取整）。
$k = 1$ 表示询问 $[l, r]$ 中各个数的和。

数据中有可能 $l > r$ ，所以遇到这种情况请交换 $l$ 和 $r$ 。

输出格式

对于询问操作，每行输出一个回答。

样例 #1

样例输入 #1

10
1 2 3 4 5 6 7 8 9 10
5
0 1 10
1 1 10
1 1 5
0 5 8
1 4 8

样例输出 #1

19
7
6

提示

对于 $30\%$ 的数据， $1\le n,m\le 10^3$ ，数列中的数不超过 $32767$ 。

对于 $100\%$ 的数据， $1\le n,m\le 10^5$ ， $1\le l,r\le n$ ，数列中的数大于 $0$ ，且不超过 $10^{12}$ 。

思路

本道题要实现区间修改，区间查询，数据规模为： $1\le n,m\le 10^5$ ，因此我们用线段树再合适不过了，但本题需要注意：

细节1：根号下取整对于x<1的不凑效，因此x<1的我们就return。
细节2：因为这道题是区间修改一个数，而且区间修改的数不能统一加减（意思就是sqrt跟里面的x有关），因此我们就得类似单点修改的写法写modify。

代码

//区间修改，区间查询=>线段树
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#define int long long

using namespace std;

const int N = 1e5+10;

int w[N],s[N],t[N];
int n,q;
struct E{
    int l,r;
    int maxv,val;
}tr[N*4];

void pushup(int u){
    tr[u].val=tr[u<<1].val+tr[u<<1|1].val;
    tr[u].maxv=max(tr[u<<1].maxv,tr[u<<1|1].maxv);
}

void build(int u,int l,int r){
    if(l==r){
        tr[u]={l,r,w[l],w[l]};
        return;
    }
    tr[u]={l,r};
    int mid=(l+r)>>1;
    build(u<<1,l,mid),build(u<<1|1,mid+1,r);
    pushup(u);
}

void modify(int u,int l,int r){
    if(tr[u].maxv<=1)return;//注意这行
    if(tr[u].l==tr[u].r){
        tr[u].val=tr[u].maxv=sqrt(tr[u].val);
        return;
    }
    
    int mid=(tr[u].l+tr[u].r)>>1;
    if(l<=mid)modify(u<<1,l,r);
    if(r>mid)modify(u<<1|1,l,r);
    pushup(u);
}

int query(int u,int l,int r){
    if(tr[u].l>=l&&tr[u].r<=r)return tr[u].val;
    int mid=(tr[u].l+tr[u].r)>>1;
    int v=0;
    if(l<=mid)v=query(u<<1,l,r);
    if(r>mid)v+=query(u<<1|1,l,r);
    return v;
}

signed main(){
    scanf("%lld",&n);
    
    for(int i=1;i<=n;i++)scanf("%lld",&w[i]);
    
    build(1,1,n);
    
    scanf("%lld",&q);
    
    while(q--){
        int op,l,r;
        // cin>>op>>l>>r;
        scanf("%lld%lld%lld",&op,&l,&r);
        if(l>r)swap(l,r);
        if(op==0){
            modify(1,l,r);
        }else{
            printf("%lld\n",query(1,l,r));
        }
    }
    
    return 0;
}