难道这道题只能用结构体来创建节点吗?

P4513 小白逛公园

Astk @ 2020-07-31 18:46:36

我偏不信,用数组写了一个线段树,查询的时候定义了一个新的变量存合并后的节点,然后进行查询。 我感觉这个想法没什么问题呀。 然而。。。。。。。它wa了. 下面是我的代码

#include<cstdio>
#include<iostream> 
#define mid (l+r>>1)
#define N 500010
using namespace std;
int sum[N<<2],dat[N<<2],lmax[N<<2],rmax[N<<2],ans,n,m,a[N];
void up(int rt){
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    dat[rt]=max(dat[rt<<1],max(dat[rt<<1|1],rmax[rt<<1]+lmax[rt<<1|1]));
    lmax[rt]=max(lmax[rt<<1],sum[rt<<1]+lmax[rt<<1|1]);
    rmax[rt]=max(rmax[rt<<1|1],sum[rt<<1|1]+rmax[rt<<1]);
}
void build(int rt,int l,int r){
    if(l==r){
        sum[rt]=dat[rt]=lmax[rt]=rmax[rt]=a[l];
        return;
    }
    build(rt<<1,l,mid);
    build(rt<<1|1,mid+1,r);
    up(rt);
    printf("rt:%d %d %d %d %d\n",rt,dat[rt],lmax[rt],rmax[rt],sum[rt]);
}
void change(int rt,int l,int r,int k,int val){
    if(l==r){
        sum[rt]=dat[rt]=lmax[rt]=rmax[rt]=val;
        return;
    }
    if(k<=mid) change(rt<<1,l,mid,k,val);
    else change(rt<<1|1,mid+1,r,k,val);
    up(rt);
}
int query(int rt,int l,int r,int L,int R){
    if(L<=l&&R>=r){
        return rt;
    }
    if(R<=mid) return query(rt<<1,l,mid,L,R);
    if(L>mid)return query(rt<<1|1,mid+1,r,L,R);
    int t1=query(rt<<1,l,mid,L,R),t2=query(rt<<1|1,mid+1,r,L,R);
    printf("t1:%d t2:%d",t1,t2);
    dat[ans]=max(dat[t1],max(dat[t2],rmax[t1]+lmax[t2]));
    lmax[ans]=max(lmax[t1],sum[t1]+lmax[t2]);
    rmax[ans]=max(rmax[t2],sum[t2]+rmax[t1]);
    sum[ans]=sum[t1]+sum[t2];
    printf("rt:%d %d %d %d %d\n",rt,dat[ans],lmax[ans],rmax[ans],sum[ans]); 
    return ans;
}
int main(){
    ans=N<<2-1;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    build(1,1,n);
    for(int i=1;i<=m;i++){
        int opt,x,y;
        scanf("%d%d%d",&opt,&x,&y);
        if(opt==1){
            if(x>y)swap(x,y);
            printf("%d\n",dat[query(1,1,n,x,y)]);
        }
        else
        change(1,1,n,x,y);
    }
}

用数组不能做是吗?


by 天命之路 @ 2020-07-31 18:59:09

也许不行


by 天命之路 @ 2020-07-31 19:00:26

你可以让query函数返回两个子区间合并后的结果,这样就只能用结构体了


by 天命之路 @ 2020-07-31 19:01:06

@Astk 一般而言,像这样状态较复杂的题目都用结构体保存节点信息


by Limit @ 2020-07-31 19:05:45

可以肯定是可以的


by Limit @ 2020-07-31 19:06:01

但是没有结构体方便


by 天命之路 @ 2020-07-31 19:09:11

你会发现,除开你的特判情况外,其余时候的query 都返回(N<<2-1)


by 万弘 @ 2020-07-31 19:13:43

@Astk

这里有一个单次询问要\mathcal O(\log n)空间的做法:

每次区间完全包含时,新建一个节点,复制当前节点所有信息并返回编号。合并时就合并左右儿子返回的两个点,并新建一个点储存这些信息,并返回编号。


by 天命之路 @ 2020-07-31 19:20:24

@Astk 所以你的query写的有点问题,最好参考 @万弘 的做法


by Astk @ 2020-07-31 19:34:30

@天命之路 @万弘 @Limit啊,谢谢你们,我刚看到


by Astk @ 2020-07-31 19:34:44

@Limit


| 下一页