[Algorithm] 나머지 한 점 (ternary(?:), XOR(^), map)

🖥 문제

직사각형을 만드는 데 필요한 4개의 점 중 3개의 좌표가 주어질 때, 나머지 한 점의 좌표를 구하려고 합니다. 점 3개의 좌표가 들어있는 배열 v가 매개변수로 주어질 때, 직사각형을 만드는 데 필요한 나머지 한 점의 좌표를 return 하도록 solution 함수를 완성해주세요. 단, 직사각형의 각 변은 x축, y축에 평행하며, 반드시 직사각형을 만들 수 있는 경우만 입력으로 주어집니다.

🤔 풀이 방법

[[1, 4], [3, 4], [3, 10], [1, 10]] -> 1, y : 2개, 3, y : 2개...
4개의 좌표로 직사각형을 만들면 각 점을 이루는 좌표는 2개씩 중복된다. 이를 이용하면 문제를 간단하게 풀 수 있다.

if로 구현

if v[0][0] == v[1][0] { ans[0] = v[2][0] }

• 첫 번째 점과 두 번째 점의 x좌표가 같으면 세 번째 점의 x좌표가 답이 된다.

ternary로 구현

if를 이용하여 구현한 것을 ternary로 간단하게 구현할 수 있다.

• a ? b : c : a가 참이면 b를 실행하고 거짓이면 c를 실행한다.

XOR 사용하여 구현

출처 : https://swift.org

ans[0] = ans[0]^arrayV.first!

• ans[0]에 v의 x좌표를 넣고 이를 XOR연산을 해준다.

map으로 구현

v.map{$0[0]}.reduce(0, ^)

• x좌표의 첫 번째 값을 뽑아 이를 ^(XOR) 연산해 준다.

👩🏻‍💻 Swift Code

if로 구현

import Foundation

func solution(_ v: [[Int]]) -> [Int]
{
    var ans = [Int](repeating: 0, count: 2)

    if v[0][0] == v[1][0] {
        ans[0] = v[2][0]
    } else if v[0][0] == v[2][0] {
        ans[0] = v[1][0]
    } else if v[1][0] == v[2][0] {
        ans[0] = v[0][0]
    }

    if v[0][1] == v[1][1] {
        ans[1] = v[2][1]
    } else if v[0][1] == v[2][1] {
        ans[1] = v[1][1]
    } else if v[1][1] == v[2][1] {
        ans[1] = v[0][1]
    }

    return ans
}

ternary로 구현

import Foundation

func solution(_ v: [[Int]]) -> [Int]
{
    var ans = [Int](repeating: 0, count: 2)

    ans[0] = v[0][0] == v[1][0] ? v[2][0] : (v[0][0] == v[2][0] ? v[1][0] : v[0][0])
    ans[1] = v[0][1] == v[1][1] ? v[2][1] : (v[0][1] == v[2][1] ? v[1][1] : v[0][1])

    return ans
}

xor 사용하여 구현

import Foundation

func solution(_ v: [[Int]]) -> [Int]
{
    var ans = [Int](repeating: 0, count: 2)

    for arrayV in v {
        ans[0] = ans[0]^arrayV.first!
        ans[1] = ans[1]^arrayV.last!
    }

    return ans
}

map으로 구현

import Foundation

func solution(_ v: [[Int]]) -> [Int]
{
    var ans = [Int](repeating: 0, count: 2)

    ans[0] = v.map{$0[0]}.reduce(0, ^)
    ans[1] = v.map{$0[1]}.reduce(0, ^)

    return ans
}

 

알고리즘 문제 해설 - 나머지 한 점