hdu 4902 Nice boat 線段樹

如果一個線段的值x不爲-1,表示該線段全爲x,否則表示該線段的值不唯一。

modify1是裸的線段樹,modify2其實就是一直找到某線段的值唯一,當modify1處理。

#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
const int N=100005;
int tree[N*4];//該線段的值全爲x,-1表示不統一

int gcd(int a,int b){
    return b==0?a:gcd(b,a%b);
}

#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

void push_up(int rt){
    if(tree[rt<<1]==tree[rt<<1|1])
        tree[rt]=tree[rt<<1];
}

void push_down(int rt){
    if(tree[rt]!=-1){
        tree[rt<<1]=tree[rt<<1|1]=tree[rt];
        tree[rt]=-1;//即將修改下面的值,等待重新維護
    }
}

void build(int l,int r,int rt){
    tree[rt]=-1;
    if(l==r){
        scanf("%d",&tree[rt]);
        return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    push_up(rt);
}

void print(int l,int r,int rt){
    if(l==r){
        printf("%d ",tree[rt]);
        return;
    }
    push_down(rt);
    int m=(l+r)>>1;
    print(lson);
    print(rson);
}

void modify1(int a,int b,int c,int l,int r,int rt){
    //printf("%d %d %d %d %d %d %d--\n",a,b,c,l,r,rt,tree[rt]);
    if(a<=l&&b>=r){
        tree[rt]=c;
        return;
    }
    push_down(rt);
    int m=(l+r)>>1;
    if(a<=m) modify1(a,b,c,lson);
    if(b>m) modify1(a,b,c,rson);
    push_up(rt);
}

void modify2(int a,int b,int c,int l,int r,int rt){
    //printf("%d %d %d %d %d %d %d--\n",a,b,c,l,r,rt,tree[rt]);
    if(a<=l&&b>=r && tree[rt]!=-1){
        if(tree[rt]>c)
            tree[rt]=gcd(tree[rt],c);
        return;
    }
    push_down(rt);
    int m=(l+r)>>1;
    if(a<=m) modify2(a,b,c,lson);
    if(b>m) modify2(a,b,c,rson);
    push_up(rt);
}

int main(){
    int tt,n,Q,op,a,b,c;
    cin>>tt;
    while(tt--){
        cin>>n;
        build(1,n,1);
        cin>>Q;
        while(Q--){
            scanf("%d %d %d %d",&op,&a,&b,&c);
            if(op==1) modify1(a,b,c,1,n,1);
            else modify2(a,b,c,1,n,1);
        }
        print(1,n,1); puts("");
    }
    return 0;
}

/*
1 l-r to x
2 l-r >x to gcd(x,a[i])
暴力+線段樹
06-28-50 44min
*/

点赞