[백준/Swift] 17140 이차원 배열과 연산

📝 문제 정보

• 문제 출처: 백준 온라인 저지
• 문제 링크: 17140 이차원 배열과 연산
• 제출 언어: Swift
• 알고리즘 분류:
  • 구현
  • 정렬
  • 시뮬레이션

---

🖊️ 풀이

복잡하지만 문제를 잘 읽고 분석하면 구현할 수 있는 문제이다. 

주어진 조건

• 행의 개수가 더 많거나 같으면 R 연산, 열의 개수가 더 많은 C 연산 실행
• 연산은 각각의 수가 몇 번 나왔는지 구하여서 오름차순 정렬하여 배열에 넣는다
• (우선순위 : 등장 횟수 -> 수) ( 넣는 순서 : 수 , 등장 횟수)
• 크기가 커진 곳은 0 으로 채운다, 수 연산 시 0은 무시한다
• 행 또는 열의 크기가 100 넘어가면 100개 제외한 나머지는 버린다
• A[r][c] == k가 되기 위한 연산의 최소 시간 출력 100초가 지나도 안되면 -1 출력

접근 방법

1. Dictionary로 등장 횟수 세기

연산은 Dictionary에 해당 원소가 등장한 횟수를 저장했다.

Swift는 Dictionary 원소에 접근할 때 항상 optional 값으로 반환한다. 

즉, Dictionary[key] 형태로 접근했을 때 value가 있는 경우 optioanl(value)값, value가 없는 경우 nil값을 반환한다.

아래의 코드와 같이 삼항 연산자로 해당 key의 value값이 nil일 경우 1, 아닐 경우 1을 더해줘서 원소의 개수를 count하였다.

dic[num] == nil ? (dic[num] = 1) : (dic[num] = dic[num]! + 1)

2. dictionary 정렬하기

클로저를 매개변수로 넣어서 dictionary를 정렬해주었다.

이 또한, 삼항 연산자를 이용하여 우선순위( 등장 횟수 -> 수 )에 맞게 정렬했다.

key가 수이고 value가 등장 횟수이므로 등장횟수가 같다면 수를 오름차순으로, 같지 않다면 등장 횟수를 오름차순으로 정렬했다.

dic.sorted(by: { $0.value == $1.value ? $0.key < $1.key : $0.value < $1.value})

3. C 연산

R 연산은 2차원 배열을 다루던 방식으로 익숙하게 수행할 수 있었지만, C 연산은 열 기준이므로 세로로 봐야했기에 복잡했다.

그래서, tempArr를 만들어 정렬된 결과 배열을 R 연산과 같이 가로로 구한 다음, tempArr의 행과 열을 바꾸어 주었다.

ex)
// 1 2 3
// 4 5 6
->
// 1 4
// 2 5
// 3 6

---

👩🏻‍💻 전체 코드

let n = readLine()!.split(separator: " ").map{Int(String($0))!}
let (r, c, k) = (n[0], n[1] ,n[2])
var A: [[Int]] = []
for _ in 1...3 {
    A.append(readLine()!.split(separator: " ").map{Int(String($0))!})
}

func operation() -> Int {
    var timer = 0
    while timer <= 100 {
        // 배열의 크기가 원하는 r, c값보다 작아질 경우 index out 방지
        if r-1 < A.count && c-1 < A[0].count {
            if A[r-1][c-1] == k { return timer }
        }
        // 행의 개수가 같거나 더 많을 경우 R 연산 실행
        if A[0].count <= A.count {
            rOperation()
        }
        // 열의 개수가 더 클경우 C 연산 실행
        else {
            cOperation()
        }
        timer += 1
    }
    return -1
}

// R 연산
func rOperation() {
    var maxLength = 0
    for i in 0..<A.count {
        var dic: [Int:Int] = [:] // 개수 count 딕셔너리
        for num in A[i] {
            if num == 0 { continue } // 0인 경우 개수를 세지 않는다
            dic[num] == nil ? (dic[num] = 1) : (dic[num] = dic[num]! + 1)
        }
        A[i] = []
        // 새로운 배열에 해당 조건으로 정렬한 딕셔너리를 돌며 수, 등장 횟수 순으로 넣어준다
        for (key, value) in dic.sorted(by: { $0.value == $1.value ? $0.key < $1.key : $0.value < $1.value}) {
            if A[i].count >= 100 { break } // 100개 이상일 경우 중단
            A[i].append(contentsOf: [key, value])
        }
        maxLength = max(maxLength, A[i].count)
    }
    // 빈 공간을 0으로 채워주기
    for i in 0..<A.count {
        A[i].append(contentsOf: Array(repeating: 0, count: maxLength - A[i].count))
    }
}
// C 연산
func cOperation() {
    var maxLength = 0
    var tempArr: [[Int]] = Array(repeating: [], count: A[0].count)
    
    for i in 0..<A[0].count {
        var dic: [Int:Int] = [:]
        for j in 0..<A.count {
            if A[j][i] == 0 { continue }
            dic[A[j][i]] == nil ? (dic[A[j][i]] = 1) : (dic[A[j][i]]! = dic[A[j][i]]! + 1)
        }
        // tempArr에 R 연산과 유사하게 하나의 열을 계산한 후, 행 형태로 넣어준다
        for (key, value) in dic.sorted(by: { $0.value == $1.value ? $0.key < $1.key : $0.value < $1.value}) {
            tempArr[i].append(contentsOf: [key, value])
        }
        maxLength = max(maxLength, tempArr[i].count)
    }
    for i in 0..<tempArr.count {
        tempArr[i].append(contentsOf: Array(repeating: 0, count: maxLength - tempArr[i].count))
    }
    
    // 행 -> 열 회전
    A = Array(repeating: [], count: tempArr[0].count)
    for i in 0..<tempArr.count {
        for j in 0..<tempArr[0].count {
            A[j].append(tempArr[i][j])
        }
    }
}
print(operation())