hdu 6201 树分治

题意：一棵树，边全，点权，点权是书的价格，边权是运费。任选两个点买卖，求最大 利益。

思路：nlogn树分治，当时一看1e5，又是求树上路径的，想也没想直接莽分治，分治的过程中维护最大作为买和卖的利益就好了。

代码：

#include<bits/stdc++.h>
using namespace std;
#define X first
#define Y second
#define PB push_back
#define MP make_pair
#define MEM(a,b) memset(a,b,sizeof(a))
typedef long long ll;
const ll mod = 1e9+7;
const int maxn =1e6+10;
int n,t;
int ans,root,Max;
struct node{
    int v,next,w;
}edge[maxn*2];
int head[maxn],tot,num1,num2;
int si[maxn],maxv[maxn],vis[maxn];//vis标记重心
int dis[maxn],a[maxn],tmpsell,tmpbuy;
void init(){
    tot=ans=0;MEM(head,-1);MEM(vis,0);
}
void add_edge(int u,int v,int w){
    edge[tot].v=v;edge[tot].w=w;edge[tot].next=head[u];head[u]=tot++;
}
//处理子树的大小（固定）
void dfssi(int u,int f){
    si[u]=1;
    maxv[u]=0;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].v;
        if(v==f||vis[v])continue;
        dfssi(v,u);
        si[u]+=si[v];
        if(si[v]>maxv[u])maxv[u]=si[v];
    }
}
//找重心（固定）
void dfsroot(int r,int u,int f){
    if(si[r]-si[u]>maxv[u])//si[r]-si[u]是u上面部分的树的尺寸，跟u的最大孩子比，找到最大孩子的最小差值节点
        maxv[u]=si[r]-si[u];
    if(maxv[u]<Max)Max=maxv[u],root=u;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].v;
        if(v==f||vis[v])continue;
        dfsroot(r,v,u);
    }
}

//求每个点到u的距离（自定义）
void dfsdis(int u,int d,int f){
    tmpsell=min(tmpsell,d-a[u]);
    tmpbuy=min(tmpbuy,d+a[u]);
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].v;
        if(v!=f&&!vis[v])//不跨过其他重心
            dfsdis(v,d+edge[i].w,u);
    }
}

//分边解决一个重心的答案（自定义）
int calc(int u,int d){
    int ret=0;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].v,w=edge[i].w;
        if(!vis[v]){
            tmpsell=tmpbuy=1e9+10000;
            dfsdis(v,w,0);
            ret=max(ret,-a[u]-tmpsell);
            ret=max(ret,a[u]-tmpbuy);
            ret=max(ret,-tmpsell-tmpbuy);
        }
    }
    return ret;
}

void dfs(int u){
    Max=n;
    dfssi(u,0);//重心预处理
    dfsroot(u,u,0);//找重心
    vis[root]=1;
    ans=max(ans,calc(root,0));//合并贡献（固定？）
    for(int i=head[root];i!=-1;i=edge[i].next){
        int v=edge[i].v;
        if(!vis[v])dfs(v);
    }
}

int main(){
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        int u,v,w;
        init();
        for(int i=1;i<n;i++){
            scanf("%d%d%d",&u,&v,&w);
            add_edge(u,v,w);add_edge(v,u,w);
        }
        dfs(1);
        printf("%d
",ans);
    }
    return 0;
}