프로그래머스 Level 1
🔒 정수 제곱근 판별
임의의 양의 정수 n에 대해, n이 어떤 양의 정수 x의 제곱인지 아닌지 판단하려 합니다.
n이 양의 정수 x의 제곱이라면 x+1의 제곱을 리턴하고, n이 양의 정수 x의 제곱이 아니라면 -1을 리턴하는 함수를 완성하세요.
n | return |
---|---|
121 | 144 |
3 | -1 |
입출력 예#1
121은 양의 정수 11의 제곱이므로, (11+1)를 제곱한 144를 리턴합니다.
입출력 예#2
3은 양의 정수의 제곱이 아니므로, -1을 리턴합니다.
class Solution {
public long solution(long n) {
long answer = 0;
double sqrt = Math.sqrt(n);
if(sqrt % 1 == 0) {
answer = (long) Math.pow(sqrt + 1, 2);
}
else answer = -1;
return answer;
}
}
Math 함수를 이용하여 쉽게 풀 수 있는 문제였다.
java.lang.Math 클래스는 수학 계산에 사용할 수 있는 메소드를 제공하고 메소드는 모두 정적이므로 Import나 Math 클래스 선언 없이 바로 사용이 가능하다.
Math 클래스에서
sqrt() 함수는 제곱근을 구할 수 있고
pow() 함수는 제곱을 구할 수 있다.
단, double 형식이 들어가야하기 때문에 double로 계산하다가 마지막에 long으로 형변환 해준 것이다.
sqrt(x)
x에 제곱근을 구하고 싶은 수를 넣으면 된다.
pow(x, y)
x를 y만큼 곱해준다.
그래서 감자는 x를 sqrt + 1로 y를 2로 하여 n의 제곱근 + 1을 제곱했다.
sqrt를 1로 나눈 나머지가 0인지 체크한 이유는 sqrt가 정수인지 확인하기 위함이다.
현재 double로 선언되어있기 때문에 실수도 가능하기 때문에 소수점이 있다면 정수가 아니므로 -1을 반환하게 했다.
class Solution {
public long solution(long n) {
if (Math.pow((int)Math.sqrt(n), 2) == n) {
return (long) Math.pow(Math.sqrt(n) + 1, 2);
}
return -1;
}
}
엥? 이렇게 짧게도 가능하구나...
제곱근을 int로 형변환을 한 후 다시 제곱했을 때 다시 n이 된다면
제곱근 + 1 의 제곱을 바로 리턴한다.
위에 조건이 맞지 않는다면 -1을 반환...
코드는 어떻게 생각하고 받아들여야 짧고 간결하게 짤 수 있는 걸까?
단순히 많이 풀어보고 다른 사람 코드도 많이 보면 사고력이 늘어서 간결한 코드가 가능해지는 걸까?
일단 많이 풀어보자!!
아직 1레벨도 많이 남았다...ㅠ