[SW Expert Academy] - (D4)7829. 보물왕 태혁

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWtInr3auH0DFASy&categoryId=AWtInr3auH0DFASy&categoryType=CODE

SW Expert Academy

코드

import java.util.Arrays;
import java.util.Scanner;

public class Solution {

	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		
		int t = scan.nextInt();
		for(int tc=1; tc<=t; tc++) {
			int p = scan.nextInt();	// 약수의 갯수
			int[] arr = new int[p];
			int ans = 0;
			
			for(int i=0; i<p; i++) {
				arr[i] = scan.nextInt();
			}
			
			Arrays.sort(arr);
			
			/*
			 * 약수의 갯수가 1개면 그 수는 제곱수이고, 
			 * 1개가 아니면 가장 작은값 * 가장 큰값이 구하는 수가 됨.
			 */
			if(arr.length == 1) {
				ans = arr[0] * arr[0];
			} else {
				ans = arr[0] * arr[arr.length-1];
			}
			
			System.out.println("#" + tc + " " + ans);
		}
		
		scan.close();
	}

}

풀이

배열을 정렬하고 mix * max 하면 되는 간단한 문제이다.

4, 9, 16 등 제곱수의 경우 약수가 1개(1, n제외)이므로 

배열의 길이가 1이면 그 값을 제곱하면 된다.