백준 [1253] kotlin

문제

N개의 수 중에서 어떤 수가 다른 수 두 개의 합으로 나타낼 수 있다면 그 수를 “좋다(GOOD)”고 한다.

N개의 수가 주어지면 그 중에서 좋은 수의 개수는 몇 개인지 출력하라.

수의 위치가 다르면 값이 같아도 다른 수이다.

입력

첫째 줄에는 수의 개수 N(1 ≤ N ≤ 2,000), 두 번째 줄에는 i번째 수를 나타내는 Ai가 N개 주어진다. (|Ai| ≤ 1,000,000,000, Ai는 정수)

출력

좋은 수의 개수를 첫 번째 줄에 출력한다.

코드

var N = 0
var ans = 0
lateinit var arr : List<Int>
fun main() {
    N = readln().toInt()
    arr = readLine()!!.split(" ").map { it.toInt() }
    arr = arr.sorted()
    for (i in 0 until N)
    {
        var left = 0
        var right = N-1
        while(left < right)
        {
            if(arr[left] + arr[right] == arr[i]){
                if(left != i && right != i){
                    ans++
                    break
                }
                else if(left == i) left++
                else if(right == i) right--
            }
            else if(arr[left] + arr[right] < arr[i]) left++
            else right--
        }
    }
    println(ans)
}