🧩 Baekjoon 30802 - 웰컴 키트

문제

2024년 2월 3일 개최 예정인 온사이트 그랜드 아레나에서는
참가자들에게 티셔츠 한 장과 펜 한 자루가 포함된 웰컴 키트를 나눠줄 예정입니다.
키트를 제작하는 업체는 다음과 같은 조건으로만 주문이 가능합니다.

티셔츠는 S, M, L, XL, XXL, 그리고 XXXL의 6가지 사이즈가 있습니다.
티셔츠는 같은 사이즈의 T장 묶음으로만 주문할 수 있습니다.
펜은 한 종류로, P자루씩 묶음으로 주문하거나 한 자루씩 주문할 수 있습니다.
총 N명의 참가자 중 S, M, L, XL, XXL, XXXL 사이즈의 티셔츠를 신청한 사람은
각각 S, M, L, XL, XXL, XXXL명입니다.
티셔츠는 남아도 되지만 부족해서는 안 되고 신청한 사이즈대로 나눠주어야 합니다.
펜은 남거나 부족해서는 안 되고 정확히 참가자 수만큼 준비되어야 합니다.

티셔츠를 T장씩 최소 몇 묶음 주문해야 하는지,
그리고 펜을 P자루씩 최대 몇 묶음 주문할 수 있고,
그 때 펜을 한 자루씩 몇 개 주문하는지 구하세요.

입력

첫 줄에 참가자의 수 N이 주어집니다. 둘째 줄에 티셔츠 사이즈별 신청자의 수 S, M, L, XL, XXL, XXXL이 공백으로 구분되어 주어집니다.
셋째 줄에 정수 티셔츠와 펜의 묶음 수를 의미하는 정수 T와 P가 공백으로 구분되어 주어집니다.

출력

첫 줄에 티셔츠를 T장씩 최소 몇 묶음 주문해야 하는지 출력하세요.
다음 줄에 펜을 P자루씩 최대 몇 묶음 주문할 수 있는지와,
그 때 펜을 한 자루씩 몇 개 주문하는지 구하세요.​

예제

✅ 입력 1

23 
3 1 4 1 5 9 
5 7

✅ 출력 1

7
3 2

작성 코드

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
	public static void main(String[] args) throws Exception {
		long startTime = System.currentTimeMillis();
		
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		int[] sizes = Arrays.stream(br.readLine().split(" "))
			.mapToInt(Integer::parseInt)
			.toArray();
				
		StringTokenizer st = new StringTokenizer(br.readLine());
		int t = Integer.parseInt(st.nextToken());
		int p = Integer.parseInt(st.nextToken());
		
		int tBundle = Arrays.stream(sizes)
			.map(i -> (i % t) > 0 ? (i / t) + 1 : (i / t))
			.sum();
				
		int pBundle = n / p;
		int pEach = n % p;
		
		System.out.println(tBundle);
		System.out.printf("%d %d\n", pBundle, pEach);
		
		long endTime = System.currentTimeMillis();
		
		System.out.println("실행 시간: " + (endTime - startTime) + " ms");
	}
}

• 실행 시간은 955 ms다.

개선 코드

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

public class Main {
	public static void main(String[] args) throws Exception {
		long startTime = System.currentTimeMillis();
		
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		
		String[] sizeStrArr = br.readLine().split(" ");
		int[] sizes = new int[sizeStrArr.length];
		for (int i = 0; i < sizeStrArr.length; i++) {
			sizes[i] = Integer.parseInt(sizeStrArr[i]);
		}
		
		StringTokenizer st = new StringTokenizer(br.readLine());
		int t = Integer.parseInt(st.nextToken());
		int p = Integer.parseInt(st.nextToken());
		
		int tBundle = 0;
		for (int size : sizes) {
			if (size % t > 0) {
				tBundle += (size / t) + 1;
			} else {
				tBundle += size / t;
			}
		}
		
		int pBundle = n / p;
		int pEach = n % p;
		
		System.out.println(tBundle);
		System.out.printf("%d %d\n", pBundle, pEach);
		
		long endTime = System.currentTimeMillis();
		
		System.out.println("실행 시간: " + (endTime - startTime) + " ms");
	}
}

• 실행 시간은 918 ms다.

회고

• Stream을 통해 작성한 코드보다 조금 빠르긴 하지만 현재는 데이터가 작아서 그런지 차이가 미미하다.
• 반복을 훨씬 많이 한다면 차이가 커질 것으로 보인다.
• for-each문이 빠르다고 무조건 Stream보다 좋은 것은 아니다.
• 코드가 너무 길어지지만 않는다면 Stream이 가독성이 더 좋다고 생각한다.
• 두 문법 간의 Trade-off를 고려해서 적절한 문법을 사용해야겠다.