弱校联盟con#8Modules (最小树形图)

蒜头有一块主板,为了提升其性能,可在主板上安置若干增强模块。蒜头有
n个不同的增强模块,增强模块可以直接安置在主板上,也可以安置在已经直接或间接连接在主板上的其他增强模块上。
每个增强模块具有一个初始强化值,其中第
i个模块的初始强化值为
Pi
在所有模块安置完成后,每个模块的最终强化值为其自身初始强化值及直接安置在其上的所有模块的最终强化值之和。

tip:
安置完所有模块形成以主板为根的树,即为一个非常耿直的最大树形图问题。
设置跟为0,因为一个块只能选择一个块放到上面,所以就是一棵树。。。要求边权和最大,就用一个很大的数-加成跑最小生成树

最小树形图:

需要有根,需要有向
模板:

/root 根节点,n点个数,m边条数,edge边数组
LL zhuliu(int root,int n,int m,Edge edge[])
{
    LL res = 0;
    int u,v;
    while(1)
    {
        for(int i = 0;i < n;i++)
            in[i] = INF;
        for(int i = 0;i < m;i++)
            if(/*edge[i].u != edge[i].v && */edge[i].cost < in[edge[i].v])
            {
                pre[edge[i].v] = edge[i].u;
                in[edge[i].v] = edge[i].cost;
            }
        for(int i = 0;i < n;i++)
            if(i != root && in[i] == INF)
                return -1.0;//不存在最小树形图
        int tn = 0;
        memset(id,-1,sizeof(id));
        memset(visit,-1,sizeof(visit));
        in[root] = 0;
        for(int i = 0;i < n;i++)
        {
            res += in[i];
            v = i;
            while( /*visit[v] != i*/ visit[v] == -1 /*&& id[v] == -1 */ && v != root)
            {
                visit[v] = i;
                v = pre[v];
            }
            if( v != root && visit[v] == i /*id[v] == -1*/)
            {
                for(int u = pre[v]; u != v ;u = pre[u])
                    id[u] = tn;
                id[v] = tn++;
            }
        }
        if(tn == 0)break;//没有有向环
        for(int i = 0;i < n;i++)
            if(id[i] == -1)
                id[i] = tn++;
        for(int i = 0;i < m;)
        {
            v = edge[i].v;
            edge[i].u = id[edge[i].u];
            edge[i].v = id[edge[i].v];
            if(edge[i].u != edge[i].v)
                edge[i++].cost -= in[v];
            else
                swap(edge[i],edge[--m]);
        }
        n = tn;
        root = id[root];
    }
    return res;
}
/*
安置完所有模块形成以主板为根的树,即为一个非常耿直的最大树形图问题。
设置跟为0,因为一个块只能选择一个块放到上面,所以就是一棵树。。。要求边权和最大,就用一个很大的数-加成跑最小生成树
*/

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;

typedef long long LL;

const LL INF = 10000000000000LL;
const int MAXN = 3010;
const int MAXM = 6010;
struct Edge{
    int u,v;
    LL cost;
};
Edge edge[MAXM];
int pre[MAXN],id[MAXN],visit[MAXN];
LL in[MAXN];

//root 根节点,n点个数,m边条数,edge边数组
LL zhuliu(int root,int n,int m,Edge edge[])
{
    LL res = 0;
    int u,v;
    while(1)
    {
        for(int i = 0;i < n;i++)
            in[i] = INF;
        for(int i = 0;i < m;i++)
            if(/*edge[i].u != edge[i].v && */edge[i].cost < in[edge[i].v])
            {
                pre[edge[i].v] = edge[i].u;
                in[edge[i].v] = edge[i].cost;
            }
        for(int i = 0;i < n;i++)
            if(i != root && in[i] == INF)
                return -1.0;//不存在最小树形图
        int tn = 0;
        memset(id,-1,sizeof(id));
        memset(visit,-1,sizeof(visit));
        in[root] = 0;
        for(int i = 0;i < n;i++)
        {
            res += in[i];
            v = i;
            while( /*visit[v] != i*/ visit[v] == -1 /*&& id[v] == -1 */ && v != root)
            {
                visit[v] = i;
                v = pre[v];
            }
            if( v != root && visit[v] == i /*id[v] == -1*/)
            {
                for(int u = pre[v]; u != v ;u = pre[u])
                    id[u] = tn;
                id[v] = tn++;
            }
        }
        if(tn == 0)break;//没有有向环
        for(int i = 0;i < n;i++)
            if(id[i] == -1)
                id[i] = tn++;
        for(int i = 0;i < m;)
        {
            v = edge[i].v;
            edge[i].u = id[edge[i].u];
            edge[i].v = id[edge[i].v];
            if(edge[i].u != edge[i].v)
                edge[i++].cost -= in[v];
            else
                swap(edge[i],edge[--m]);
        }
        n = tn;
        root = id[root];
    }
    return res;
}

void init(){
    cin>>n>>m;
    LL sum=0;
    for(int i = 0;i < n;i++){
        LL x; cin>>x; sum+=x;
    }
    int L=0;
    while(m--){
        int u,v; LL w;
        cin>>u>>v>>w;
        edge[L].u=v;
        edge[L].v=u;
        edge[L++].cost=1000000001LL-w;
    }
    for(int i = 0;i < n;i++){
        edge[L].u=0;
        edge[L].v=i+1;
        edge[L++].cost=1000000001LL;
    }
        //for(int i=0;i<L;i++) cout<<edge[i].u<<" -> "<<edge[i].v<<" "<<edge[i].cost<<"\n";
    LL ans = zhuliu(0,n+1,L,edge);
    cout<<1000000001LL*n-ans+sum<<"\n";
}

//最小树形图
int n,m,t;
int main(){
    cin>>t;
    while(t--){
        init();
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值