[백준 1939] 중량제한 (자바)

백준 1939번 중량제한 (자바)

 

 

 

출처

www.acmicpc.net/problem/1939

1939번: 중량제한

 

 

 

문제

N(2≤N≤10,000)개의 섬으로 이루어진 나라가 있다. 이들 중 몇 개의 섬 사이에는 다리가 설치되어 있어서 차들이 다닐 수 있다.

영식 중공업에서는 두 개의 섬에 공장을 세워 두고 물품을 생산하는 일을 하고 있다. 물품을 생산하다 보면 공장에서 다른 공장으로 생산 중이던 물품을 수송해야 할 일이 생기곤 한다. 그런데 각각의 다리마다 중량제한이 있기 때문에 무턱대고 물품을 옮길 순 없다. 만약 중량제한을 초과하는 양의 물품이 다리를 지나게 되면 다리가 무너지게 된다.

한 번의 이동에서 옮길 수 있는 물품들의 중량의 최댓값을 구하는 프로그램을 작성하시오.

 

 

 

입력

첫째 줄에 N, M(1≤M≤100,000)이 주어진다. 다음 M개의 줄에는 다리에 대한 정보를 나타내는 세 정수 A, B(1≤A, B≤N), C(1≤C≤1,000,000,000)가 주어진다. 이는 A번 섬과 B번 섬 사이에 중량제한이 C인 다리가 존재한다는 의미이다. 서로 같은 두 도시 사이에 여러 개의 다리가 있을 수도 있으며, 모든 다리는 양방향이다. 마지막 줄에는 공장이 위치해 있는 섬의 번호를 나타내는 서로 다른 두 정수가 주어진다. 공장이 있는 두 섬을 연결하는 경로는 항상 존재하는 데이터만 입력으로 주어진다.

 

 

 

출력

첫째 줄에 답을 출력한다.

 

 

 

입출력 예

입출력 예

 

 

 

걸린 시간

1시간 13분..!!!

 

 

 

접근 방법

백트래킹으로 접근하려했지만 어림도 없지~~!

이분탐색과 너비 우선 탐색을 사용하면 풀 수 있다.(조건 범위가 너무 크다)

 

right에 중량제한의 최댓값을 넣는다.

이분 탐색을 사용해서 최대 중량 기준을 mid로 두고 bfs를 실행한다.

bfs에서 방문하지 않았던 나라이고, mid보다 제한이 큰 경우 계속 탐색하도록 한다.

이후에, end까지 도달하는 경우 true를 반환하고, 도달하지 못하면 false를 반환한다.

true를 반환하는 경우는 더 큰 값이 있는 것이므로 오른쪽으로 범위를 옮긴다.

false를 반환하는 경우는 큰 값이 없는 것이므로 왼쪽으로 범위를 옮긴다.

 

위를 반복하면 최종적으로 right에 최대 중량 제한이 들어가게 된다.

 

 

 

내 코드

	package binarySearch;
	import java.io.BufferedReader;
	import java.io.IOException;
	import java.io.InputStreamReader;
	import java.util.ArrayList;
	import java.util.LinkedList;
	import java.util.Queue;
	import java.util.StringTokenizer;
	public class WeightLimit {
	/**
	* 백준 1939 중량제한 (https://www.acmicpc.net/problem/1939)
	*/
	private static int n,m;
	private static ArrayList<ArrayList<Island>> list = new ArrayList<>();
	private static boolean[] visit;
	public static void main(String[] args) throws IOException {
	BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
	StringTokenizer st = new StringTokenizer(reader.readLine());
	n = Integer.parseInt(st.nextToken());
	m = Integer.parseInt(st.nextToken());
	int left = 0;
	int right = 0;
	for (int i=0; i<=n; i++) {
	list.add(new ArrayList<>());
	}
	int max = 0;
	for (int i=0; i<m; i++) {
	st = new StringTokenizer(reader.readLine());
	int a = Integer.parseInt(st.nextToken());
	int b = Integer.parseInt(st.nextToken());
	int c = Integer.parseInt(st.nextToken());
	list.get(a).add(new Island(b,c));
	list.get(b).add(new Island(a,c));
	max = Math.max(max, c);
	}
	right = max;
	st = new StringTokenizer(reader.readLine());
	int start = Integer.parseInt(st.nextToken());
	int end = Integer.parseInt(st.nextToken());
	while (left <= right) {
	int mid = (left+right)/2;
	visit = new boolean[n+1];
	if (bfs(start,end,mid)) {
	left = mid+1;
	} else {
	right = mid-1;
	}
	}//while
	System.out.println(right);
	}
	private static boolean bfs(int start, int end, int mid) {
	Queue<Integer> q = new LinkedList<>();
	q.add(start);
	visit[start] = true;
	while (!q.isEmpty()) {
	int island = q.poll();
	if (island == end) {
	return true;
	}
	for (Island i : list.get(island)) {
	if (!visit[i.destination] && mid <= i.cost) {
	visit[i.destination] = true;
	q.add(i.destination);
	}
	}
	}//while
	return false;
	}//bfs
	static class Island {
	int destination, cost;
	Island (int destination, int cost) {
	this.destination = destination;
	this.cost = cost;
	}
	}
	}

view raw WeightLimit.java hosted with ❤ by GitHub

 

 

 

고려할 점

1. 이분탐색

2. bfs

3. 중량 제한