-
[백준] 2303. 숫자 게임PS/Java 2022. 4. 6.
N명이 모여 숫자 게임을 하고자 한다. 각 사람에게는 1부터 10사이의 수가 적혀진 다섯 장의 카드가 주어진다. 그 중 세 장의 카드를 골라 합을 구한 후 일의 자리 수가 가장 큰 사람이 게임을 이기게 된다. 세 장의 카드가 (7, 8, 10)인 경우에는 합은 7+8+10 = 25가 되고 일의 자리 수는 5가 된다. 어떤 사람이 받은 카드가 (7, 5, 5, 4, 9)인 경우 (7, 4, 9)를 선택하면 합이 20이 되어 일의 자리 수는 0이 되고, (5, 5, 9)를 선택하면 합이 19가 되어 일의 자리 수는 9가 된다. 게임을 이기기 위해서는 세 장의 카드를 선택할 때 그 합의 일의 자리 수가 가장 크게 되도록 선택하여야 한다.
예를 들어, N=3일 때
- 1번 사람이 (7, 5, 5, 4, 9),
- 2번 사람이 (1, 1, 1, 1, 1),
- 3번 사람이 (2, 3, 3, 2, 10)의
카드들을 받았을 경우, 세 수의 합에서 일의 자리 수가 가장 크게 되도록 세 수를 선택하면
- 1번 사람은 (5, 5, 9)에서 9,
- 2번 사람은 (1, 1, 1)에서 3,
- 3번 사람은 (2, 3, 3)에서 8의
결과를 각각 얻을 수 있으므로 첫 번째 사람이 이 게임을 이기게 된다.
N명에게 각각 다섯 장의 카드가 주어졌을 때, 세 장의 카드를 골라 합을 구한 후 일의 자리 수가 가장 큰 사람을 찾는 프로그램을 작성하시오. 가장 큰 수를 갖는 사람이 두 명 이상일 경우에는 번호가 가장 큰 사람의 번호를 출력한다.
입력
첫 줄에는 사람의 수를 나타내는 정수 N이 주어진다. N은 2이상 1,000이하이다. 그 다음 N 줄에는 1번부터 N번까지 각 사람이 가진 카드가 주어지는 데, 각 줄에는 1부터 10사이의 정수가 다섯 개씩 주어진다. 각 정수 사이에는 한 개의 빈칸이 있다.
출력
게임에서 이긴 사람의 번호를 첫 번째 줄에 출력한다. 이긴 사람이 두 명 이상일 경우에는 번호가 가장 큰 사람의 번호를 출력한다.
예제 입력 1
3 7 5 5 4 9 1 1 1 1 1 2 3 3 2 10
예제 출력 1
1
import java.io.BufferedReader; import java.io.BufferedWriter; import java.io.InputStreamReader; import java.io.OutputStreamWriter; import java.util.Collections; import java.util.LinkedList; import java.util.StringTokenizer; public class Main { public static BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); public static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out)); public static int[] bitMask = {7, 11, 13, 14, 19, 21, 22, 25, 26, 28}; public static class Game implements Comparable<Game> { int seq; int num; public Game(int seq, int num) { this.seq = seq; this.num = num; } @Override public int compareTo(Game o) { if(o.num == this.num) { return o.seq - this.seq; } else { return o.num - this.num; } } } public static void main(String[] args) throws Exception { int N = Integer.parseInt(br.readLine()); LinkedList<Game> list = new LinkedList<>(); for(int i = 1; i <= N; i++) { StringTokenizer st = new StringTokenizer(br.readLine()); int[] arr = { Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()) }; int sumMax = Integer.MIN_VALUE; for(int j = 0; j < bitMask.length; j++) { int k = bitMask[j]; int tmpSum = 0; for(int l = 0; l < 5; l++) { if((k & 1 << l) > 0) { tmpSum += arr[l]; } } sumMax = sumMax < (tmpSum % 10) ? (tmpSum % 10) : sumMax; } list.add(new Game(i, sumMax)); } Collections.sort(list); bw.write(Integer.toString(list.get(0).seq)); br.close(); bw.flush(); bw.close(); } }
5장중 3장을 고르는 경우의 수는 작고 입력이 고정되어 있으므로 BitMask를 이용하였다.
BitMask를 이용 시 5개중 3개를 뽑는 경우의 수를 계산하지 않고
00000의 5칸에서 7(00111), 11(01011), 13(01101), 14(01110), 19(10011), 21(10101), 22(10110), 25(11001), 26(11010), 28(11100)를 미리 저장해 둔 다음 해당 경우의 수 만큼 돌면서 1을 << 1씩(1칸씩 이동)하면서 해당되는 인덱스만 더해서 합산값의 최대값을 저장한다.
'PS > Java' 카테고리의 다른 글
[백준] 19939. 박 터뜨리기 (0) 2022.05.16 [백준] 1251. 단어 나누기 (0) 2022.04.06 [백준] 16435. 스네이크버드 (0) 2022.04.06 [백준] 11170. 0의 개수 (0) 2022.04.06 [백준] 14467. 소가 길을 건너간 이유 1 (0) 2022.04.06