글 읽기 - 4716 풍선

yeonbn 8년 전 0

어디서 틀린지를 잘 모르겠네요.

혹시 오류 발견하신 분 도움좀..

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String[] str = br.readLine().trim().split(" ");

		int n = Integer.parseInt(str[0]);
		int balloonA = Integer.parseInt(str[1]); // A팀 풍선 총 갯수
		int balloonB = Integer.parseInt(str[2]); // B팀 풍선 총 갯수

		int[][] testCase = new int[n][3];

		for (int i = 0; i < n; i++) {
			str = br.readLine().trim().split(" ");
			for (int j = 0; j < 3; j++) {
				testCase[i][j] = Integer.parseInt(str[j]);
			}
		}
		br.readLine(); // 0 0 0 입력하는 것.
		br.close();

		// A를 기준으로 거리가 짧은 순서대로 정렬. 그 다음에는 B를 기준으로 거리가 짧은 순서대로 정렬.
		Arrays.sort(testCase, new Comparator<int[]>() {
			@Override
			public int compare(int[] o1, int[] o2) {
				if(Math.abs(o2[1] - o2[2]) != Math.abs(o1[1] - o1[2])) {
					return Math.abs(o2[1] - o2[2]) - Math.abs(o1[1] - o1[2]);
				} else {
					return o2[0] - o1[0];
				}
			}
		});

		// 확인하기
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < 3; j++) {
				System.out.print(testCase[i][j] + " ");
			}
			System.out.println();
		}

		long sum = 0;
		for (int i = 0; i < n; i++) {
			// 1번의 거리가 더 짧을 때.
			if (testCase[i][1] < testCase[i][2]) { // 정답이 아니라면 부등식 체크 해보자.
				if (balloonA > testCase[i][0]) {
					sum += testCase[i][0] * testCase[i][1];
					balloonA -= testCase[i][0]; // 풍선을 쓴만큼 빼준다.
				} else {
					sum += balloonA * testCase[i][1];
					sum += (testCase[i][0] - balloonA) * testCase[i][2]; // 나머지는
					balloonA = 0;
				}
			}
			// 2번의 거리가 더 짧을 때
			else {
				if (balloonB > testCase[i][0]) {
					sum += testCase[i][0] * testCase[i][2];
					balloonB -= testCase[i][0]; // 풍선을 쓴만큼 빼준다.
				} else {
					sum += balloonB * testCase[i][2];
					sum += (testCase[i][0] - balloonB) * testCase[i][1]; // 나머지는
					balloonB = 0;
				}
			}

		} // end for

		System.out.println(sum);

	}

}

kyg516 8년 전 0

여러번의 테스트케이스가 있다고 문제에 나와있네요

마지막 0 0 0은 테스트케이스 종료를 알려주기위한 거구요

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

4716 풍선

yeonbn 8년 전 0

kyg516 8년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말