V - Can you answer these queries? HDU - 4027 线段树

Posted 2022-05-30 echozqn

V - Can you answer these queries?

 HDU - 4027 

 

这个题目开始没什么思路，因为不知道要怎么去区间更新这个开根号。

然后稍微看了一下题解，因为每一个数开根号最多开十几次就变成1了，所以就直接单点更新，但是这个可以剪枝。

如果碰到区间长度和区间大小相同就可以不用更新了。

我也是无语了，想到刚刚的做法之后很好写了，

不过要注意以下x,y 大小可能x>y, 这个bug如果不是之前看题解的时候瞄到了，我可能找不出来了。

因为我写对拍我肯定会保证x<y 的，还是就是注意看清楚题目要求输出一行空行。

#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <string>
#include <iostream>
#include <algorithm>
#define inf 0x3f3f3f3f
#define inf64 0x3f3f3f3f3f3f3f3f
using namespace std;
const int maxn = 1e5 + 10;
typedef long long ll;
struct node

    ll len, sum;
tree[maxn*8];
ll a[maxn];

void push_up(int id)

    tree[id].sum = tree[id << 1].sum + tree[id << 1 | 1].sum;


void build(int id,int l,int r)

    tree[id].len = r - l + 1;
    if(l==r)
    
        tree[id].sum = a[l];
        return;
    
    int mid = (l + r) >> 1;
    build(id << 1, l, mid);
    build(id << 1 | 1, mid + 1, r);
    push_up(id);


void update(int id,int l,int r,int x,int y)

    if (x <= l && y >= r && tree[id].len == tree[id].sum) return;
    if(l==r)
    
        tree[id].sum = sqrt(tree[id].sum);
        return;
    
    int mid = (l + r) >> 1;
    if (x <= mid) update(id << 1, l, mid, x, y);
    if (y > mid) update(id << 1 | 1, mid + 1, r, x, y);
    push_up(id);


ll query(int id,int l,int r,int x,int y)

    if(x<=l&&y>=r) return tree[id].sum;
    int mid = (l + r) >> 1;
    ll ans = 0;
    if (x <= mid) ans += query(id << 1, l, mid, x, y);
    if (y > mid) ans += query(id << 1 | 1, mid + 1, r, x, y);
    return ans;


int main()

    int n, m, cas = 1;
    while(scanf("%d", &n)!=EOF)
    
        for (int i = 1; i <= n; i++) scanf("%lld", &a[i]);
        build(1, 1, n);
        scanf("%d", &m);
        cout << "Case #" << cas++ << ":" << endl;
        while(m--)
        
            int opt, x, y;
            scanf("%d%d%d", &opt, &x, &y);
            if (x > y) swap(x, y);
            if(opt==0) update(1, 1, n, x, y);
            else
            
                ll ans = query(1, 1, n, x, y);
                printf("%lld\n", ans);
            
        
        printf("\n");
    
    return 0;

线段树