洛谷 P4012 深海机器人问题【最大费用最大流】

和火星那个有点像，但是这个价值直接在路径上，不用拆点，对于每条价值为w的边(i,j)，连接(i,j,1,w)(i,j,inf,0)，表示价值只能取一次，然后连接源点和所有出发点(s,i,k,0)，所有终点和t(i,t,r,0)，然后跑最大费用最大流即可。

#include<iostream>

#include<cstdio>

#include<queue>

using namespace std;

const int N=1000005,inf=1e9;

int a,b,n,m,h[N],cnt=1,dis[N],fr[N],id[20][20],tot,ans,s,t;

bool v[N];

struct qwe

{

	int ne,no,to,va,c;

}e[N<<2];

int read()

{

	int r=0,f=1;

	char p=getchar();

	while(p>'9'||p<'0')

	{

		if(p=='-')

			f=-1;

		p=getchar();

	}

	while(p>='0'&&p<='9')

	{

		r=r*10+p-48;

		p=getchar();

	}

	return r*f;

}

void add(int u,int v,int w,int c)

{

	cnt++;

	e[cnt].ne=h[u];

	e[cnt].no=u;

	e[cnt].to=v;

	e[cnt].va=w;

	e[cnt].c=c;

	h[u]=cnt;

}

void ins(int u,int v,int w,int c)

{//cout<<u<<" "<<v<<" "<<w<<endl;

	add(u,v,w,c);

	add(v,u,0,-c);

}

bool spfa()

{

	queue<int>q;

	for(int i=s;i<=t;i++)

		dis[i]=-inf;

	dis[s]=0;

	v[s]=1;

	q.push(s);

	while(!q.empty())

	{

		int u=q.front();

		q.pop();

		v[u]=0;

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

			if(e[i].va>0&&dis[e[i].to]<dis[u]+e[i].c)

			{

				dis[e[i].to]=dis[u]+e[i].c;

				fr[e[i].to]=i;

				if(!v[e[i].to])

				{

					v[e[i].to]=1;

					q.push(e[i].to);

				}

			}

	}

	return dis[t]!=-inf;

}

void mcf()

{//cout<<"OK"<<endl;

	int x=inf;

	for(int i=fr[t];i;i=fr[e[i].no])

		x=min(x,e[i].va);

	for(int i=fr[t];i;i=fr[e[i].no])

	{

		e[i].va-=x;

		e[i^1].va+=x;

		ans+=x*e[i].c;

	}

}

int main()

{

	a=read(),b=read(),n=read()+1,m=read()+1;

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

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

			id[i][j]=++tot;

	s=0,t=tot+1;

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

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

		{

			int x=read();

			ins(id[i][j],id[i][j+1],1,x);

			ins(id[i][j],id[i][j+1],inf,0);

		}

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

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

		{

			int x=read();

			ins(id[i][j],id[i+1][j],1,x);

			ins(id[i][j],id[i+1][j],inf,0);

		}

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

	{

		int k=read(),x=read()+1,y=read()+1;

		ins(s,id[x][y],k,0);

	}

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

	{

		int r=read(),x=read()+1,y=read()+1;

		ins(id[x][y],t,r,0);

	}//cout<<"ok1"<<endl;

	while(spfa())

		mcf();//cout<<"ok2"<<endl;

	printf("%d\n",ans);

	return 0;

}

巴特西

洛谷 P4012 深海机器人问题【最大费用最大流】

最新文章

热门文章

巴特西

洛谷 P4012 深海机器人问题 【最大费用最大流】

最新文章

热门文章

洛谷 P4012 深海机器人问题【最大费用最大流】