115. 给树染色

最新推荐文章于 2025-07-14 13:12:17 发布

原创最新推荐文章于 2025-07-14 13:12:17 发布 · 468 阅读

0 ·

CC 4.0 BY-SA版权

greedy 专栏收录该内容

3 篇文章

订阅专栏

博主边看视频边敲代码，感觉自己还不具备单独完成提高组+题目的能力。文中以一个贪心（greedy）题为例，通过假设不同节点选择顺序计算成本，得出结果取决于节点平均值，做法是找出非根节点最大节点并入父节点，还给出了代码实现步骤。

继续看yan老哥的视频还是一边看yan老哥视频一边敲代码总感觉自己现在还不具备单独过提高组+题目的能力写代码5分钟调试半小时不是这错就是那错；这是个greedy题而且要找出一个顺序；那么我们假设子节点b其父节点a

以及一个八杆子打不着的节点c ；如果先选b，a 那么 cost为 a+2b+3c;

如果先选c 则为 c+2a+3b；结果是看c 和（a+b）/2的关系

那么推广到一般情况结果就是看那一坨节点的平均值；所以我们不停的找出非根节点的最大节点；然后将其并入父节点，最终根节点就是答案；

做法；

1 begin：f[i]=i; 最初各自为一组

2 接下来每次会将两组点合并，将其中一组点接在另一组点的后面

代码

#include <algorithm>
#include <iostream>
#include <sstream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <string>
#include <bitset>
#include <vector>
#include <queue>
#include <stack>
#include <cmath>
#include <list>
#include <map>
#include <set>
#include<iomanip>
#include <bits/stdc++.h>
/***************************************/
#define  ll  long long
#define int64 __int64
#define PI 3.1415927
#define INF 99999
#define MAXN 1000010
using namespace std;

 struct node
 {
     int father, size ,sum;//父节点     子树个数  权值和
     double avg;
 }nodes[1010];     int n,root;

int find()//求当前这个点权值的非根节点
{
 double avg=0;  int ans=-1;


 for(int i=1;i<=n;i++)
 {
       if(i!=root&&nodes[i].avg>avg)//根节点无法往上和并
       {
          avg=nodes[i].avg;
          ans=i;
       }
 }

    return ans;
}

  int main()

{

       cin>>n>>root; int res=0;
       for(int i=1;i<=n;i++)
       {    cin>>nodes[i].sum;
           nodes[i].avg = nodes[i].sum;
        nodes[i].size = 1;
        res += nodes[i].sum;
       }
         int a,b;
         for(int i=0;i<n-1;i++)
         {
             cin>>a>>b;
nodes[b].father=a ;

         }
         for(int i=0;i<n-1;i++)
         {
              int ver=find();
              int father_baba =nodes[ver ].father;
              res+= nodes[ver].sum*nodes[father_baba].size;  //更新答案
              nodes [ver].avg=-1;//已经被遍历
             for(int  j=1 ;j<=n;j++)//合并   删边 连结点
             {
                  if(nodes[j].father== ver)
                  {
                  nodes[j].father =father_baba ;

             } 
             }
 nodes[father_baba].sum+=nodes[ver].sum;
             nodes[father_baba].size+=nodes[ver].size;
               nodes[father_baba].avg=(double)nodes[father_baba].sum/nodes[father_baba].size ;

         }  cout<<res<<endl;

    return 0;
}