Spoj 1825 Freetour2

点分治第二题，分治过程写的比较熟练了，但搞答案的时候还是想了好久，写了好久.......
主要解释一下求g的那块吧：
首先dep[i]表示rot的第i个儿子的子树里黑点最多的路径的的黑点数。

黑点限制是K，我们可以求出rot的每一个儿子i黑点数为j的最大长度g[i,j]，然后每次求出一个j1+j2<=K,且g[i1,j1]+g[i2,j2]最大的。

如果我们枚举i2，那么只需要求出一个可以和i2配并且最长的i1就行了。这个可以用个什么数据结构维护，也可以“暴力”。由于顺序是对答案没有影响的，那么我们按照dep升序排序，然后按照这个顺序计算，mg[j]表示前i个点的g[j]的最大值，如果维护好mg，就可以用g+mg来更新答案了。最恶心就是更新答案这里，假设当前枚举到了g[j]，我们需要找到一个≤K-j-black[rot]的，但是我们不能直接用mg[K-j-black[rot]]，因为前面i-1个点可能没有达到K-j-black[rot]个黑点的，这样我们得用第i-1个点的dep。就说这些吧，其他地方看论文就行了。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#define inf 2147483647
#define maxn 420000
using namespace std;

struct et
{
    int s,t,val,next;
}e[maxn];
int fa[maxn],fir[maxn],size[maxn];
int dep[maxn],son[maxn],g[maxn],mg[maxn],way[maxn];
bool ex[maxn],black[maxn];
int n,m,p,tot,ans,num;

void dfs(int now)
{
    size[now]=1;
    for (int j=fir[now];j;j=e[j].next)
    {
        int k=e[j].t;
        if (k!=fa[now])
        {
            fa[k]=now;
            dfs(k);
            size[now]+=size[k];
        }
    }
}

void choose(int &rot)
{
    int now,mx=0;
    for (int j=fir[rot];j;j=e[j].next)
    {
        int k=e[j].t;
        if (ex[k]&&k!=fa[rot]&&size[k]>mx)
        {
            mx=size[k];
            now=k;
        }
    }
    if (mx>num/2) 
    {
        size[rot]-=size[now];
        size[now]=num;
        fa[rot]=now;
        fa[now]=0;
        choose(rot=now);
    }
}

int getdep(int now)
{
    int tmp=0;
    for (int j=fir[now];j;j=e[j].next)
    {
        int k=e[j].t;
        if (ex[k]&&k!=fa[now]) tmp=max(tmp,getdep(k));
    }
    return black[now]+tmp;
}

int getway(int now,int sum,int dis)
{
    g[sum]=max(g[sum],dis);
    for (int j=fir[now];j;j=e[j].next)
    {
        int k=e[j].t;
        if (ex[k]&&k!=fa[now]) getway(k,sum+black[k],dis+e[j].val);
    }
}

bool cmp(int a,int b)
{
     return dep[e[a].t]<dep[e[b].t];
}

void solve(int rot)
{
    if (num==1) return;
    choose(rot);
    int sum=0;
    for (int j=fir[rot];j;j=e[j].next)
    {
        int k=e[j].t;
        if (ex[k]) sum++,dep[k]=getdep(k),way[sum]=j; 
    }
    sort(way+1,way+sum+1,cmp);
    for (int i=0;i<=dep[e[way[sum]].t];i++) mg[i]=-inf;
    for (int i=1;i<=sum;i++)
    {
        int k=e[way[i]].t;
        for (int j=0;j<=dep[k];j++) g[j]=-inf;
        getway(k,black[k],e[way[i]].val);
        if (i!=1)
        {
           for (int j=0;j<=p-black[rot]&&j<=dep[k];j++)
           {
               int len=min(p-black[rot]-j,dep[e[way[i-1]].t]);
               if (mg[len]==-inf) break;
               if (g[j]!=-inf) ans=max(ans,mg[len]+g[j]);
           }    
        }
        for (int j=0;j<=dep[k];j++)
        {
            mg[j]=max(g[j],mg[j]);
            if (j) mg[j]=max(mg[j-1],mg[j]);
            if (j+black[rot]<=p) ans=max(ans,mg[j]);
        }   
    }
    ex[rot]=0;
    for (int j=fir[rot];j;j=e[j].next)
    {
        int k=e[j].t;
        if (ex[k]) 
        {
            num=size[k];
            solve(k);
        }
    }
}

void add(int x,int y,int z)
{
    e[++tot].s=x; e[tot].t=y; e[tot].val=z; e[tot].next=fir[x]; fir[x]=tot;
    e[++tot].s=y; e[tot].t=x; e[tot].val=z; e[tot].next=fir[y]; fir[y]=tot;
}

int main()
{
    //freopen("freetour.in","r",stdin);
    scanf("%d %d %d",&n,&p,&m);
    int x,y,z;
    tot=0;
    for (int i=1;i<=m;i++)
    {
        scanf("%d",&x);
        black[x]=1;
    }
    for (int i=1;i<n;i++)
    {
        scanf("%d %d %d",&x,&y,&z);
        add(x,y,z);
    }
    memset(ex,1,sizeof(ex));
    dfs(1);
    num=n;ans=0;
    solve(1);
    printf("%d\n",ans);
}