hdu 1055 Color a Tree

简介:

Color a Tree


       题意:每个节点有一个权值,从根节点开始着色,每一个被着色的节点,其父亲一定已经被着色了,根据着色顺序得到一个序列,求最小的序列对应的权值的乘积和最小;
       分析:寻找最大权值,合并这个节点和他的父亲节点,记下这两个节点的拓扑序列,同时新节点的权值为这些节点的算术平均值,直到只有一个节点。
 
       证明: http://hi.baidu.com/cheezer94/blog/item/d98eca065202a2f237d122da.html
/*
author:jxy
lang:C/C++
university:China,Xidian University
**If you need to reprint,please indicate the source**
*/
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#define INF 1E9
using namespace std;
double v[1010];
int num[1010],q[1010];
bool cmp(int a,int b)
{
    return v[a]/num[a]>v[b]/num[b];
}
int ff[1010],ss[1010],fa[1010];
int n,r,val[1010];
int farther(int i)
{
    if(ff[i]==i)return i;
    return ff[i]=farther(ff[i]);
}
int son(int i)
{
    if(ss[i]==i)return i;
    return son(ss[i]);
}
bool input()
{
    scanf("%d%d",&n,&r);
    if(n==0&&r==0)return 0;
    int i,f,s;
    r--;
    for(i=0;i<n;i++)
    {
        scanf("%d",&val[i]);
        fa[i]=ff[i]=ss[i]=q[i]=i;
        num[i]=1;v[i]=val[i]*1.0;
    }
    for(i=1;i<n;i++)
    {
        scanf("%d%d",&f,&s);
        fa[s-1]=f-1;
    }
    return 1;
}
void solve()
{
    int ans=0,k=1,i,now,t;
    for(i=0;i<n;i++)//取出比值最大的没被访问的点。
    {
        sort(q+i,q+n,cmp);
        now=q[i];t=son(fa[now]);
        if(now==r)continue;
        ss[t]=now;ff[now]=t;
        t=farther(now);
        v[t]+=v[now];num[t]+=num[now];
    }
    while(n--)
    {
        ans+=val[r]*(k++);
        r=ss[r];
    }
    printf("%d\n",ans);
}
int main()
{
    while(input())solve();
    return 0;
}

目录
相关文章
|
10月前
HDU - 1312 Red and Black(DFS)
There is a rectangular room, covered with square tiles. Each tile is colored either red or black. A man is standing on a black tile. From a tile, he can move to one of four adjacent tiles. But he can’t move on red tiles, he can move only on black tiles. Write a program to count the number of black
57 0
|
11月前
|
C++
【PAT甲级 - C++题解】1135 Is It A Red-Black Tree
【PAT甲级 - C++题解】1135 Is It A Red-Black Tree
69 0
AtCoder Beginner Contest 222 E - Red and Blue Tree(dfs dp)
AtCoder Beginner Contest 222 E - Red and Blue Tree(dfs dp)
91 0
|
机器学习/深度学习 Windows
Codeforces Round #748 (Div. 3) F - Red-Black Number (记忆化搜索)
Codeforces Round #748 (Div. 3) F - Red-Black Number (记忆化搜索)
78 0
CodeForces - 1469B Red and Blue (前缀和)
CodeForces - 1469B Red and Blue (前缀和)
76 0
hdu 1312 Red and Black(BFS)
hdu 1312 Red and Black(BFS)
122 0
hdu 1312 Red and Black
一个人从@点出发,求他所能到达的'.'的数目,'#'不可走,@本身算1个点。 思路:搜索入门题。
123 0
|
机器学习/深度学习
HDOJ/HDU 1556 Color the ball(树状数组)
HDOJ/HDU 1556 Color the ball(树状数组)
82 0