[Baekjoon] 6754_Bridge transport

문제(출처: https://www.acmicpc.net/problem/6754)

---

< Bridge transport >

 

문제 풀이 

 

최대 4개씩 다리를 건너며 최대 무게를 넘는지 확인한다. 

 

my solution (Java)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class _6754_ { // Bridge transport

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));

		int w = Integer.parseInt(bf.readLine());
		int n = Integer.parseInt(bf.readLine());

		int arr[] = new int[n];
		for (int i = 0; i < n; i++) {
			arr[i] = Integer.parseInt(bf.readLine());
		}

		int sum = 0;
		boolean flag = false;
		for (int i = 0; i < Math.min(4, n); i++) {
			sum += arr[i];
			if (sum > w) {
				System.out.println(i);
				flag = true;
				break;
			}
		}

		if (!flag) {
			if (n < 5) {
				System.out.println(n);
			} else {
				flag = false;
				for (int i = 4; i < n; i++) {
					sum -= arr[i - 4];
					sum += arr[i];

					if (sum > w) {
						System.out.println(i);
						flag = true;
						break;
					}
				}
				if (!flag) {
					System.out.println(n);
				}
			}
		}
	}
}

변수)
w : 최대 무개
n : 차량의 수
arr : 각 차량 무개
sum : 다리 위에 있는 차량 무게 합

 

최대 무개, 차량의 수를 입력받는다. 차량의 수만큼 각 차량의 무게를 입력받아 arr에 저장한다. 다리에 최대 4대가 올라갈 수 있으므로 초기에 4대의 차량 무게 합을 구한다. 이때, n < 4 라면 n개만큼만 더한다. 합이 최대 무개보다 크다면 최대 무개를 넘지 않는 차량 개수를 출력한다. 최대 무개를 넘지 않는데 n < 5라면 n대의 차량이 다 건널 수 있으므로 n을 출력한다. n이 5 이상이라면 맨 앞의 한대를 보내고, 다음 한 대를 다리 위로 올린다. 이때도 최대 무개보다 커진다면 최대 무개를 넘지 않는 차량 개수를 출력한다. 다 건넜다면 n을 출력한다. 

---

---