[bzoj4860]树的难题

题目大意

[bzoj4860]树的难题_第1张图片

点分治

点剖。
每颗子树按最上面那条边的颜色排序。
顺序扫,同颜色的用一颗线段树,其他也用一颗线段树。
每扫过一个颜色合并一下两颗线段树。
这个方法比单调队列不知道低到哪里去了。

#include
#include
#define max(a,b) (a>b?a:b)
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=200000+10,inf=2000000010;
int tree[maxn*4][2],d[maxn],num[maxn],belong[maxn],a[maxn],b[maxn],co[maxn],size[maxn],tt[maxn];
int bb[maxn],wz[maxn];
bool bz[maxn],pd[maxn*4][2];
int c[maxn],h[maxn],go[maxn*2],dis[maxn*2],next[maxn*2];
int i,j,k,l,r,t,n,m,L,R,tot,top,ans,cnt;
int read(){
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9'){
        if (ch=='-') f=-1;
        ch=getchar();
    }
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
void add(int x,int y,int z){
    go[++tot]=y;
    dis[tot]=z;
    next[tot]=h[x];
    h[x]=tot;
}
void travel(int x,int y){
    top++;
    int t=h[x];
    size[x]=1;
    while (t){
        if (!bz[go[t]]&&go[t]!=y){
            travel(go[t],x);
            size[x]+=size[go[t]];
        }
        t=next[t];
    }
}
void dfs(int x,int y,int bl,int cc,int z){
    a[++top]=x;
    belong[x]=bl;
    co[x]=cc;
    size[x]=1;
    int t=h[x];
    while (t){
        if (!bz[go[t]]&&go[t]!=y){
            num[go[t]]=num[x];
            if (dis[t]!=z) num[go[t]]+=c[dis[t]];
            d[go[t]]=d[x]+1;
            dfs(go[t],x,bl,cc,dis[t]);
            size[x]+=size[go[t]];
        }
        t=next[t];
    }
}
bool cmp(int x,int y){
    return co[x]y]||co[x]==co[y]&&belong[x]y];
}
void mark(int p,int x){
    //if (pd[p][x]) return;
    pd[p][x]=1;
    tree[p][x]=-inf;
}
void down(int p,int x){
    if (pd[p][x]){
        mark(p*2,x);
        mark(p*2+1,x);
        pd[p][x]=0;
    }
}
void change(int p,int l,int r,int a,int b,int x){
    if (l==r){
        tree[p][x]=max(tree[p][x],b);
        return;
    }
    down(p,x);
    int mid=(l+r)/2;
    if (a<=mid) change(p*2,l,mid,a,b,x);else change(p*2+1,mid+1,r,a,b,x);
    tree[p][x]=max(tree[p*2][x],tree[p*2+1][x]);
}
int query(int p,int l,int r,int a,int b,int x){
    if (tree[p][x]==-inf) return -inf;
    if (l==a&&r==b) return tree[p][x];
    down(p,x);
    int mid=(l+r)/2;
    if (b<=mid) return query(p*2,l,mid,a,b,x);
    else if (a>mid) return query(p*2+1,mid+1,r,a,b,x);
    else return max(query(p*2,l,mid,a,mid,x),query(p*2+1,mid+1,r,mid+1,b,x));
}
void calc(){
    int i,j,k,l,r,t;
    fo(i,1,top)
        if (L<=d[a[i]]&&d[a[i]]<=R) ans=max(ans,num[a[i]]);
    fo(i,1,top) tt[i]=-inf;
    j=1;l=1;
    mark(1,0);mark(1,1);
    while (j<=top){
        k=j;
        while (k1]]==belong[a[j]]) k++;
        fo(i,j,k){
            if (d[a[i]]>=R) continue;
            r=query(1,1,n,max(L-d[a[i]],1),R-d[a[i]],0);
            t=query(1,1,n,max(L-d[a[i]],1),R-d[a[i]],1);
            t=max(r,t);
            if (t==-inf) continue;
            ans=max(ans,t+num[a[i]]);
        }
        fo(i,j,k) change(1,1,n,d[a[i]],num[a[i]]-c[co[a[i]]],1);
        if (k1]]!=co[a[l]]){
            fo(i,l,k){
                if (num[a[i]]<=tt[d[a[i]]]) continue;
                tt[d[a[i]]]=num[a[i]];
                change(1,1,n,d[a[i]],num[a[i]],0);
            }
            mark(1,1);
            l=k+1;
        }
        j=k+1;
    }
}
void solve(int x){
    top=0;
    travel(x,0);
    if (topreturn;
    int i,j=x,k=0,t;
    while (1){
        t=h[j];
        while (t){
            if (!bz[go[t]]&&go[t]!=k&&size[go[t]]>top/2){
                k=j;
                j=go[t];
                break;
            }
            t=next[t];
        }
        if (!t) break;
    }
    t=h[j];
    d[j]=num[j]=0;
    top=cnt=0;
    while (t){
        if (!bz[go[t]]){
            d[go[t]]=1;
            num[go[t]]=c[dis[t]];
            dfs(go[t],j,go[t],dis[t],dis[t]);
            bb[++cnt]=go[t];
        }
        t=next[t];
    }
    //sort(a+1,a+top+1,cmp);
    sort(bb+1,bb+cnt+1,cmp);
    k=0;
    fo(i,1,cnt){
        wz[bb[i]]=k+1;
        k+=size[bb[i]];
    }
    i=1;
    while (i<=top){
        k=i;
        while (k1]]==belong[a[i]]) k++;
        fo(t,i,k)
            bb[wz[a[i]]+t-i]=a[t];
        i=k+1;
    }
    fo(i,1,top) a[i]=bb[i];
    calc();
    bz[j]=1;
    t=h[j];
    while (t){
        if (!bz[go[t]]) solve(go[t]);
        t=next[t];
    }
}
int main(){
    freopen("journey.in","r",stdin);freopen("journey.out","w",stdout);
    n=read();m=read();L=read();R=read();
    fo(i,1,m) c[i]=read();
    fo(i,1,n-1){
        j=read();k=read();l=read();
        add(j,k,l);add(k,j,l);
    }
    ans=-inf;
    solve(n/2);
    printf("%d\n",ans);
}

你可能感兴趣的:(点分治,线段树,单调队列)