백준알고리즘 6086번, 최대 유량
Contents
드디어 네트워크플로우
처음에 목표한 알고리즘이 MCMF까지였기 때문에, 곧 결과에 쫓기듯 공부할 필요는 없을꺼같다.
배울 알고리즘이 참 많지만 사실 그런알고리즘은 내가 앞으로 쓸일이 있을까 하는 의문이 들어서 필요성을 느끼지 못하겠다. 그리고 필요하다면 그 때 배워서 사용하면 될꺼같기도 하다. 시험문제로 나온다면.. 그저 슬플수 밖에
어쨋든 앞으로는 네트워크 플로우를 공부좀 하다가 어려운 dp들을 풀어봐야겠다. 이후에는 새로운 알고리즘은 세그먼트 트리, 문자열 알고리즘, 컨벡스 홀 중에서 선택하지 싶다.
인접행렬을 사용한 풀이
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 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 |
/**
*
*/
import java.io.BufferedReader;
import java.io.FileReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;
import java.util.StringTokenizer;
/**
* @author jaeji
*
*/
public class Main {
static int atoi(String a) {
if(a.charAt(0)<='Z') {
return a.charAt(0)-'A';
}
else {
return a.charAt(0)-'a'+26;
}
}
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader( new InputStreamReader( System.in ) );
//BufferedReader br = new BufferedReader( new FileReader("input.txt" ) );
StringTokenizer st = new StringTokenizer(br.readLine());
int n = Integer.parseInt(st.nextToken());
int[][] f = new int[52][52];
int[][] c = new int[52][52];
int[] prev = new int[52];
ArrayList[] adj = new ArrayList[52];
Queue<Integer> q = new LinkedList<Integer>();
int S = atoi("A");
int E = atoi("Z");
int u=0,v=0,w=0;
int total = 0;
for(int i=0;i<52;i++) {
adj[i] = new ArrayList<Integer>();
}
for(int i=0;i<n;i++) {
st = new StringTokenizer(br.readLine());
u = atoi(st.nextToken());
v = atoi(st.nextToken());
w = Integer.parseInt(st.nextToken());
c[u][v] += w;
adj[u].add(v);
adj[v].add(u);
}
int curr = 0;
int next = 0;
int adj_size = 0;
int flow = 0;
while(true) {
Arrays.fill(prev, -1);
q.clear();
q.add(S);
curr = 0;
next = 0;
adj_size = 0;
while(!q.isEmpty()) { // 소스에서 흐를수 있는 유량찾기
curr = q.poll();
adj_size = adj[curr].size();
for(int i=0;i<adj_size;i++) {
next = (int) adj[curr].get(i);
if(c[curr][next] - f[curr][next] > 0 && prev[next] == -1) {
q.add(next);
prev[next] = curr;
if(next == E) break;
}
}
}
if(prev[E] == -1) break; // 경로를 탐색 한 후 싱크로 가는 경로가 없는 경우
flow = Integer.MAX_VALUE;
for(int i=E;i!=S;i=prev[i]) {
flow = Math.min(flow, c[prev[i]][i]-f[prev[i]][i]);
}
for(int i=E;i!=S;i=prev[i]) {
f[prev[i]][i]+=flow;
f[i][prev[i]]-=flow;
}
total +=flow;
}
System.out.println(total);
}
} |
인접리스트를 사용한 풀이
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 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 |
/**
*
*/
import java.io.BufferedReader;
import java.io.FileReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;
import java.util.StringTokenizer;
/**
* @author jaeji
*
*/
class edge{
int to;
int flow;
int cap;
edge rever;
edge(){
this(0,0,0,null);
}
edge(int ar_to, int ar_flow, int ar_cap, edge ar_rever){
to = ar_to;
flow = ar_flow;
cap = ar_cap;
rever = ar_rever;
}
int residu() {
return cap-flow;
}
void addflow(int ar_flow) {
flow+=ar_flow;
rever.flow -=ar_flow;
}
}
public class Main {
static int atoi(String a) {
if(a.charAt(0)<='Z') {
return a.charAt(0)-'A';
}
else {
return a.charAt(0)-'a'+26;
}
}
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader( new InputStreamReader( System.in ) );
//BufferedReader br = new BufferedReader( new FileReader("input.txt" ) );
StringTokenizer st = new StringTokenizer(br.readLine());
int n = Integer.parseInt(st.nextToken());
int[] prev = new int[52];
ArrayList[] adj = new ArrayList[52];
Queue<Integer> q = new LinkedList<Integer>();
int S = atoi("A");
int E = atoi("Z");
int u=0,v=0,w=0;
int total = 0;
for(int i=0;i<52;i++) {
adj[i] = new ArrayList<edge>();
}
for(int i=0;i<n;i++) {
st = new StringTokenizer(br.readLine());
u = atoi(st.nextToken());
v = atoi(st.nextToken());
w = Integer.parseInt(st.nextToken());
edge temp_edge = new edge(v,0,w,null);
edge temp_rever_edge = new edge(u,0,0,temp_edge);
temp_edge.rever = temp_rever_edge;
adj[u].add(temp_edge);
adj[v].add(temp_rever_edge);
}
int curr = 0;
edge next = null;
int adj_size = 0;
int flow = 0;
while(true) {
edge[] path = new edge[52];
Arrays.fill(prev, -1);
q.clear();
q.add(S);
curr = 0;
next = null;
adj_size = 0;
while(!q.isEmpty()) { // 소스에서 흐를수 있는 유량찾기
curr = q.poll();
adj_size = adj[curr].size();
for(int i=0;i<adj_size;i++) {
next = (edge) adj[curr].get(i);
if(next.residu() > 0 && prev[next.to] == -1) {
q.add(next.to);
prev[next.to] = curr;
path[next.to] = next;
if(next.to == E) break;
}
}
}
if(prev[E] == -1) break; // 경로를 탐색 한 후 싱크로 가는 경로가 없는 경우
flow = Integer.MAX_VALUE;
for(int i=E;i!=S;i=prev[i]) {
flow = Math.min(flow, path[i].residu());
}
for(int i=E;i!=S;i=prev[i]) {
path[i].addflow(flow);
}
total +=flow;
}
System.out.println(total);
}
} |
Author Jaejin Jang
LastMod 2018-01-04
License Jaejin Jang