网络流二十四题 ————(二十二）、P3356 火星探险问题 费用流经典建模

每个点访问第一次才有贡献，对于点的限制，我们一般把点拆成出点和入点，入边连向入点，出点连向出边。

入点连向出点限制点选的次数。我们从入点向出点连一条流量1，花费1的边表示第一次访问该点可以采集一个石头，再连一条流量inf，花费0的边，表示后面可以再次访问该点，但是不会获得石头。

S连起点，终点连汇点。

跑最大费用流即可。细节较多仔细检查。

#include<bits/stdc++.h> 
using namespace std;
typedef long long ll;
const int N=1000005;
const int M=1000005;
const ll inf=0x7fffffff;
struct MCMF{
	struct EDGE {
		int nxt, to, cost, flow;
	}ee[M << 1];
	int n,s,t;
	int cur[N],head[N],vis[N];
	ll dis[N];
	ll mincost=0,maxflow=0;
	inline void AD(int from, int to, int flow, int cost) {
		ee[++cnt].nxt = head[from];
		ee[cnt].to = to;
		ee[cnt].cost = cost;
		ee[cnt].flow = flow;
		head[from] = cnt;
	}
	int cnt=1;
	inline void add(int u,int v,int flow,int cost){
	    AD(u,v,flow,cost);
	    AD(v,u,0,-cost);
	 //   cout<<u<<"  ->  "<<v<<endl;
	}
	inline int spfa()
	{
	    for(int i=1;i<=n;++i)dis[i]=inf,vis[i]=0;
	    queue<int>q;q.push(s);
	    dis[s]=0;vis[s]=1;
	    while(!q.empty()){
	        int u=q.front();
	        vis[u]=0;q.pop();
	        for(int i=head[u];i;i=ee[i].nxt)
	        {
	            int v=ee[i].to;
	            if(!ee[i].flow)continue;
	            if(ee[i].flow&&dis[v]>dis[u]+ee[i].cost){
	                dis[v]=dis[u]+ee[i].cost;
	                if(!vis[v]){
	                    vis[v]=1;
	                    q.push(v);
	                }
	            }
	        }
	    }
	    return dis[t]!=inf;
	}
	inline ll dfs(int u,ll flow)
	{
	    if(u==t)return flow;
	    ll rest=flow;
	    vis[u]=1;
	    for(int i=cur[u];i&&rest;i=ee[i].nxt){
	        cur[u]=i;
	        int v=ee[i].to;
	        ll w=ee[i].cost,flow=ee[i].flow;
	        if(ee[i].flow&&(dis[v]==dis[u]+w)&&!vis[v]){
	            ll k=dfs(v,min(flow,rest));
	            if(k){
	                mincost+=k*w;
	                ee[i].flow-=k;
	                ee[i^1].flow+=k;
	                rest-=k;
	            }
	        }
	    }
	    vis[u]=0;
	    return flow-rest;
	}
	void gao(){
		while(spfa()){
	        for(int i=1;i<=n;++i)cur[i]=head[i];
	        maxflow+=dfs(s,inf);
	    }
	}
	void init(int nn,int ss,int tt){
		n=nn,s=ss,t=tt;
		maxflow=mincost=0;
		for(int i=0;i<=n;i++)head[i]=0;
	}
}mc;
int s,t,nm,n,m,k,x,y;
void path(int ip,int p,int t){//打印路径 
	while(p!=t){
	//	cout<<p<<" -> ";
		for(int i=mc.head[p];i;i=mc.ee[i].nxt){
			if(i&1)continue;//cout<<p<<"  -   "<<mc.ee[i].to<<"     "<<mc.ee[i].flow<<endl;
			if(mc.ee[i^1].flow){
				mc.ee[i^1].flow--;
				int to=mc.ee[i].to;
			//	cout<<p<<" -> "<<to<<"  =  ";
				if(to>n*m){
					if(to-n*m-p==1)cout<<ip<<" "<<1<<endl;
					else cout<<ip<<" "<<0<<endl;
				}
				p=to;
				break;
			}
		}
	}
}

int id(int x,int y){
	return (x-1)*m+y;
}
int a[110][110];
int main()
{
	ios::sync_with_stdio(false);
  	cin.tie(0);
  	cin>>nm;
  	cin>>m>>n;
  	s=2*n*m+1,t=2*n*m+2;
  	mc.init(2*n*m+2,s,t);
  	for(int i=1;i<=n;i++)
  		for(int j=1;j<=m;j++)
  			cin>>a[i][j];
  	for(int i=1;i<=n;i++)
  		for(int j=1;j<=m;j++){
  			if(a[i][j]==1)continue;
			int pu=id(i,j)+n*m,u=id(i,j),v,pv;
  			if(i+1<=n&&a[i+1][j]!=1){
  				pv=id(i+1,j)+n*m,v=id(i+1,j);//入点和出点 
  				mc.add(u,pv,inf,0);mc.add(pv,v,inf,0);
			}
			if(j+1<=m&&a[i][j+1]!=1){
				pv=id(i,j+1)+n*m,v=id(i,j+1);
				mc.add(u,pv,inf,0);mc.add(pv,v,inf,0);
			}
  				
			if(a[i][j]==2)mc.add(pu,u,1,-1);
		}
	if(a[1][1]==2)mc.add(s,1,1,-1);
	mc.add(s,1,inf,0);
	mc.add(id(n,m),t,nm,0);
	mc.gao();
	for(int i=1;i<=nm;i++){
		path(i,1,id(n,m));
	}
	//cout<<mc.maxflow<<" "<<-mc.mincost<<endl;
	return 0;
}