처음 문제풀이때는 그냥 생각나는대로 코드를 짜보았습니다만, 테스트케이스에 따라 2~3초가 걸리는 경우가 있었기에 다른 방법을 찾아보았습니다. Math 클래스의 메소드를 사용하면 1초 내로 해결 가능했습니다.
class Solution {
public int solution(int n) {
int answer = 2;
// 성능 개선 이전의 풀이
// for(int i=1;i<=n;i++) {
// if(i*i == n) {
// answer = 1;
// }
// }
if(Math.sqrt(n) % 1 == 0) {
answer = 1;
}
return answer;
}
}