[Swift] 튜플 배열, 이차원 배열, 객체 배열 정렬하기

알고리즘을 풀다보면 이차원 배열이나 튜플,객체를 원하는 조건대로 정렬하고 싶을 때가 있습니다.

그럴 때 어떻게 해야하는지 알아보도록 하겠습니다.

정석적으로 조건문 이용해서 정렬하는 것과 밑에 쪽에서는 튜플을 이용해서 간단하게 한 줄로 정렬 하는 방법이 있습니다.

객체 정렬

struct Movie {
    let title: String
    let year: Int
    let director: String
}

let movieList = [
    Movie(title: "Star Wars", year: 1898, director: "Leorge Mucas"),
    Movie(title: "A Star", year: 1922, director: "Michal Platpus"),
    Movie(title: "Be yourself", year: 1700, director: "Coey Vamn"),
    Movie(title: "The big bang", year: 2011, director: "Rom Natas"),
    Movie(title: "The big bang", year: 1922, director: "Adra Kngr"),
    Movie(title: "Finding Nemoy", year: 2022, director: "Raxip"),
    Movie(title: "The big bang", year: 2011, director: "Jos Klimbr")
]

한 가지 기준으로 정렬하기

제목 오름차 순 한 뒤 제목이 같으면 나머지는 정렬 전 배열 순서 기준

 

let sortedMovieList = movieList.sorted { // Mark 1
    $0.title < $1.title
}

for movie in sortedMovieList {
    print(movie)
}

두 가지 기준으로 정렬하기

제목 오름차순 → 제목이 같으면 연도 오름차순 → 연도가 같으면 정렬 전 배열 순서 기준

let sortedMovieList = movieList.sorted {
    if $0.title == $1.title { // Mark 1
        return $0.year < $1.year // Mark 2
    }
    return $0.title < $1.title // Mark 3
}

for movie in sortedMovieList {
    print(movie)
}

세가지 기준으로 정렬하기

제목 오름차순 → 제목이 같으면 연도 오름차순 → 연도가 같으면 감독 오름차순

let sortedMovieList = movieList.sorted {
    if $0.title == $1.title {
        if $0.year == $1.year {
            return $0.director < $1.director
        }
        return $0.year < $1.year
    }
    return $0.title < $1.title
}

for movie in sortedMovieList {
    print(movie)
}

조건문이 아닌 튜플 이용해서 정렬하기

튜플끼리 값의 비교를 할 때는, 앞 요소끼리 각각 순서대로 크기 비교를 할 수 있다.

let sortedMovieList = movieList.sorted {
    ($0.title, $0.year, $0.director) < ($1.title, $1.year, $1.director)
}

우선순위 순서 바꾸기

let sortedMovieList = movieList.sorted {
    ($0.year, $0.title, $0.director) < ($1.year, $1.title, $1.director)
}

하나만 역순

let sortedMovieList = movieList.sorted {
    ($0.title, $1.year, $0.director) < ($1.title, $0.year, $1.director)
}

for movie in sortedMovieList {
    print(movie)
}

이차원 배열 정렬

let unsorted = [ [1,2] , [12,4] ,[12,1] ]
let sorted = unsorted.sorted {
    ($0[0] , $0[1] ) < ( $1[0] , $1[1] )
}
print( sorted) // output [[1, 2], [12, 1], [12, 4]]

let unsorted = [ [1,2] , [12,4] ,[12,1] ]
let sorted = unsorted.sorted {
    ($0[0] , $1[1] ) < ( $1[0] , $0[1] )
}
print(sorted) // [[1, 2], [12, 4], [12, 1]]

튜플 배열 정렬

let unsorted = [ (1,2) , (12,4) ,(12,1) ]
let sorted = unsorted.sorted {
    ($0.0 , $0.1 ) < ( $1.0 , $1.1 )
}
print( sorted) // output [(1, 2), (12, 1), (12, 4)]

let unsorted = [ (1,2) , (12,4) ,(12,1) ]
let sorted = unsorted.sorted {
    ($0.0 , $1.1 ) < ( $1.0 , $0.1 )
}
print( sorted) // output [(1, 2), (12, 4), (12, 1)]

Reference

[Swift] Nested Array (2차원 배열)에서 튜플을 이용해 다중 정렬하는 법 (sort, tuple, Closure)

[Swift] Nested Array (2차원 배열)에서 튜플을 이용해 다중 정렬하는 법 (sort, tuple, Closure)

Using Tuples to do Sorting Operations in Swift - Holy Swift

Using Tuples to do Sorting Operations in Swift - Holy Swift

Sorting using tuples | Functional Swift

Sorting using tuples | Functional Swift