Codeforces 240E. Road Repairs 最小树形图+输出路径
最小树形图裸题,只是须要记录路径
/* ***********************************************
Author        :CKboss
Created Time  :2015年07月07日 星期二 22时48分41秒
File Name     :CF240E.cpp
************************************************ */
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <string>
#include <cmath>
#include <cstdlib>
#include <vector>
#include <queue>
#include <set>
#include <map>
using namespace std;
const int INF=0x3f3f3f3f;
const int maxn=2001000;
int n,m;
struct Edge
{
	int u,v,cost,id,ru,rv,rcost;
}edge[maxn];
void add_Edge(int id,int u,int v,int c)
{
	edge[id].id=id;
	edge[id].u=edge[id].ru=u;
	edge[id].v=edge[id].rv=v;
	edge[id].cost=edge[id].rcost=c;
}
int pre[maxn],id[maxn],vis[maxn],in[maxn];
//// !!!!
int preid[maxn],useE[maxn];
int eA[maxn],eD[maxn];
int ex;
int zhuliu(int root,int n,int m,Edge edge[])
{
	int ex=m,res=0;
	while(true)
	{
		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;
				//// !!!!
				preid[edge[i].v]=edge[i].id;
			}
		}
		for(int i=0;i<n;i++)
			if(i!=root&&in[i]==INF) return -1;
		int tn=0;
		memset(id,-1,sizeof(id));
		memset(vis,-1,sizeof(vis));
		in[root]=0;
		for(int i=0;i<n;i++)
		{
			res+=in[i];
			int v=i;
			//// !!!!
			if(i!=root) useE[preid[i]]++;
			while(vis[v]!=i&&id[v]==-1&&v!=root)
			{
				vis[v]=i; v=pre[v];
			}
			if(v!=root&&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;i++)
		{
			int 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];
				//// !!!!
				eA[ex]=edge[i].id;
				eD[ex]=preid[v];
				edge[i].id=ex;
				ex++;
			}
		}
		n=tn;
		root=id[root];
	}
	//// !!!!
	for(int i=ex-1;i>=m;i--)
	{
		if(useE[i])
		{
			useE[eA[i]]++; useE[eD[i]]--;
		}
	}
	return res;
}
int main()
{
	freopen("input.txt","r",stdin);
	freopen("output.txt","w",stdout);
    
	scanf("%d%d",&n,&m);
	for(int i=0,a,b,c;i<m;i++)
	{
		scanf("%d%d%d",&a,&b,&c);
		a--; b--;
		add_Edge(i,a,b,c);
	}
	int lens = zhuliu(0,n,m,edge);
	if(lens==0||lens==-1) { printf("%d\n",lens); return 0; }
	printf("%d\n",lens);
	for(int i=0;i<m;i++)
	{
		if(useE[i]&&edge[i].rcost)
			printf("%d ",i+1);
	}
	putchar(10);
    return 0;
}
                    
                
                
            
        
浙公网安备 33010602011771号