POJ 2054 贪心

题意

传送门 POJ 2054 Color a Tree

题解

本质是在树上遍历顺序的限制下，找到一个染色顺序 $chs[i]$，使数列 { 1 , 2 , … , n } \{1,2,\dots,n\} {1,2,…,n} 与 { C 1 , C 2 , … , C n } \{C_1,C_2,\dots,C_n\} {C1​,C2​,…,Cn​} 在对应位置相乘，即 $\sum _{i=1}^{n}i\times C_{chs[i]}$ 取到最小值。假设没有限制，显然将 $C_i$ 降序排序满足条件。考虑简单的贪心策略：用堆维护当前未染色节点，每次染色以该节点为根节点的子树权值最大的节点；很容易找到反例，因为当前染色顺序会影响到后续能够染色的节点，这样的贪心并不能保证全局最优。

考虑未染色的 m a x { C i } max\{C_i\} max{Ci​} 节点父节点已经染色，此时选取 m a x { C i } max\{C_i\} max{Ci​} 的节点一定比不会比其它染色顺序更差，考虑数列即可，假如 m a x { C i } max\{C_i\} max{Ci​} 后移 $1$ 个位置，那么必有一个 C j ( C j ≤ m a x { C i } ) C_j(C_j\leq max\{C_i\}) Cj​(Cj​≤max{Ci​}) 左移一个位置，对总权和的影响为 m a x { C i } − C j ≥ 0 max\{C_i\}-C_j\geq 0 max{Ci​}−Cj​≥0。则得到结论：树中除根节点以外权值最大的节点一定在其父节点染色后立即被染色。

考虑 $3$ 个节点 $x,y,z$，假设前 $2$ 个节点为连续染色的节点，则染色顺序得到的权和有 $2$ 种可能，即 $x+2y+3z,z+2x+3y$。考虑右项大于左项，则有 $z<\frac{x+y}{2}$，可以等效为 $2$ 个节点染色；推广到更多的节点合并时，使用同样的推导过程，可以得到等效权值为：合并节点的 $C_i$ 求和除以合并节点的个数。

那么考虑非根节点，每次选取等效权值最大的节点与父节点合并，同时保存染色顺序。当合并到只剩一个节点（即根节点）时，按照保存的顺序进行染色同时计算答案即可。

#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
#define maxn 1005
struct node
{
    int p, v, vs[maxn], c, ch[maxn];
    double w;
} tree[maxn];
int N, R, C[maxn];
bool del[maxn];

int main()
{
    while (~scanf("%d%d", &N, &R) && (N | R))
    {
        for (int i = 1; i <= N; ++i)
        {
            scanf("%d", C + i);
            tree[i].v = 1, tree[i].vs[1] = i;
            tree[i].p = tree[i].c = 0;
            tree[i].w = C[i];
        }
        for (int i = 2; i <= N; ++i)
        {
            int u, v;
            scanf("%d%d", &u, &v);
            tree[v].p = u;
            tree[u].ch[++tree[u].c] = v;
        }
        memset(del, 0, sizeof(del));
        for (int i = 2; i <= N; ++i)
        {
            int v = -1;
            for (int i = 1; i <= N; ++i)
            {
                if (i != R && !del[i] && (v == -1 || tree[v].w < tree[i].w))
                    v = i;
            }
            del[v] = 1;
            node &cur = tree[v], &p = tree[cur.p];
            memcpy(p.vs + p.v + 1, cur.vs + 1, sizeof(int) * cur.v);
            p.w = (p.w * p.v + cur.w * cur.v) / (p.v + cur.v);
            p.v += cur.v;
            for (int i = 1; i <= cur.c; ++i)
            {
                p.ch[++p.c] = cur.ch[i];
                tree[cur.ch[i]].p = cur.p;
            }
        }
        int res = 0;
        for (int i = 1, t = 1; i <= N; ++i, ++t)
            res += C[tree[R].vs[i]] * t;
        printf("%d\n", res);
    }
    return 0;
}