【BZOJ2245】[SDOI2011]工作安排（费用流）

时间：2023-03-08 21:32:16

【BZOJ2245】[SDOI2011]工作安排（费用流）

题面

BZOJ

洛谷

题解

裸的费用流吧。

不需要拆点，只需要连边就好了，保证了\(W_j<W_{j+1}\)。

#include<iostream>

#include<cstdio>

#include<cstring>

#include<queue>

using namespace std;

#define ll long long

#define MAX 600

#define inf 1000000000

inline int read()

{

	int x=0;bool t=false;char ch=getchar();

	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();

	if(ch=='-')t=true,ch=getchar();

	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();

	return t?-x:x;

}

struct Line{int v,next,w,fy;}e[1000000];

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++;

}

int S,T,n,m;

ll Cost,Flow;

ll dis[MAX];bool vis[MAX];

int pe[MAX],pv[MAX];

bool SPFA()

{

	memset(dis,63,sizeof(dis));dis[S]=0;

	queue<int> Q;Q.push(S);vis[S]=true;

	while(!Q.empty())

	{

		int u=Q.front();Q.pop();

		for(int i=h[u];i;i=e[i].next)

		{

			int v=e[i].v;

			if(e[i].w&&dis[v]>dis[u]+e[i].fy)

			{

				pe[v]=i;pv[v]=u;

				dis[v]=dis[u]+e[i].fy;

				if(!vis[v])vis[v]=true,Q.push(v);

			}

		}

		vis[u]=false;

	}

	if(dis[T]>1e9)return false;

	int flow=inf;

	for(int i=T;i!=S;i=pv[i])flow=min(flow,e[pe[i]].w);

	for(int i=T;i!=S;i=pv[i])e[pe[i]].w-=flow,e[pe[i]^1].w+=flow;

	Flow+=flow;Cost+=dis[T]*flow;

	return true;

}

int TT[MAX],WW[MAX],C[MAX];

int main()

{

	m=read();n=read();S=0;T=m+n+1;

	for(int i=1;i<=n;++i)Add(m+i,T,C[i]=read(),0);

	for(int i=1;i<=m;++i)

		for(int j=1;j<=n;++j)

			if(read())Add(i,m+j,inf,0);

	for(int i=1;i<=m;++i)

	{

		int D=read(),lt=0;

		for(int j=1;j<=D;++j)TT[j]=read();

		for(int j=1;j<=D+1;++j)WW[j]=read();

		for(int j=1;j<=D;++j)Add(S,i,TT[j]-lt,WW[j]),lt=TT[j];

		Add(S,i,inf,WW[D+1]);

	}

	while(SPFA());

	printf("%lld\n",Cost);

	return 0;

}

相关文章