BZOJ 1061 [Noi2008]志愿者招募（费用流）

题目链接：https://www.lydsy.com/JudgeOnline/problem.php?id=1061

 

每个时间像下一个时间连一条容量为INF－这个时间需要的人，费用为０。
Ｓ向1连容量为INF，费用为０。
n向Ｔ连容量为INF，费用为０。
然后对于每个人，时间段为x到y,费用为ｚ，从x到y+1连一条容量为inf,费用为ｚ的边。
然后跑最小费用最大流。

 

代码：

#include<bits/stdc++.h>
#define xx first
#define yy second
#define mp make_pair
#define pb push_back
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int MAXN=10000;
const int MAXM=100000;
const int INF=0x3f3f3f3f;
namespace MCMF
{
	struct Edge
	{
		int to,next,cap,flow,cost;
	}edge[MAXM];
	int head[MAXN],tol;
	int pre[MAXN];
	ll dis[MAXN];
	bool vis[MAXN];
	int N;//节点总个数，节点编号从0~N-1
	void init(int n)
	{
		N=n;
		tol=0;
		memset(head,-1,sizeof(head));
	}
	void addedge(int u,int v,int cap,int cost)
	{
		edge[tol].to=v;edge[tol].cap=cap;edge[tol].cost=cost;edge[tol].flow=0;edge[tol].next=head[u];
		head[u]=tol++;
		edge[tol].to=u;edge[tol].cap=0;edge[tol].cost=-cost;edge[tol].flow=0;edge[tol].next=head[v];
		head[v]=tol++;
	}
	bool spfa(int s,int t) 
	{ 
		queue<int>q;
		for(int i=0;i<N;i++)
		{
			dis[i]=INF;
			vis[i]=false;
			pre[i]=-1;
		}
		dis[s]=0;
		vis[s]=true;
		q.push(s);
		while(!q.empty())
		{
			int u=q.front();
			q.pop();
			vis[u]=false;
			for(int i=head[u];i!=-1;i=edge[i].next)
			{
				int v=edge[i].to;
				if(edge[i].cap>edge[i].flow&&dis[v]>dis[u]+edge[i].cost)
				{
					dis[v]=dis[u]+edge[i].cost;
					pre[v]=i;
					if(!vis[v])
					{
						vis[v]=true;
						q.push(v);
					}
				}
			}
		}
		if(pre[t]==-1)
			return false;
		else
			return true;
	}
	//返回的是最大流，cost存的是最小费用 
	int mcmf(int s,int t,ll &cost)
	{
		int flow=0;
		cost=0;
		while(spfa(s,t))
		{
			int Min=INF;
			for(int i=pre[t];i!=-1;i=pre[edge[i^1].to])
			{
				if(Min>edge[i].cap-edge[i].flow)
					Min=edge[i].cap-edge[i].flow;
			}
			for(int i=pre[t];i!=-1;i=pre[edge[i^1].to])
			{
				edge[i].flow+=Min;
				edge[i^1].flow-=Min;
				cost+=edge[i].cost*Min;
			}
			flow+=Min;
		}
		return flow;
	}
}
using namespace MCMF;
int n,m;
int x[MAXN];
void solve()
{
	init(n+2);
	for(int i=1;i<=n;i++)
		scanf("%d",&x[i]);
	for(int i=1;i<=m;i++)
	{
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		addedge(u,v+1,INF,w);
	}
	addedge(0,1,INF,0);
	for(int i=1;i<=n;i++)
	{
		addedge(i,i+1,INF-x[i],0);
	}
	ll cost=0;
	mcmf(0,n+1,cost);
	printf("%lld\n",cost);
}
int main()
{
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	while(~scanf("%d%d",&n,&m))
	{
		solve();
	}
	return 0;
}