Problem K. King of Renovations

题意:给定n(300)个点m条带权边的有向图,要求保留一组权和最小的边使得每个点既有入度又有出度。
思路:
每条边沟通一个点的入度和另一个点的出度,因此可以构建一个二分图,左列为入度,右列为出度。问题转化为:带边权二分图的最小边覆盖。
如果没有权,那就是点数-最大匹配。
有权怎么办呢?没办法采取这样的减法,正解是:s连到每一个入点、每一个出点连到t两条边,一条容量为1费用为-INF,另一条容量为INF费用为0,左右两列连边容量为1费用为k。跑一个不固定流量的最小费用流,当增广一条路径的费用非负时退出,答案就是 c o s t + 2 ∗ I N F cost+2*INF cost+2INF,如果这个值大于INF,说明没有覆盖到所有的点,输出NIE,否则直接输出。

代码:

#include
using namespace std;
const int maxn=300*2+100;
const int INF=0x3f3f3f3f;
typedef long long ll;

struct Edge{
	int from,to,cap,flow,cost;
};

struct MCMF{
	int n,m,s,t;
	vector<Edge> edges;
	vector<int> G[maxn];
	int inq[maxn];
	ll d[maxn];
	int p[maxn];
	int a[maxn];

	void init()
	{
		int m0,x,y,k;
		cin>>n>>m0;
		s=0,t=1;
		for(int i=1;i<=n;i++)AddEdge(s,i*2,1,-INF),AddEdge(s,i*2,INF,0),AddEdge(i*2+1,t,1,-INF),AddEdge(i*2+1,t,INF,0);
		for(int i=1;i<=m0;i++)
		{
			scanf("%d%d%d",&x,&y,&k);
			AddEdge(y*2,x*2+1,1,k);
		}
	}

	void AddEdge(int from,int to,int cap,int cost)
	{
		edges.push_back((Edge){from,to,cap,0,cost});
		edges.push_back((Edge){to,from,0,0,-cost});
		m=edges.size();
		G[from].push_back(m-2);
		G[to].push_back(m-1);
	}

	bool BellmanFord(ll& flow,ll& cost)
	{
		for(int i=0;i<=2*n+1;i++)d[i]=(1LL<<50);/***切记这里要改***/
		memset(inq,0,sizeof(inq));
		d[s]=0;inq[s]=1;p[s]=0;a[s]=INF;

		queue<int> Q;
		Q.push(s);
		while(!Q.empty())
		{
			int u=Q.front();Q.pop();
			inq[u]=0;
			for(int i=0;i<G[u].size();i++)
			{
				Edge& e=edges[G[u][i]];
				if(e.cap>e.flow && d[e.to]>d[u]+e.cost)
				{
					d[e.to]=d[u]+e.cost;
					p[e.to]=G[u][i];
					a[e.to]=min(a[u],e.cap-e.flow);
					if(inq[e.to]==0){inq[e.to]=1;Q.push(e.to);}
				}
			}
		}
		if(d[t]==(1LL<<50))return false;
		flow+=a[t];
		if(d[t]>=0)return 0;
		cost+=a[t]*d[t];
		int u=t;
		while(u!=s)
		{
			edges[p[u]].flow+=a[t];
			edges[p[u]^1].flow-=a[t];
			u=edges[p[u]].from;
		}
		return true;
	}

	ll Mincost()
	{
		ll flow=0,cost=0;
		while(BellmanFord(flow,cost));
		return cost;
	}
}ans;

int main()
{
	//freopen("input.in","r",stdin);
	ans.init();
	ll sum=ans.Mincost();
	sum+=2LL*ans.n*INF;
	if(sum>INF)cout<<"NIE"<<endl;
	else cout<<sum<<endl;
	return 0;
}

你可能感兴趣的:(网络流,网络流)