[백준(BOJ)] 3273 두 수의 합 - 스위프트(Swift)

[Silver III] 두 수의 합 - 3273

문제 링크

성능 요약

메모리: 76884 KB, 시간: 48 ms

분류

정렬, 두 포인터

문제 설명

n개의 서로 다른 양의 정수 a₁, a₂, ..., a_n으로 이루어진 수열이 있다. a_i의 값은 1보다 크거나 같고, 1000000보다 작거나 같은 자연수이다. 자연수 x가 주어졌을 때, a_i + a_j = x (1 ≤ i < j ≤ n)을 만족하는 (a_i, a_j)쌍의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 수열의 크기 n이 주어진다. 다음 줄에는 수열에 포함되는 수가 주어진다. 셋째 줄에는 x가 주어진다. (1 ≤ n ≤ 100000, 1 ≤ x ≤ 2000000)

출력

문제의 조건을 만족하는 쌍의 개수를 출력한다.

 

풀이

투 포인터 문제로, 배열을 정렬하고 시작과 끝점을 지정합니다.

그리고 해당 값들을 더하면서 만약 더한 값이 같다면 양 옆의 점을 옮기고 정답을 +1해줍니다.

만약 더한 값이 작다면 시작점을 옮기고, 크다면 끝점을 옮겨주면 됩니다.

 

let n = Int(readLine()!)!
let material = readLine()!.split(separator: " ").map { Int($0)! }.sorted()
let x = Int(readLine()!)!
var (start, end, ans) = (0, material.count-1, 0)

for _ in 0..<n {

    if start >= end {
        break
    }

    if (material[start] + material[end]) == x {     // 값이 같을 때
        ans += 1
        start += 1
        end -= 1
    }

    else if (material[start] + material[end]) < x {     // 더한 값이 M보다 더 작을 때 -> start 옮기기
        start += 1
    }

    else if (material[start] + material[end] > x) {     // 더한 값이 M보다 더 클 때 -> end 옮기기
        end -= 1
    }
}

print(ans)