728x90
def solution(n):
return (int(n ** 0.5) + 1) ** 2 if n == int(n ** 0.5) ** 2 else -1삼항 연산자를 이용하면 한줄로 적을 수 있다.
728x90
'알고리즘 > 프로그래머스' 카테고리의 다른 글
| 프로그래머스 연습문제: 정수 내림차순으로 배열하기 (java) (0) | 2022.10.11 |
|---|---|
| 프로그래머스 연습문제: 문자열 내 p와 y의 개수 (python) (1) | 2022.10.11 |
| 프로그래머스 연습문제: 자연수 뒤집어 배열로 만들기 (python) (0) | 2022.10.11 |
| 프로그래머스 연습문제: 자릿수 더하기 (java, python) (0) | 2022.10.11 |
| 프로그래머스 연습문제: 약수의 합 (java) (0) | 2022.10.11 |