백준11378 열혈강호4

문제 링크

  • http://icpc.me/11378

사용 알고리즘

  • Ford Fulkerson

시간복잡도

  • O(VE2)

풀이

source와 사람, 사람과 일, 일과 sink를 모두 용량이 1인 간선으로 만들어줍시다.
그 다음에 bridge라는 정점을 새로 만들어 source에서 bridge까지 용량이 k인 간선으로 이어주고, bridge와 사람도 용량이 k인 간선으로 이어주면 문제에 적합한 그래프를 만들 수 있습니다.

전체 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#include <bits/stdc++.h>
using namespace std;

typedef pair<int, int> p;

const int inf = 1e9+7;

class FordFulkerson{
	private:
		vector< vector<int> > flow, capacity, adjList;
		int s, e, n; //source, sink, verties
	public:
		FordFulkerson(){ n = 0, s = -1, e = -1; }
		FordFulkerson(int n){
			this->n = n;
			flow.resize(n+1), capacity.resize(n+1), adjList.resize(n+1);
			for(int i=0; i<=n; i++){
				flow[i].resize(n+1);
				capacity[i].resize(n+1);
			}
			s = -1, e = -1;
		}
		FordFulkerson(int n, int source, int sink){
			this->n = n;
			flow.resize(n+1), capacity.resize(n+1), adjList.resize(n+1);
			for(int i=0; i<=n; i++){
				flow[i].resize(n+1);
				capacity[i].resize(n+1);
			}
			s = source, e = sink;
		}
		void setSource(int t){ s = t; }
		void setSink(int t){ e = t; }
		void addEdge(int start, int end, int cap, bool directed){
			adjList[start].push_back(end);
			adjList[end].push_back(start);
			capacity[start][end] += cap;
			if(!directed) capacity[end][start] += cap;
		}

		int maxFlow(){
			if(s==-1 || e==-1) return -1;
			vector<int> par(n+1);
			queue<int> q;
			int ret = 0;
			while(1){
				fill(par.begin(), par.end(), -1);
				q = queue<int>(); q.push(s);
				while(!q.empty()){
					int now = q.front(); q.pop();
					for(auto nxt : adjList[now]){
						if(par[nxt] == -1 && capacity[now][nxt]-flow[now][nxt] > 0){
							q.push(nxt); par[nxt] = now;
							if(nxt == e) break;
						}
					}
				}
				if(par[e] == -1) break;
				int cost = inf;
				for(int i=e; i!=s; i = par[i]){
					int a = par[i], b = i;
					cost = min(cost, capacity[a][b] - flow[a][b]);
				}
				ret += cost;
				for(int i=e; i!=s; i = par[i]){
					int a = par[i], b = i;
					flow[a][b] += cost; flow[b][a] -= cost;
				}
			}
			return ret;
		}
};

int main(){
	ios_base::sync_with_stdio(0); cin.tie(0);
	int n, m, k; cin >> n >> m >> k;
	int s = 2001, e = 2003, bridge = 2002;
	FordFulkerson flow(2003, s, e);
	for(int i=1; i<=n; i++){
		flow.addEdge(s, i, 1, 1);
		flow.addEdge(bridge, i, k, 1);
		int x; cin >> x;
		while(x--){
			int t; cin >> t; flow.addEdge(i, 1000+t, 1, 1);
		}
	}
	flow.addEdge(s, bridge, k, 1);
	for(int i=1; i<=m; i++) flow.addEdge(1000+i, e, 1, 1);
	cout << flow.maxFlow();
}