【最小树形图 - 朱刘算法模板】HDU - 2121 - Ice_cream’s world II

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

---

题意

在一张有向图上，选择一个首都，然后选择若干条边使得首都能去其他的城市。问代价最小是多少

---

题解

最小树形图模板题，虚拟一个点连向所有的点，跑一边朱刘算法，虚拟点连的就是首都

---

#include<iostream>
#include<cstdio>
#include<string.h>
#include<string>
#include<algorithm>
#include<stack>
using namespace std;
typedef long long ll;
const ll N=2e4+7;
const ll inf=1e18;
ll in[N],pre[N],vis[N],idx[N],pos;
struct Edge{
    ll u,v,c;
}e[N];
ll zhuliu(ll rt,ll totn,ll totm){
    ll ans=0;
    while(true){
        for(ll i=0;i<totn;i++)in[i]=inf;
        for(ll i=0;i<totm;i++){
            ll u=e[i].u,v=e[i].v,c=e[i].c;
            if(u!=v&&c<in[v]) {
                in[v]=c;
                pre[v]=u;
                if(u==rt) pos=i;
            }
        }
        for(ll i=0;i<totn;i++){
            if(in[i]==inf&&i!=rt)return -1;
        }
        ll tot=0;
        for(ll i=0;i<totn;i++) vis[i]=idx[i]=-1;
        in[rt]=0;
        for(ll i=0;i<totn;i++){
            ans+=in[i];
            ll y=i;
            while(vis[y]!=i&&idx[y]==-1&&y!=rt){
                vis[y]=i;
                y=pre[y];
            }
            if(y!=rt&&idx[y]==-1){
                for(ll x=pre[y];x!=y;x=pre[x]) idx[x]=tot;
                idx[y]=tot++;
            }
        }
        if(!tot) break;
        for(ll i=0;i<totn;i++) if(idx[i]==-1) idx[i]=tot++;
        for(ll i=0;i<totm;i++){
            ll u=e[i].u,v=e[i].v;
            e[i].u=idx[u];
            e[i].v=idx[v];
            if(e[i].u!=e[i].v) e[i].c-=in[v];
        }
        totn=tot;
        rt=idx[rt];
    }
    return ans;
}

int main(){
    ll n,m;
    while(scanf("%lld%lld",&n,&m)!=EOF){
        ll sum=1;
        for(ll i=0;i<m;i++){
            scanf("%lld%lld%lld",&e[i].u,&e[i].v,&e[i].c);
            sum+=e[i].c;
        }
        for(ll i=m,j=0;j<n;i++,j++){
            e[i].u=n,e[i].v=j,e[i].c=sum;
        }
        ll ans=zhuliu(n,n+1,n+m);
        if(ans>=2*sum||ans==-1) printf("impossible\n\n");
        else printf("%lld %lld\n\n",ans-sum,pos-m);
    }
}