[Swift] 2021 KAKAO INTERNSHIP 거리두기 확인하기

 

Problem

 

코딩테스트 연습 - 거리두기 확인하기

---

Solution

 

1. 응시자의 위치를 파악한다.

 

func getSeatLocation(places:[String],xMap:inout [[String]],applicants:inout [Seat]) {
    for (y,place) in places.enumerated() {
        xMap.append(place.map{String($0)})
        for (x,p) in place.map({String($0)}).enumerated() {
            if p == "P" {
                applicants.append((x,y))
            }
        }
    }
}

 

2. 맨하튼 거리가 2이하일 경우 3가지 조건으로 검사한다.

 

    2-1. x위치가 같을 경우

 

     y위치가 더 적은 응시자의 y에 +1을 해준 곳이 "X"인지 확인한다.

 

    2-2. y위치가 같을 경우

 

     x위치가 더 적은 응시자의 x에 +1을 해준 곳이 "X"인지 확인하다.

 

    2-3.x와 y모두 다를 경우

 

     x위치가 더 적은 곳은 x에 +1을 해준 곳이 "X"인지 확인하고

 

     x위치가 더 큰 곳은 x -1을 해준 곳이 "X"인지 확인한다.

 

func getManhattanDistance(p1:Seat,p2:Seat) -> Int {
    return abs(p1.0-p2.0) + abs(p1.1-p2.1)
}

func isKeepDistance(p1:Seat,p2:Seat,xMap:inout [[String]]) -> Bool {
    let minX = p1.0 < p2.0 ? p1 : p2
    let minY = p1.1 < p2.1 ? p1 : p2
    if p1.0 == p2.0 {
        return xMap[minY.1+1][p1.0] == "X"
    }else if p1.1 == p2.1 {
        return xMap[p1.1][minX.0+1] == "X"
    }else {
        if p1.0 < p2.0 {
            return xMap[p1.1][p1.0+1] == "X" && xMap[p2.1][p2.0-1] == "X"
        }else {
            return xMap[p2.1][p2.0+1] == "X" && xMap[p1.1][p1.0-1] == "X"
        }
    }
}

 

3. 모든 대기실을 검사하고 조건에 맞게 0과 1을 넣은 뒤 반환한다.

 

func solution(_ places:[[String]]) -> [Int] {
    var answer:[Int] = []
    for place in places {
        var check:Bool = false
        var applicants:[Seat] = []
        var xMap:[[String]] = []
        getSeatLocation(places: place,xMap: &xMap,applicants: &applicants)
        outer:while !applicants.isEmpty {
            let first = applicants.removeFirst()
            for applicant in applicants {
                if getManhattanDistance(p1: first, p2: applicant) <= 2 {
                    if !isKeepDistance(p1: first, p2: applicant,xMap: &xMap) {
                        answer.append(0)
                        check = true
                        break outer
                    }
                }
            }
        }
        if !check {
            answer.append(1)
        }
    }
    return answer
}

---

Source Code