主要是建图有些小烦。若几个节点流量的来源或者去向全然同样,且流量为 INF,将它们合并成一个节点。
若从两点间有且仅有一条容量为 INF 的边。将两点合并成一个节点。
#include <iostream>
#include <cstring>
#include <queue>
using namespace std;
const int INF = 0x7fffffff;
const int MAXN = 110;
int capacity[MAXN][MAXN], flow[MAXN][MAXN];
int parent[MAXN], residual[MAXN];
int max_flow, node_num;
void edmond_karp( int source, int target ) {
queue< int > Q;
memset( flow, 0, sizeof( flow ) );
memset( parent, 0, sizeof( parent ) );
max_flow = 0;
while ( true ) {
memset( residual, 0, sizeof( residual ) );
residual[source] = INF;
Q.push( source );
while ( !Q.empty() ) {
int u = Q.front();
Q.pop();
for( int v = 0; v <= node_num + 1; ++v ) {
if( !residual[v] && capacity[u][v] > flow[u][v] ) {
parent[v] = u;
Q.push( v );
residual[v] = residual[u] < capacity[u][v] - flow[u][v] ?
residual[u] : capacity[u][v] - flow[u][v]; } } } if( residual[target] == 0 ) break; for( int u = target; u != source; u = parent[u] ) { flow[parent[u]][u] += residual[target]; flow[u][parent[u]] -= residual[target]; } max_flow += residual[target]; } } int main() { int pighouses_num, customers_num; int pighouses_capacity[1010]; int visited_by_somebody[1010]; while( cin >> pighouses_num >> customers_num ) { memset( capacity, 0, sizeof( capacity ) ); memset( visited_by_somebody, 0, sizeof( visited_by_somebody ) ); int source = 0, target = customers_num + 1; node_num = customers_num; for( int i = 1; i <= pighouses_num; ++i ) cin >> pighouses_capacity[i]; for( int i = 1; i <= customers_num; ++i ) { int keys_num; cin >> keys_num; for( int j = 0; j < keys_num; ++j ) { int temp; cin >> temp; if( visited_by_somebody[temp] == 0 ) capacity[source][i] += pighouses_capacity[temp]; else capacity[visited_by_somebody[temp]][i] = INF; visited_by_somebody[temp] = i; } cin >> capacity[i][target]; } edmond_karp( source, target ); cout << max_flow << endl; } return 0; }