HDU 4027（线段树）

HDU4027

题意：操作指令为0时，对区间[x,y]之间的数字进行开平方；指令为1的时候，对区间[x,y]之间的数字求和并输出；

思路：线段树处理就OK了，但是64位内的数最多开8次平方就为1了（开始不信，试了试之后orz.......），所以在开平方的时候加一下限制条件使开平方操作提前结束没必要的操作就可以了，不然会超时。

代码中的这句：en - st + 1 == evil[rt]表示区间st到en中所有的数都是1，所以可以提前结束了。

代码：

/*
    Time:2018/8/20
    Author:sykai1
    Funtion:solve HDU 4027
*/
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
#define INF 0x3f3f3f3f

using namespace std;
typedef long long ll;
typedef pair<double, int> P;
const int maxn = 2e5 + 10;
ll evil[maxn * 4];
int n;

void Build(int rt, int st, int en)
{
    if (st == en)
    {
        scanf("%I64d", &evil[rt]);
        return;
    }

    int mid = (st + en) / 2;
    Build(rt << 1, st, mid);
    Build(rt << 1 | 1, mid + 1, en);
    evil[rt] = evil[rt << 1] + evil[rt << 1 | 1];
}

void Add(int rt, int x, int y, int st, int en)
{
    if (st == en)
    {
        evil[rt] = (ll)sqrt(evil[rt]);
        return;
    }
    if (en - st + 1 == evil[rt])
    {
        //printf("GG %d\n",evil[rt]);
        return;
    }

    int mid = (st + en) / 2;
    if (y <= mid)
        Add(rt << 1, x, y, st, mid);
    else if (x > mid)
        Add(rt << 1 | 1, x, y, mid + 1, en);
    else
    {
        Add(rt << 1, x, mid, st, mid);
        Add(rt << 1 | 1, mid + 1, y, mid + 1, en);
    }
    evil[rt] = evil[rt << 1] + evil[rt << 1 | 1];
}

ll query(int rt, int x, int y, int st, int en)
{
    ll res = 0;
    if (x == st && y == en)
        return evil[rt];
    int mid = (st + en) / 2;

    if (y <= mid)
        res = query(rt << 1, x, y, st, mid);
    else if (x > mid)
        res = query(rt << 1 | 1, x, y, mid + 1, en);
    else
    {
        res = query(rt << 1, x, mid, st, mid);
        res += query(rt << 1 | 1, mid + 1, y, mid + 1, en);
    }
    return res;
}


int main()
{
    int cnt = 0;
    while (scanf("%d", &n) != EOF)
    {
        printf("Case #%d:\n", ++cnt);
        Build(1, 1, n);
        int q, op, x, y;
        scanf("%d", &q);
        for (int i = 0; i < q; i++)
        {
            scanf("%d%d%d", &op, &x, &y);
            if (x > y) swap(x, y);
            if (op == 0)
                Add(1, x, y, 1, n);
            else
            {
                printf("%lld\n", query(1, x, y, 1, n));
            }
        }
        printf("\n");
    }
    return 0;
}