码迷,mamicode.com
首页 > 其他好文 > 详细

HDU 4578 Transformation (线段树区间多种更新)

时间:2016-07-13 01:08:41      阅读:199      评论:0      收藏:0      [点我收藏+]

标签:

http://acm.hdu.edu.cn/showproblem.php?pid=4578

题目大意:对于一个给定序列,序列内所有数的初始值为0,有4种操作。1:区间(x, y)内的所有数字全部加上C;2:区间(x, y)内所有数字全部乘C; 3:区间(x, y)内的所有数字全部重置为C;

4:输出区间(x, y)内所有数字的P次方的和。由于题目为多实例(至少10组样例),故很耿直的更新到叶子节点明显会TLE;因此需优化。可发现题目所有操作都是对区间进行,因此只需要更新

到区间内数字相同即可。再者注意可进行状态压缩,不需要的累加和累乘只标记即可,需要此部分时再往下更新;更新时先更新3,因为3会覆盖掉1和2;之后再进行累乘,因为累乘影响累加,累

加不影响累乘。注意细节即可。

 

 

 

///至少10组样例,则八秒实际上不多,需优化。
///由于所有操作都是对区间进行,故数字保存情况为
///分区间相同,因此只需要操作到区间数字相同时即可,不必处理到最下面的叶子节点
#include <stdio.h>
#include <algorithm>
using namespace std;
#define N 110000
#define mod 10007
#define lson rt<<1
#define rson rt<<1|1
struct tree
{
    int l, r, add, mul, op, num;
    ///add记录累加的数,mul记录累乘的数,op记录操作的状态
    ///op为1表示区间内数字相同,op为0表示区间内数字不同,需向下
    ///继续进行操作,op为2表示区间被重新赋值,需向下更新(操作3)。
    int len()
    {
        return r-l+1;
    }
}a[N<<2];
void build(int l, int r, int rt)///初始化
{
    a[rt].l = l;
    a[rt].r = r;
    a[rt].mul = a[rt].op = 1;
    a[rt].add = a[rt].num = 0;
    if(l==r)return ;
    int mid = (l+r)/2;
    build(l, mid, lson);
    build(mid+1, r, rson);
}
void Change(int rt, int op, int k)
{
    if(op==3)///重新赋值,即再次初始化
    {
        a[rt].num = k%mod;
        a[rt].mul = 1;
        a[rt].add = 0;
        a[rt].op = 2;///重新赋值后子区间所有都重新覆盖
    }
    else if(op==2)
    {
        (a[rt].add *= k) %= mod;
        (a[rt].mul *= k) %= mod;
        (a[rt].num *= k) %= mod;
    }
    else
    {
        (a[rt].add += k) %= mod;
        (a[rt].num += k) %= mod;
    }
}

void Up(int rt)
{
    if(a[lson].op && a[rson].op)///若子区间为同数字区间且两子区间数字相同,
    if(a[lson].num == a[rson].num)///则可向上合并给父区间
    {
        a[rt].num = a[lson].num;
        a[rt].op = 1;
    }
}

void Down(int rt)///向下的状态压缩,若不需此区间作答此区间暂时储存;
{                ///若需此区间作答则向下更新一层直到叶子节点
    if(a[rt].l != a[rt].r)
    {
        if(a[rt].op==2)
        {
            a[lson].num = a[rson].num = a[rt].num;

            a[lson].op = a[rson].op = 2;
            a[lson].add = a[rson].add = 0;
            a[lson].mul = a[rson].mul = 1;

            a[rt].add = 0;
            a[rt].mul = 1;
            a[rt].op = 1;
        }

        if(a[rt].mul>1)///注意此处,先更新乘法,因为累乘会影响累加的状态
        {
            (a[lson].num *= a[rt].mul) %= mod;
            (a[lson].add *= a[rt].mul) %= mod;
            (a[lson].mul *= a[rt].mul) %= mod;

            (a[rson].num *= a[rt].mul) %= mod;
            (a[rson].add *= a[rt].mul) %= mod;
            (a[rson].mul *= a[rt].mul) %= mod;

            a[rt].mul = 1;
        }

        if(a[rt].add)
        {
            (a[lson].num += a[rt].add) %= mod;
            (a[lson].add += a[rt].add) %= mod;

            (a[rson].num += a[rt].add) %= mod;
            (a[rson].add += a[rt].add) %= mod;

            a[rt].add = 0;
        }
    }
}
void Update(int rt, int op, int l, int r, int k)
{
    if(a[rt].l==l && a[rt].r==r && a[rt].op)///找到数字相同区间
    {
        Change(rt, op, k);///执行操作
        return ;
    }

    Down(rt);
    a[rt].op = 0;///假设默认区间数字已改变,标记为不同。

    int mid = (a[rt].l + a[rt].r)/2;
    if(mid>=r)Update(lson, op, l, r, k);
    else if(mid<l)Update(rson, op, l, r, k);
    else
    {
        Update(lson, op, l, mid, k);
        Update(rson, op, mid+1, r, k);
    }

    Up(rt);///执行操作后向上回溯,用已得到的子区间反馈负区间的状态
}
int Query(int rt, int l, int r, int p)
{
    if(a[rt].l==l && a[rt].r==r && a[rt].op)///找到同数字区间即可计算
    {
        int ans = 1;
        for(int i=1; i<=p; i++)///一个p次方
            (ans *= a[rt].num) %= mod;
        ans = (ans * a[rt].len())%mod; ///区间内所有p次方
        return ans;
    }

    Down(rt);

    int mid = (a[rt].l + a[rt].r)/2;

    if(mid>=r)return Query(lson, l, r, p);
    else if(mid<l)return Query(rson, l, r, p);
    else
    {
        int lans = Query(lson, l, mid, p);
        int rans = Query(rson, mid+1, r, p);
        return (lans+rans)%mod;
    }
}
int main()
{
    int n, m;
    while(scanf("%d %d", &n, &m), m+n)
    {
        build(1, n, 1);
        int op, l, r, k;
        while(m--)
        {
            scanf("%d %d %d %d", &op, &l, &r, &k);
            if(op!=4)Update(1, op, l, r, k);///只要不为4都是更新操作
            else printf("%d\n", Query(1, l, r, k));
        }
    }
    return 0;
}

 

HDU 4578 Transformation (线段树区间多种更新)

标签:

原文地址:http://www.cnblogs.com/zznulw/p/5665214.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!