[CF 343E]Pumping Stations解題報告

題目翻譯


題解

首先,有一個東西叫Gomory-Hu(戈莫里-胡)樹。就是說,對於一張題中這樣的圖可以建出來一棵樹,使得圖中s~t的最小割等於樹中s~t路徑上的最小邊權。

那麼問題來了:怎麼建樹呢?

有兩篇資料:
Wiki百科:

yangff的博客:

如果你看完之後還不知道怎麼建,不用擔心,你不是一個人……

其實這個建樹過程寫出來並不難,看一下代碼就知道了。大致思路是:一開始所有點都在同一個集合內。然後每次選一個集合,對其中兩個點做最小割,按照源彙集分割這個集合,並添加一條邊。最後所有集合都剩下一個點,樹也就建完了。共需要做O(n)次最大流。

現在有了這棵樹,應該怎麼辦呢?

很簡單:選擇樹中權值最小的邊,那麼最優解一定是先在一側走完,再經過這條邊,再走另外一側(只經過該權值最小的邊一次,最優方案必然如此)。然後分成兩部分遞歸下去即可。


所以這篇題解的核心其實是貼一下Gomory-Hu樹的代碼,相信我,代碼真的非常簡單……


代碼

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
using namespace std;
const int SIZEN=210,INF=0x7fffffff/2;
class EDGE{
public:
	int from,to,cap,flow;
};
vector<EDGE> edges;
vector<int> c[SIZEN];
int S,T;
bool visit[SIZEN]={0};
int depth[SIZEN]={0};
int cur[SIZEN]={0};
void clear_graph(void){
	edges.clear();
	for(int i=0;i<SIZEN;i++) c[i].clear();
}
void clear_flow(void){
	for(int i=0;i<edges.size();i++) edges[i].flow=0;
}
void addedge_2(int from,int to,int cap){//加雙向邊
	EDGE temp;
	temp.from=from,temp.to=to,temp.cap=cap,temp.flow=0;
	edges.push_back(temp);
	temp.from=to,temp.to=from,temp.cap=cap,temp.flow=0;
	edges.push_back(temp);
	int tot=edges.size()-2;
	c[from].push_back(tot);
	c[to].push_back(tot+1);
}
bool BFS(void){
	memset(visit,0,sizeof(visit));
	memset(depth,-1,sizeof(depth));
	queue<int> Q;
	Q.push(S);visit[S]=true;depth[S]=0;
	while(!Q.empty()){
		int x=Q.front();Q.pop();
		for(int i=0;i<c[x].size();i++){
			EDGE &now=edges[c[x][i]];
			if(!visit[now.to]&&now.cap>now.flow){
				visit[now.to]=true;
				depth[now.to]=depth[x]+1;
				Q.push(now.to);
			}
		}
	}
	return visit[T];
}
int DFS(int x,int a){
	if(x==T||!a) return a;
	int flow=0,cf=0;
	for(int i=cur[x];i<c[x].size();i++){
		cur[x]=i;
		EDGE &now=edges[c[x][i]];
		if(depth[x]+1==depth[now.to]){
			cf=DFS(now.to,min(a,now.cap-now.flow));
			if(cf){
				flow+=cf;
				a-=cf;
				now.flow+=cf,edges[c[x][i]^1].flow-=cf;
			}
			if(!a) break;
		}
	}
	if(!flow) depth[x]=-1;
	return flow;
}
int Dinic(void){
	int flow=0;
	while(BFS()){
		memset(cur,0,sizeof(cur));
		flow+=DFS(S,INF);
	}
	return flow;
}
int N,M;
int fa[SIZEN],falen[SIZEN];
int now;
void find_min(int x,int fa){
	for(int i=0;i<c[x].size();i++){
		EDGE &e=edges[c[x][i]];
		if(e.to!=fa&&e.cap!=-1){
			if(now==-1||e.cap<edges[now].cap) now=c[x][i];
			find_min(e.to,x);
		}
	}
}
void Solve(int x){
	now=-1;
	find_min(x,0);
	if(now==-1){
		printf("%d ",x);
		return;
	}
	edges[now].cap=edges[now^1].cap=-1;
	int p=now;
	Solve(edges[p].from);
	Solve(edges[p].to);
}
int ans=0;
void build_tree(void){//建樹
	for(int i=1;i<=N;i++) fa[i]=1;
	for(int i=2;i<=N;i++){
		clear_flow();
		S=i,T=fa[i];
		falen[i]=Dinic();
		BFS();
		for(int j=i+1;j<=N;j++)
			if(visit[j]&&fa[j]==fa[i]) fa[j]=i;
	}
	clear_graph();
	for(int i=2;i<=N;i++)
		addedge_2(i,fa[i],falen[i]),ans+=falen[i];
}
void answer(void){
	printf("%d\n",ans);
	Solve(1);
	printf("\n");
}
void init(void){
	scanf("%d%d",&N,&M);
	int a,b,w;
	for(int i=1;i<=M;i++){
		scanf("%d%d%d",&a,&b,&w);
		addedge_2(a,b,w);
	}
}
int main(){
	//freopen("pumpingstations.in","r",stdin);
	//freopen("pumpingstations.out","w",stdout);
	init();
	build_tree();
	answer();
	return 0;
}
建樹就是build_tree函數。


發佈了53 篇原創文章 · 獲贊 17 · 訪問量 12萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章