POJ 2987 最大权闭合子图

题意

传送门 POJ 2987

题解

最大权闭合子图建图

最大权闭合子图建图基本方法：源点向每一个正权值节点连边，每一个负权值节点向汇点连边，边权值为点权值的绝对值；每一条有向边连一条容量为 INF 的边（这保证了求最小割后，边两端节点位于同一个点集）。

算法证明

参考《最小割模型在信息学竞赛中的应用》。考虑闭合子图的节点位于包含源点的点集 $S$，其余节点位于包含汇点的点集 $T$，则设最大权为 $W$，最小割流量为 $F$，下标 $+$，$-$ 表示点集中权值为正或为负的点

$W=\sum _{v\in S_{+}}{w}_v+\sum _{v\in S_{-}}{w}_v$

$F=\sum _{v\in S_{-}}(-w_v)+\sum _{v\in T_{+}}{w}_v$

联立求最大权

W = − F + ∑ v ∈ { S + ∪   T + } w v W=-F+\sum\limits_{v\in \{S_{+}\cup \ T_{+}\}}w_{v} W=−F+v∈{S+​∪ T+​}∑​wv​

则闭合子图的最大权等于正权值节点的权值和减去最小割流量，此时最大权闭合子图的节点对应从源点可达的节点。因为边容量为 $0$ 时不搜索下一个节点，所以如果有多个的最小割，dfs 仅搜索到使包含源点点集最小点数的最小割；故 dfs 的点集是最小点集。

代码

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
#define min(a,b)    (((a) < (b)) ? (a) : (b))
#define max(a,b)    (((a) > (b)) ? (a) : (b))
#define abs(x)    ((x) < 0 ? -(x) : (x))
#define INF 0x3f3f3f3f
#define delta 0.85
#define eps 1e-5
#define PI 3.14159265358979323846
using namespace std;

typedef long long LL;
#define MAX_V 5005
struct edge{
	int to, cap, rev;
	edge(int to, int cap, int rev) : to(to), cap(cap), rev(rev){}
};

vector<edge> G[MAX_V];
int level[MAX_V], iter[MAX_V];

void add_edge(int from, int to, int cap){
	G[from].push_back(edge(to, cap, G[to].size()));
	G[to].push_back(edge(from, 0, G[from].size() - 1));
}

void bfs(int s){
	memset(level, -1, sizeof(level));
	queue<int> que;
	level[s] = 0;
	que.push(s);
	while(!que.empty()){
		int v = que.front(); que.pop();
		for(int i = 0; i < G[v].size(); i++){
			edge &e = G[v][i];
			if(e.cap > 0 && level[e.to] < 0){
				level[e.to] = level[v] + 1;
				que.push(e.to);
			}
		}
	}
}

int dfs(int v, int t, int f){
	if(v == t) return f;
	for(int &i = iter[v]; i < G[v].size(); i++){
		edge &e = G[v][i];
		if(e.cap > 0 && level[v] < level[e.to]){
			int d = dfs(e.to, t, min(f, e.cap));
			if(d > 0){
				e.cap -= d;
				G[e.to][e.rev].cap += d;
				return d;
			}
		}
	}
	return 0;
}

LL max_flow(int s, int t){
	LL flow = 0;
	for(;;){
		bfs(s);
		if(level[t] < 0) return flow;
		memset(iter, 0, sizeof(iter));
		int f;
		while((f = dfs(s, t, INF)) > 0){
			flow += f;
		}
	}
}

#define MAX_N 5000
#define MAX_M 60000
int N, M;
int b[MAX_N];
int a[MAX_M], c[MAX_M];
bool used[MAX_N];
// 求包含源点的点集 S 点数
int dfs(int u){
	used[u] = 1;
	int sum = 1;
	for(int i = 0; i < G[u].size(); i++){
		edge &e = G[u][i];
		if(e.cap == 0 || used[e.to]) continue;
		sum += dfs(e.to);
	}
	return sum;
}

void solve(){
	LL sum = 0;
	int s = N, t = s + 1;
	// 最大权闭合子图建图
	for(int i = 0; i < N; i++){
		if(b[i] > 0){
			sum += b[i];
			add_edge(s, i, b[i]);
		}
		else add_edge(i, t, -b[i]);
	}
	for(int i = 0; i < M; i++) add_edge(a[i], c[i], INF);
	memset(used, 0, sizeof(used));
	printf("%d %lld\n", dfs(s) - 1, sum - max_flow(s, t));
}

int main(){
	while(~scanf("%d%d", &N, &M)){
		for(int v = 0; v < N + 2; v++) G[v].clear();
		for(int i = 0; i < N; i++) scanf("%d", b + i);
		for(int i = 0; i < M; i++){
			scanf("%d%d", a + i, c + i);
			--a[i], --c[i];
		}
		solve();
	}
	return 0;
}