bzoj3743 [Coci2015]Kamp 常州模拟赛d6t2

【数据规模】

K <= N <= 500000

1 <= x,y <= N, 1 <= z <= 1000000

分析：本来想用状压dp的，但是n这么大，怎么压的下来。其实这道题有一个结论：如果我们把要送达的点建立一棵树，那么答案就是这棵树上的边权和*2+i到这棵树的最短距离(假设为j点)-j到树上最远一点的距离.其实也很好想为什么，要遍历树上的点最少肯定要每条边都走两次，i点要到这棵树上肯定要走最短距离，我们到一个点后就不回起点了，那么肯定终点在最远的那个点上.

那么想要做出这道题就要用到4个dfs.首先求出每个点在不在树上,然后求出i到树上最近的点和最短距离，接下来求出每个点到它的叶子节点的最长距离和次长距离，最后求出每个点的最长距离.

第一步和第二步很好理解，关键是最后两步，我们求最长距离为什么还要求每个点到它的叶子节点的最长距离和次长距离呢？其实这有点像树链剖分，我们把树链分为最长链和次长链，位于最长链的儿子成为重儿子，重儿子的最远距离点肯定在它的子节点中或者次长链中，这就包含了最长链和次长链两种情况，没有其它情况了，轻儿子的最远距离也肯定在它的子节点或者最长链中，所以我们先求出父节点到子节点的最远距离和次远距离，然后利用父节点更新子节点，很难想到啊.

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int maxn = 500010;

int n, k,head[maxn],to[maxn * 2],nextt[maxn * 2],w[maxn * 2],tot = 1,ner[maxn],son[maxn];
long long mind[maxn],maxd[maxn],secd[maxn],d[maxn],sum;
bool flag[maxn],vis[maxn];

void add(int x, int y, int z)
{
    w[tot] = z;
    to[tot] = y;
    nextt[tot] = head[x];
    head[x] = tot++;
}

bool dfs1(int x)   //找树
{
    vis[x] = 1;
    for (int i = head[x]; i; i = nextt[i])
    {
        int v = to[i];
        if (!vis[v])
        {
            if (dfs1(v))
            {
                flag[x] = 1;
                sum += w[i];
            }
        }
    }
    return flag[x];
}

void dfs2(int x)  //找最短距离
{
    vis[x] = 1;
    if (flag[x])
        ner[x] = x;
    for (int i = head[x]; i; i = nextt[i])
    {
        int v = to[i];
        if (!vis[v] && !flag[v])
        {
            mind[v] = mind[x] + w[i];
            ner[v] = ner[x];
        }
        if (!vis[v])
            dfs2(v);
    }
}

long long dfs3(int x)  //最长距离+次长距离
{
    vis[x] = 1;
    for (int i = head[x]; i; i = nextt[i])
    {
        int v = to[i];
        if (!vis[v] && flag[v])
        {
            long long t = dfs3(v) + w[i];
            if (t > maxd[x])
            {
                secd[x] = maxd[x];
                maxd[x] = t;
                son[x] = v;
            }
            else
                if (t > secd[x])
                    secd[x] = t;
        }
    }
    return maxd[x];
}

void dfs4(int x, long long lmax)   //最后的更新
{
    vis[x] = 1;
    for (int i = head[x]; i; i = nextt[i])
    {
        int v = to[i];
        d[x] = max(lmax, maxd[x]);
        if (flag[v] && !vis[v])
        {
            if (v == son[x])
                dfs4(v, max(lmax, secd[x]) + w[i]);
            else
                dfs4(v, max(lmax, maxd[x]) + w[i]);
        }
    }
}

int main()
{
    memset(flag, false, sizeof(flag));
    scanf("%d%d", &n, &k); 
    for (int i = 1; i < n; i++)
    {
        int x, y, z;
        scanf("%d%d%d", &x, &y, &z);
        add(x, y, z);
        add(y, x, z);
    }
    int t;
    for (int i = 1; i <= k; i++)
    {
        scanf("%d", &t);
        flag[t] = 1;
    }
    dfs1(t);
    memset(vis, 0, sizeof(vis));
    dfs2(t);
    memset(vis, 0, sizeof(vis));
    dfs3(t);
    memset(vis, 0, sizeof(vis));
    dfs4(t, 0);
    for (int i = 1; i <= n; i++)
        printf("%lld\n", sum * 2 + mind[i] - d[ner[i]]);

    return 0;
}

bzoj3743 [Coci2015]Kamp 常州模拟赛d6t2

3743: [Coci2015]Kamp

Description

Input

Output

Sample Input

Sample Output

HINT