BOJ-11650,11651 좌표 정렬하기 Swift

문제

2차원 평면 위의 점 N개가 주어진다. 좌표를 x좌표가 증가하는 순으로, x좌표가 같으면 y좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.

 

입력

첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.

 

출력

첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.

 

풀이

1.  x좌표를 오름차순으로 정렬한다.

2. x좌표가 같을 경우 y좌표를 오름차순으로 정렬한다.

 

- 2중 for문으로 index 값을 확인해서 insert 로 넣은 결과 시간초과가 떴다. 시간복잡도 O(n^2)

func test() { 	// 시간초과 N^2
    guard let count = Int(readLine()!) else { return }
    var nums = [[Int]]()

    for i in 0..<count {
        guard let input = readLine()?.split(separator: " ") else { return }
        var arr = input.map{ Int(String($0))!}
        var index = 0

        if nums.isEmpty {
            nums.append(arr)
            continue
        }

        for j in 0..<nums.count {
            let x1 = nums[j][0]
            let y1 = nums[j][1]
            let x2 = arr[0]
            let y2 = arr[1]

            if x1 < x2 {
                if j == nums.count - 1 {
                    index = nums.count
                }
                continue
            } else if x1 == x2 {
                if y1 < y2 {
                    index = j+1
                } else {
                    index = j
                }
                break
            } else {
                index = j
                break
            }
        }
        nums.insert(arr, at: index)
    }

    for j in nums {
        print(j.map{ String($0)}.joined(separator: " "))
    }

}

test()

 

시간제한 1초 였던 문제라 시간 복잡도를 좀  줄였어야했다.

그래서 array 함수중 sort를 이용했다. sort의 시간복잡도는 O(nlogn)으로 2중 배열보다 빨랐다.

하지만 sort 함수를 사용해서 2차원배열 y좌표 접근하는 방법은 아직 미숙했다.

< 11650 >

func test() {
    guard let input = Int(readLine()!) else {return}
    var nums = [[Int]]()
    
    for i in 0..<input {
        guard let inputs = readLine()?.split(separator: " ") else {return}
        var arr = inputs.map{ Int(String($0))!}
        nums.append(arr)
    }
    nums = nums.sorted{ if $0[0] == $1[0] {
        return $0[1] < $1[1]
    } else {
        return $0[0] < $1[0]
    }}
    
    for j in nums {
        print(j.map{ String($0)}.joined(separator: " "))
    }
}
test()

마지막 출력할때 사용된 map함수의 시간복잡도는 O(n)이므로 결국 n + nlogn 이므로 n^2 보다 빠르니 시간초과가 뜨지않았다

 

11651 은 정렬기준만 다르기때문에 위 코드와 큰 차이는 없었다

< 11651 >

import Foundation

func test() {
    guard let input = Int(readLine()!) else {return}
    var nums = [[Int]]()
    
    for i in 0..<input {
        guard let inputs = readLine()?.split(separator: " ") else {return}
        var arr = inputs.map{ Int(String($0))!}
        nums.append(arr)
    }
    nums = nums.sorted{ if $0[1] == $1[1] {
        return $0[0] < $1[0]
    } else {
        return $0[1] < $1[1]
    }}
    
    for j in nums {
        print(j.map{ String($0)}.joined(separator: " "))
    }
}
test()