HDU 4902 Nice boat 线段树 区间更新

HDU 4902 Nice boat

题意:

一个序列, 两种操作. 

1. 把一个区间全部更改为x

2. 把一个区间比x大的数全部修改为 gcd(ai,x)

思路:

lazy tag

pushDown的原则是把该节点所有tag都推到子节点

所以分情况讨论一下即可  最后到叶节点直接更新数列

#include <bits/stdc++.h>
#define lson num<<1
#define rson num<<1|1
#define gl l,m,lson
#define gr m+1,r,rson
#define PARA int l=1,int r=n,int num=1
using namespace std;
const int MAXN = 1e5+100;

int n;
int val[MAXN];
int gcd(int a,int b)
{
    if(b==0) return a;
    else return gcd(b,a%b);
}
struct SegTree
{
    struct Node
    {
        int l,r;
        int tag[2];
        void init(int a,int b)
        {
            l=a,r=b;
            tag[0]=tag[1]=-1;
        }
        void deal()
        {
            if(tag[0]!=-1)
                val[l]=tag[0];
            if(tag[1]!=-1&&val[l]>tag[1])
                val[l]=gcd(tag[1],val[l]);

        }
    }st[MAXN<<2];
    void init(PARA)
    {
        st[num].init(l,r);
        int m=l+r>>1;
        if(l!=r)
            init(gl),init(gr);
    }

    void pushDown(int num)
    {
        int l=st[num].l,r=st[num].r;
        if(l!=r)
        {
            if(st[num].tag[0]!=-1)
            {
                st[lson].tag[0]=st[num].tag[0];
                st[rson].tag[0]=st[num].tag[0];
                st[lson].tag[1]=-1;
                st[rson].tag[1]=-1;
            }
            if(st[num].tag[1]!=-1)
            {
                if(st[lson].tag[0]!=-1)
                {
                    if(st[lson].tag[0]>st[num].tag[1])
                        st[lson].tag[0]=gcd(st[lson].tag[0],st[num].tag[1]);
                }
                else
                {
                    if(st[lson].tag[1]!=-1)
                        pushDown(lson);
                    st[lson].tag[1]=st[num].tag[1];
                }
                if(st[rson].tag[0]!=-1)
                {
                    if(st[rson].tag[0]>st[num].tag[1])
                        st[rson].tag[0]=gcd(st[rson].tag[0],st[num].tag[1]);
                }
                else
                {
                    if(st[rson].tag[1]!=-1)
                        pushDown(rson);
                    st[rson].tag[1]=st[num].tag[1];
                }
            }
        }
        else
            st[num].deal();
        st[num].tag[0]=-1;
        st[num].tag[1]=-1;
    }
    void update(int a,int b,int f,int v,PARA)
    {
        pushDown(num);
        if(a<=l&&r<=b)
            st[num].tag[f]=v;
        else
        {
            int m=l+r>>1;
            if(b<=m)
                update(a,b,f,v,gl);
            else if(a>m)
                update(a,b,f,v,gr);
            else
                update(a,b,f,v,gl),update(a,b,f,v,gr);
        }
    }
    void dfs(PARA)
    {
        pushDown(num);
        int m=l+r>>1;
        if(l!=r)
            dfs(gl),dfs(gr);
    }
}soul;

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d",&val[i]);
        soul.init();
        int m;
        scanf("%d",&m);
        while(m--)
        {
            int a,b,c,d;
            scanf("%d%d%d%d",&a,&b,&c,&d);
            soul.update(b,c,a-1,d);
        }
        soul.dfs();
        for(int i=1;i<=n;i++)
            printf("%d ",val[i]);
        puts("");
    }
    return 0;
}