Going Home
node
本身寫的第一道費用流,圖建好一波板子AC。不過仍是有幾個地方有點迷。spa
先來看看思想:題意是平面圖上每一個點有若干我的和若干個房子,保證人的數量等於房子的數量。每一個格子上能夠同時容納全部人,也容許一我的在一個房子的格子上但不進入房子。求全部人都找到一個房子住所走的距離的和的最小值。code
典型的最小費用最大流問題,用0點做爲源點,給人編號,而後從源點引入一條邊鏈接人,容量爲1,費用爲0,給房子編號,每一個人到每一個房子連一條邊,容量爲1,費用爲他們的曼哈頓距離。而後再設置一個點做爲匯點,全部的房子都與匯點連一條邊,容量爲1,費用爲0.求一遍最小費用流就好了。blog
const int N=1e5+10; struct Edge { int to,next,cap,flow,cost; } e[N*10]; int head[N],tot; int pre[N],dis[N]; int vis[N],Vis[201][201]; int n,m,tn; char s[201][201]; void init(int num) { tot=0;//邊的數量 tn=num; memset(head,-1,sizeof(head)); } void add(int u,int v,int cap,int cost) { e[tot].to=v,e[tot].cap=cap,e[tot].cost=cost,e[tot].flow=0; e[tot].next=head[u]; head[u]=tot++; e[tot].to=u,e[tot].cap=0,e[tot].cost=-cost,e[tot].flow=0; e[tot].next=head[v]; head[v]=tot++; } bool spfa(int s,int t) { queue<int>q; for(int i=0; i<=tn; i++) { dis[i]=INF; vis[i]=0; pre[i]=-1; } dis[s]=0; vis[s]=1; q.push(s); while(!q.empty()) { int u=q.front(); q.pop(); vis[u]=0; for(int i=head[u]; i+1; i=e[i].next) { int v=e[i].to; if(e[i].cap>e[i].flow&&dis[v]>dis[u]+e[i].cost) { dis[v]=dis[u]+e[i].cost; pre[v]=i; if(!vis[v]) { vis[v]=1; q.push(v); } } } } if(pre[t]==-1) return false; return true; } int mincost_maxflow(int s,int t) { int cost=0; while(spfa(s,t)) { int Min=INF; for(int i=pre[t]; i!=-1; i=pre[e[i^1].to]) if(Min>e[i].cap-e[i].flow) Min=e[i].cap-e[i].flow; for(int i=pre[t]; i!=-1; i=pre[e[i^1].to]) { e[i].flow+=Min; e[i^1].flow-=Min; cost+=e[i].cost*Min; } } return cost; } struct node { int x,y; }; vector<node>M,H; int main() { while(~scanf("%d%d",&n,&m)) { if(n==0&&m==0) return 0; M.clear(); H.clear(); int num=1,tmp=1; for(int i=0; i<n; i++) { scanf("%s",s[i]); for(int j=0; j<m; j++) if(s[i][j]!='.') tmp++; } init(tmp); for(int i=0; i<n; i++) for(int j=0; j<m; j++) if(s[i][j]!='.') { if(s[i][j]=='m') { add(0,num,1,0);//0爲源點; M.push_back(node {i,j}); } else { add(num,tmp,1,0);//m+h+1爲匯點 H.push_back(node {i,j}); } Vis[i][j]=num++; } for(int i=0; i<M.size(); i++) { node tmpM=M[i]; for(int j=0; j<H.size(); j++) { node tmpH=H[j]; int diss=abs(tmpM.x-tmpH.x)+abs(tmpM.y-tmpH.y); add(Vis[tmpM.x][tmpM.y],Vis[tmpH.x][tmpH.y],1,diss); } } printf("%d\n",mincost_maxflow(0,tmp)); } return 0; }