【BZOJ1061】【NOI2008】志愿者招募

【BZOJ1061】【NOI2008】志愿者招募

题面

BZOJ

题解

我们设每类志愿者分别招募了\(B[i]\)个

那么，我们可以得到一系列的方程

\[\sum_{S[i]\leq x\leq T[i]}B[i]\geq A[x]
\]

一共\(n\)天，所以是\(n\)个方程

因为有不等式让我们很不爽，所以换成等式。

这里，设\(y[i]\)表示第\(i\)天多招募的人数

这样，方程变成了

\[(\sum_{S[i]\leq x\leq T[i]}B[i])-y[i]=A[x]
\]

所以我们得到了一个方程组

\[\begin{cases}0&=0\\\sum_{S[i]\leq 1\leq T[i]}B[i]-y[1]&=A[1]\\ \sum_{S[i]\leq 2\leq T[i]}B[i]-y[2]&=A[2]\\...\\\sum_{s[i]\leq n\leq T[i]}B[i]-y[n]&=A[n]\\0&=0\end{cases}
\]

对于每一类志愿者，他一定出现在连续的一段

如果我们对于相邻的两个方程做差，得到\(n-1\)个方程，

这样每一类志愿者就只会在两个方程中出现过

所以，方程组变成了

\[(\sum_{S[i]=x}B[i]-\sum_{T[i]+1=x}B[i])-y[x]+y[x-1]=A[x]-A[x-1]
\]

这样一个等式很像流量平衡

所以我们考虑如何连边？

我们把\(+\)看成流入，\(-\)看成流出

这样子相当于每天就是一个点

前面两个\(\sum\)的差，相当于从\(S[i]\)流入，向\(T[i]+1\)流出

也就是如果这个点作为一类志愿者的\(S[i]\)那么他就会流入一定的流

如果作为一个志愿者的\(T[i]+1\)，那么它就要流出去，显然是流到对应的\(S[i]\)去

而\(-y[x]+y[x-1]\)相当于从前一天多出来的志愿者可以流入，然后可以向后一天流出

右边如果非零的话，可以看做从源（汇）点流入（出）

如果大于零则证明流入多于流出，则多出来的流流向汇点

如果小于零则证明流出多于流入，则多出来的流从源点流入

这样建图就可以跑费用流了

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 1111
#define inf 1000000000
inline int read()
{
    RG int x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
int n,m,a[MAX];
struct Line{int v,next,w,fy;}e[50000];
int h[MAX],cnt=2;
inline void Add(int u,int v,int w,int fy)
{
	e[cnt]=(Line){v,h[u],w,fy};h[u]=cnt++;
	e[cnt]=(Line){u,h[v],0,-fy};h[v]=cnt++;
}
bool vis[MAX];
int S,T;
ll Cost,dis[MAX];
bool SPFA(int S,int T)
{
	memset(vis,0,sizeof(vis));
	memset(dis,63,sizeof(dis));
	queue<int> Q;Q.push(S);
	dis[S]=0;vis[S]=true;
	while(!Q.empty())
	{
		int u=Q.front();Q.pop();
		for(int i=h[u];i;i=e[i].next)
		{
			if(!e[i^1].w)continue;
			int v=e[i].v;
			if(dis[v]>dis[u]-e[i].fy)
			{
				dis[v]=dis[u]-e[i].fy;
				if(!vis[v])vis[v]=true,Q.push(v);
			}
		}
		vis[u]=false;
	}
	if(dis[T]>=1e18)return false;
	return true;
}
int dfs(int u,int flow)
{
	if(u==T||!flow)return flow;
	int ret=0;vis[u]=true;
	for(int i=h[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(!vis[v]&&e[i].w&&dis[v]==dis[u]-e[i].fy)
		{
			int d=dfs(v,min(flow,e[i].w));
			e[i].w-=d;e[i^1].w+=d;flow-=d;ret+=d;
			if(!flow)break;
		}
	}
	return ret;
}
int main()
{
	n=read();m=read();S=0;T=n+2;
	for(int i=1;i<=n;++i)a[i]=read();
	for(int i=1;i<=n+1;++i)
	{
		if(a[i]-a[i-1]>=0)Add(i,T,a[i]-a[i-1],0);
		else Add(S,i,a[i-1]-a[i],0);
		if(i<=n)Add(i,i+1,inf,0);
	}
	for(int i=1;i<=m;++i)
	{
		int S=read(),T=read(),C=read();
		Add(T+1,S,inf,C);
	}
	while(SPFA(T,S))Cost+=1ll*dis[S]*dfs(S,inf);
	printf("%lld\n",Cost);
	return 0;
}