공부공간

BOJ - 1939 ) 중량제한 본문

알고리즘/완전탐색(BFS,DFS)

BOJ - 1939 ) 중량제한

개발자가될수있을까? 2020. 9. 28. 19:49


www.acmicpc.net/problem/1939

 

1939번: 중량제한

첫째 줄에 N, M(1≤M≤100,000)이 주어진다. 다음 M개의 줄에는 다리에 대한 정보를 나타내는 세 정수 A, B(1≤A, B≤N), C(1≤C≤1,000,000,000)가 주어진다. 이는 A번 섬과 B번 섬 사이에 중량제한이 C인 다리

www.acmicpc.net

 


0부터 ~ 최대 Capacity 까지의 수를 이분탐색으로 구하고, 구한 숫자로 유량을 보낼 수 있는지?

 

확인하는 이분탐색 + bfs 문제

 


import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.StringTokenizer;

public class Main {

	public static ArrayList<int[]> node[] ;
	public static int flow,start,end,step=1;
	public static int visit[];
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		int n = Integer.parseInt(st.nextToken());
		int m = Integer.parseInt(st.nextToken());
		node =new ArrayList[n];
		visit = new int[n];
		for(int i =0;i<n;i++)node[i] = new ArrayList<>();
		int max = -1;
		for(int i =0;i<m;i++) {
			st = new StringTokenizer(br.readLine());
			int s = Integer.parseInt(st.nextToken())-1;
			int e = Integer.parseInt(st.nextToken())-1;
			int c = Integer.parseInt(st.nextToken());
			max = max > c ? max : c;
			node[s].add(new int[] {e,c});
			node[e].add(new int[] {s,c});
		}
		st = new StringTokenizer(br.readLine());
		 start = Integer.parseInt(st.nextToken())-1;
		 end = Integer.parseInt(st.nextToken())-1;
		int min = 0;
		int answer = 0;
		while(min <= max) {
			
			 flow = ((min+max)>>1);
			
			if(bfs()) {
				answer = flow;
				min = flow+1;
			} else {
				max = flow-1;
			}
			step++;
			
		}System.out.println(answer);
	}

	private static boolean bfs() {
		// flow로 bfs가 가능한지?
		ArrayDeque<int[]> dq = new ArrayDeque<>();
		dq.add(new int[] {start,0});
		visit[start] = step;
		while(!dq.isEmpty()) {
			int [] now = dq.poll();
			int ss = now[0];
			if(ss==end) {
				return true;
			} else {
				for(int i =0;i< node[ss].size();i++) {
					int next = node[ss].get(i)[0];
					int cc =  node[ss].get(i)[1];
					if(flow<= cc && visit[next]< step){
						visit[next]= step;
						dq.add(new int[] {next,cc});
					}
				}
			}
		}
		return false;
	}
}

Comments