1977번: 완전제곱수

문제 출처 : https://www.acmicpc.net/problem/1977

1977번: 완전제곱수

언어 : Kotlin

 

문제 설명 :

M과 N이 주어질 때 M이상 N이하의 자연수 중 완전제곱수인 것을 모두 골라 그 합을 구하고 그 중 최솟값을 찾는 프로그램을 작성하시오. 예를 들어 M=60, N=100인 경우 60이상 100이하의 자연수 중 완전제곱수는 64, 81, 100 이렇게 총 3개가 있으므로 그 합은 245가 되고 이 중 최솟값은 64가 된다.

첫째 줄에 M이, 둘째 줄에 N이 주어진다. M과 N은 10000이하의 자연수이며 M은 N보다 같거나 작다.

• M이상 N이하의 자연수 중 완전제곱수인 것을 모두 찾아 첫째 줄에 그 합을, 둘째 줄에 그 중 최솟값을 출력한다. 단, M이상 N이하의 자연수 중 완전제곱수가 없을 경우는 첫째 줄에 -1을 출력한다.

 

제한 사항 :

• 시간 제한 : 2초
• 메모리 제한 : 128MB

 

입출력 예 :

입력	출력
60 100	245 64
75 80	-1

풀이 : 

import java.io.BufferedWriter
import java.io.OutputStreamWriter
import kotlin.math.sqrt

fun main() = with(System.`in`.bufferedReader()) {
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val m = readLine().toInt()
    val n = readLine().toInt()
    val arr = (m .. n).map { if (sqrt(it.toDouble()).toInt() * sqrt(it.toDouble()).toInt() == it) it else 0 }.filter { it != 0 }

    if (arr.isEmpty()) bw.write("-1") else bw.appendLine("${arr.sumOf { it }}").appendLine("${arr.minOf { it }}")
    bw.flush()
    bw.close()
}