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

题目：http://acm.hdu.edu.cn/showproblem.php?pid=2121

题意：给定n个点和m条有向边，边有权值，选择一个点，从这个点可以到达其他所有点且权值最小，若这样的点有多个，选择编号最小的那一个

思路：不定根最小树形图，因为没有给定根，所以虚拟一个根，从虚拟根向所有点连虚拟边，权值设为一个可以和其他边区分的数字，设比其他所有边权值和大1。然后求最小树形图，如果求出的最小树形图中含有2条及以上的虚拟边，则原图肯定没有最小树形图。至于最小树形图存在时如何求编号最小的点，可以知道所有实际根一定在一个环中，当一条边的起点是虚拟根时，那么终点一定是实际的根，在从虚拟根向其他点连边时按照从小到大的顺序，其中 边的编号 - m = 点的编号，更新in数组时也按从小到大的顺序，这样记录下的边一定是编号最小的实际根所在的边，因为算法过程中点的编号会变，所以要用边。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <queue>
#include <map>
using namespace std;

typedef long long ll;
const int N = 1010;
const ll INF = 0x3f3f3f3f3f3f3f3f;
struct edge
{
    int v, u;
    ll cost;
} g[N*20];
int n, m;
int id[N], pre[N], vis[N], ans;
ll in[N];
ll ma(int s, int n, int m)
{
    ll res = 0;
    while(true)
    {
        for(int i = 0; i < n; i++) in[i] = INF;
        for(int i = 0; i < m; i++)
        {
            int v = g[i].v, u = g[i].u;
            if(v != u && g[i].cost < in[u])
            {
                in[u] = g[i].cost, pre[u] = v;
                if(v == s) ans = i;//记录边的编号
            }
        }
        for(int i = 0; i < n; i++)
        {
            if(i == s) continue;
            if(in[i] == INF) return -1;
        }
        int num = 0;
        memset(id, -1, sizeof id);
        memset(vis, -1, sizeof vis);
        in[s] = 0;
        for(int i = 0; i < n; i++)
        {
            res += in[i];
            int v = i;
            while(vis[v] != i && id[v] == -1 && v != s) vis[v] = i, v = pre[v];
            if(v != s && id[v] == -1)
            {
                for(int j = pre[v]; j != v; j = pre[j]) id[j] = num;
                id[v] = num++;
            }
        }
        if(num == 0) break;
        for(int i = 0; i < n; i++)
            if(id[i] == -1) id[i] = num++;
        for(int i = 0; i < m; i++)
        {
            int v = g[i].v, u = g[i].u;
            g[i].v = id[v], g[i].u = id[u];
            if(id[v] != id[u]) g[i].cost -= in[u];
        }
        n = num, s = id[s];
    }
    return res;
}
int main()
{
    while(~ scanf("%d%d", &n, &m))
    {
        ll sum = 0;
        for(int i = 0; i < m; i++)
            scanf("%d%d%lld", &g[i].v, &g[i].u, &g[i].cost), g[i].v++, g[i].u++, sum += g[i].cost;
        sum++;
        for(int i = m; i < n + m; i++)
            g[i].v = 0, g[i].u = i - m + 1, g[i].cost = sum;
        ll res = ma(0, n + 1, n + m);
        if(res == -1 || res / sum >= 2) printf("impossible\n\n");
        else printf("%lld %d\n\n", res - sum, ans - m);
    }
    return 0;
}