[Java] SWEA 7829번 보물왕 태혁

SW Expert Academy

---

풀이

입력받는 수들은 약수이므로 한개만 입력받을 경우 그 수를 제곱하면 원래의 수가 되고 한개 이상일 경우에는 가장 큰 수와 가장 작은 수를 곱해주면 원래의 수가 됨을 이용하여 해결했습니다.

---

코드

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

public class Solution {
	public static int N, answer;
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder("");
		StringTokenizer st;
		
		int T = Integer.parseInt(br.readLine());
		for (int tc = 1; tc <= T; ++tc) {
			N = Integer.parseInt(br.readLine());
			
			st = new StringTokenizer(br.readLine(), " ");
			
			int max = Integer.MIN_VALUE;
			int min = Integer.MAX_VALUE;
			int num;
			
			if (st.countTokens() == 1) {
				num = Integer.parseInt(st.nextToken());
				answer = num * num;
			}
			else {
				for (int i = 0; i < N; ++i) {
					num = Integer.parseInt(st.nextToken());
					max = Math.max(max, num);
					min = Math.min(min, num);
				}
				
				answer = max * min;
			}
			
			sb.append("#").append(tc).append(" ").append(answer).append("\n");
		}
		
		System.out.println(sb);
	}
}

---

pyo7410/Algorithm