BZOJ3611: [Heoi2014]大工程

最新推荐文章于 2020-03-19 10:34:03 发布

weixin_30703911

最新推荐文章于 2020-03-19 10:34:03 发布

阅读量58

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/huibixiaoxing/p/8551096.html

3611: [Heoi2014]大工程

Time Limit: 60 Sec Memory Limit: 512 MB
Submit: 1965 Solved: 822
[Submit][Status][Discuss]

Description

国家有一个大工程，要给一个非常大的交通网络里建一些新的通道。
我们这个国家位置非常特殊，可以看成是一个单位边权的树，城市位于顶点上。
在 2 个国家 a,b 之间建一条新通道需要的代价为树上 a,b 的最短路径。
现在国家有很多个计划，每个计划都是这样，我们选中了 k 个点，然后在它们两两之间新建 C(k,2)条新通道。
现在对于每个计划，我们想知道：
1.这些新通道的代价和
2.这些新通道中代价最小的是多少
3.这些新通道中代价最大的是多少

Input

第一行 n 表示点数。
接下来 n-1 行，每行两个数 a,b 表示 a 和 b 之间有一条边。
点从 1 开始标号。接下来一行 q 表示计划数。
对每个计划有 2 行，第一行 k 表示这个计划选中了几个点。
第二行用空格隔开的 k 个互不相同的数表示选了哪 k 个点。

Output

输出 q 行，每行三个数分别表示代价和，最小代价，最大代价。

Sample Input

2 1

3 2

4 1

5 2

6 4

7 5

8 6

9 7

10 9

5 4

10 4

5 2

6 1

Sample Output

3 3 3

6 6 6

1 1 1

2 2 2

HINT

n<=1000000

q<=50000并且保证所有k之和<=2*n

Source

鸣谢佚名上传

题解

虚树。

\(sum[i]\)表示子树的边所能产生的权值
\(mi[i]\)表示子树内的点到根的最小权值
\(ma[i]\)表示子树内的点到根的最大权值

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <map>
#include <cmath>
inline long long max(long long a, long long b){return a > b ? a : b;}
inline long long min(long long a, long long b){return a < b ? a : b;}
inline void swap(int &x, int &y){int tmp = x;x = y;y = tmp;}
template <class T>
inline void read(T &x)
{
    x = 0;char ch = getchar(), c = ch;
    while(ch < '0' || ch > '9') c = ch, ch = getchar();
    while(ch <= '9' && ch >= '0') x = x * 10 + ch - '0', ch = getchar();
    if(c == '-') x = -x;
}
const long long INF = 0x3f3f3f3f3f3f3f3f;
const int MAXN = 1000000 + 10;

struct Edge
{
    int v, nxt;
    Edge(int _v, int _nxt){v = _v, nxt = _nxt;}
    Edge(){}
}edge1[MAXN << 1], edge2[MAXN << 1];
int n, q, k, t, size[MAXN], stack[MAXN * 25], top, lk[MAXN * 25], tot, dfn[MAXN], node[MAXN], head1[MAXN], cnt1, head2[MAXN], cnt2, deep[MAXN], p[30][MAXN], M, tag[MAXN];
long long ansmi, ansma, sum[MAXN], mi[MAXN], ma[MAXN];
inline void insert1(int a, int b)
{
    edge1[++ cnt1] = Edge(b, head1[a]), head1[a] = cnt1;
    edge1[++ cnt1] = Edge(a, head1[b]), head1[b] = cnt1;
}
inline void insert2(int a, int b)
{
    edge2[++ cnt2] = Edge(b, head2[a]), head2[a] = cnt2;
    lk[++ tot] = a;
}
void dfs(int x)
{
    dfn[x] = ++ t;
    for(int pos = head1[x];pos;pos = edge1[pos].nxt)
    {
        int v = edge1[pos].v;
        if(v == p[0][x]) continue;
        deep[v] = deep[x] + 1, p[0][v] = x, dfs(v);
    }
}
void yuchuli()
{
    while((1 << M) <= n) ++ M;-- M;
    for(int i = 1;i <= M;++ i)
        for(int j = 1;j <= n;++ j)
            p[i][j] = p[i - 1][p[i - 1][j]];
}
int LCA(int va, int vb)
{
    if(deep[va] < deep[vb]) swap(va, vb);
    for(int i = M;i >= 0;-- i)
        if(deep[va] - deep[vb] >= (1 << i))
            va = p[i][va];
    if(va == vb) return va;
    for(int i = M;i >= 0;-- i)
        if(p[i][va] != p[i][vb])
            va = p[i][va], vb = p[i][vb];
    return p[0][va];
}
bool cmp(int a, int b)
{
    return dfn[a] < dfn[b];
}
void DP(int x)
{
    sum[x] = ma[x] = 0, mi[x] = tag[x] ? 0 : INF, size[x] = tag[x];
    long long tmpmi = INF, tmpma = 0;
    for(int pos = head2[x];pos;pos = edge2[pos].nxt)
    {
        int v = edge2[pos].v;
        DP(v);
        sum[x] += sum[v] + (long long)(deep[v] - deep[x]) * size[v] * (k - size[v]);
        
        long long a = mi[v] + (deep[v] - deep[x]);
        if(mi[x] == a) tmpmi = mi[x];
        else if(mi[x] > a) tmpmi = mi[x], mi[x] = a;
        else tmpmi = min(tmpmi, a);
        
        long long b = ma[v] + (deep[v] - deep[x]);
        if(ma[x] == b) tmpma = b;
        else if(ma[x] < b) tmpma = ma[x], ma[x] = b;
        else tmpma = max(tmpma, b);
        
        size[x] += size[v];
    }
    ansma = max(ansma, tmpma + ma[x]), ansmi = min(ansmi, tmpmi + mi[x]);
}
void build_VT()
{
    tot = 0, top = 0, cnt2 = 0;
    std::sort(node + 1, node + 1 + k, cmp);
    for(int i = 1;i <= k;++ i)
    {
        if(!top)
        {
            stack[++ top] = node[i];
            continue;
        }
        int lca = LCA(stack[top], node[i]);
        while(deep[lca] < deep[stack[top]])
        {
            if(deep[lca] >= deep[stack[top - 1]])
            {
                insert2(lca, stack[top]);
                if(stack[-- top] != lca) stack[++ top] = lca;
                break;
            }
            insert2(stack[top - 1], stack[top]), -- top;
        }
        stack[++ top] = node[i];
    }
    while(top > 1) insert2(stack[top - 1], stack[top]), -- top;
    ansmi = INF, ansma = 0, DP(stack[1]);
    printf("%lld %lld %lld\n", sum[stack[1]], ansmi, ansma);
    for(int i = 1;i <= tot;++ i) head2[lk[i]] = 0;
}
int main()
{
    read(n);
    for(int i = 1;i < n;++ i)
    {
        int tmp1, tmp2;read(tmp1), read(tmp2);
        insert1(tmp1, tmp2);
    }
    read(q);dfs(1);yuchuli();
    for(int i = 1;i <= q;++ i)
    {
        read(k);
        for(int i = 1;i <= k;++ i) read(node[i]), tag[node[i]] = 1;
        build_VT();
        for(int i = 1;i <= k;++ i) tag[node[i]] = 0;
    }
    return 0;
}

转载于:https://www.cnblogs.com/huibixiaoxing/p/8551096.html