小Y学长的GCD难题【线段树】

  求一段区间的数的GCD的值，一道基础的线段树的题。

题目链接

思路：

用一个Lazy标记记录更新节点时候的值，然后当update的时候，如果不是正好包含区间，那么就得pushdown，查询操作也是一样的。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
using namespace std;
typedef long long ll;
const int maxN=200005;
int N,M;
struct node
{
    int l,r,val;
}trie[maxN<<2];
int lazy[maxN<<2];
int a[maxN];
int gcd(int a, int b)
{
    if(b==0) return a;
    return gcd(b, a%b);
}
void buildTree(int rt, int l, int r)
{
    trie[rt].l=l;
    trie[rt].r=r;
    if(l==r)
    {
        trie[rt].val=a[l];
        return;
    }
    int mid=(l+r)>>1;
    buildTree(rt<<1, l, mid);
    buildTree(rt<<1|1, mid+1, r);
    trie[rt].val=gcd(trie[rt<<1].val, trie[rt<<1|1].val);
}
void pushdown(int rt)
{
    if(lazy[rt])
    {
        lazy[rt<<1]=lazy[rt];
        lazy[rt<<1|1]=lazy[rt];
        trie[rt<<1].val=lazy[rt];
        trie[rt<<1|1].val=lazy[rt];
        lazy[rt]=0;
    }
}
void update(int rt, int l, int r, int ql, int qr, int val)
{
    if(ql<=l && qr>=r)
    {
        trie[rt].val=val;
        lazy[rt]=val;
        return;
    }
    int mid=(l+r)>>1;
    pushdown(rt);
    if(qr<=mid) update(rt<<1, l, mid, ql, qr, val);
    else if(ql>mid) update(rt<<1|1, mid+1, r, ql, qr, val);
    else
    {
        update(rt<<1, l, mid, ql, mid, val);
        update(rt<<1|1, mid+1, r, ql, qr, val);
    }
    trie[rt].val=gcd(trie[rt<<1].val, trie[rt<<1|1].val);
}
int Query(int rt, int l, int r, int ql, int qr)
{
    if(ql<=l && qr>=r) return trie[rt].val;
    if(lazy[rt]) return lazy[rt];
    int mid=(l+r)>>1;
    pushdown(rt);
    if(qr<=mid) return Query(rt<<1, l, mid, ql, qr);
    else if(ql>mid) return Query(rt<<1|1, mid+1, r, ql, qr);
    else
    {
        return gcd(Query(rt<<1, l, mid, ql, mid), Query(rt<<1|1, mid+1, r, mid+1, qr));
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(trie, 0, sizeof(trie));
        memset(lazy, 0, sizeof(lazy));
        memset(a, 0, sizeof(a));
        scanf("%d%d",&N,&M);
        for(int i=1; i<=N; i++) scanf("%d",&a[i]);
        buildTree(1, 1, N);
        while(M--)
        {
            int e1,e2,e3,e4;
            scanf("%d",&e1);
            if(e1==1)
            {
                scanf("%d%d%d",&e2,&e3,&e4);
                update(1, 1, N, e2, e3, e4);
            }
            else
            {
                scanf("%d%d",&e2,&e3);
                printf("%d\n",Query(1, 1, N, e2, e3));
            }
        }
    }
    return 0;
}