HDU 2121 Ice_cream’s world II(最小树形图+不定根)

思路:先制造一个虚拟结点:它向所有结点连一条边,权值要大于原图中所有权值的和,以它为根跑最小树形图,如果ans >= 2 * sum,那么说明从虚拟结点连出去两条以上的边,即原图不连通,反之原图必联通,且权值和为ans - sum。根节点就是树形图上虚拟节点的边指向的那个结点

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 1e3 + 5;
const int M = 2e4;
const long long Inf = 1e12;
long long d[N];
int ID[N], vis[N], pre[N], pos;
struct edge {
    int u, v;
    long long cost;
} es[M];
long long MST(int root, int V, int E) {
    long long res = 0;
    while(true) {
        for(int i = 0; i < V; i += 1) d[i] = Inf;
        for(int i = 0; i < E; i += 1) {
            int u = es[i].u;
            int v = es[i].v;
            if(d[v] > es[i].cost && u != v) {
                d[v] = es[i].cost;
                pre[v] = u;
                if(u == root) pos = i;
            }
        }
        for(int i = 0; i < V; i += 1) {
            if(i == root) continue;
            if(d[i] == Inf) return -1;
        }
        int c;
        d[root] = c = 0;
        memset(vis,-1,sizeof(vis));
        memset(ID,-1,sizeof(ID));
        for(int i = 0; i < V; i += 1) {
            int v = i;
            res += d[i];
            while(ID[v] == -1 && vis[v] != i && v != root) {
                vis[v] = i;
                v = pre[v];
            }
            if(ID[v] == -1 && v != root) {
                for(int u = pre[v]; u != v; u = pre[u]) ID[u] = c;
                ID[v] = c;
                c += 1;
            }
        }
        if(!c)
            break;
        for(int i = 0; i < V; i += 1) {
            if(ID[i] == -1) { ID[i] = c; c += 1; }
        }
        for(int i = 0; i < E; i += 1) {
            int u = es[i].u;
            int v = es[i].v;
            es[i].u = ID[u];
            es[i].v = ID[v];
            if(ID[u] != ID[v]) es[i].cost -= d[v];
        }
        V = c;
        root = ID[root];
    }
    return res;
}
int main() {
    int V, E, x, y;
    long long ci;
    while(~scanf("%d%d", &V, &E)) {
        long long all = 0;
        for(int i = 0; i < E; i += 1) {
            scanf("%d%d%lld", &x, &y, &ci);
            if(x == y) {
                --i;
                --E;
                continue;
            }
            es[i] = edge{x, y, ci};
            all += ci;
        }
        all += 1;
        for(int i = E; i < E + V; i += 1) es[i] = edge{V, i - E, all};
        long long res = MST(V, V + 1, V + E);
        if(res >= 2 * all)
            printf("impossible\n\n");
        else
            printf("%lld %d\n\n", res - all, pos - E);
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值