글 읽기 - [Swift] 시간초과의 해결방법이 있을까요?

whdvy3 2년 전 0

같은 로직으로 파이썬으로 작성시 합격인거같은데, Swift로 통과할 방법이 있을까요?

import Foundation

class FileIO {
    @inline(__always) private var buffer: [UInt8] = Array(FileHandle.standardInput.readDataToEndOfFile()) + [0], byteIdx = 0
    
    @inline(__always) private func readByte() -> UInt8 {
        defer { byteIdx += 1 }
        return buffer.withUnsafeBufferPointer { $0[byteIdx] }
    }
    
    @inline(__always) func readInt() -> Int {
        var number = 0, byte = readByte(), isNegative = false
        while byte == 10 || byte == 32 { byte = readByte() }
        if byte == 45 { byte = readByte(); isNegative = true }
        while 48...57 ~= byte { number = number * 10 + Int(byte - 48); byte = readByte() }
        return number * (isNegative ? -1 : 1)
    }
    
    @inline(__always) func readStirngSum() -> Int {
        var byte = readByte()
        while byte == 10 || byte == 32 { byte = readByte() }
        var sum = Int(byte)
        while byte != 10 && byte != 32 && byte != 0 { byte = readByte(); sum += Int(byte) }
        return sum - Int(byte)
    }
    
    @inline(__always) private func write(_ output: String) {
        FileHandle.standardOutput.write(output.data(using: .utf8)!)
    }
}
let fileIO = FileIO()
let n = fileIO.readInt()
let m = fileIO.readInt()
let r = fileIO.readInt()

var arr = Array(repeating: Array(repeating: 0, count: m), count: n)
for i in 0..<n{
    for j in 0..<m{
        arr[i][j] = fileIO.readInt()
    }
}

//그룹의 개수 구하기
var groupCnt = min(n, m) / 2
//회전횟수
for _ in 0..<r{
    //그룹 개수만큼 반복

    for i in 0..<groupCnt{
        var x = i
        var y = i
        var curValue = arr[x][y]
        var prevValue = 0
        for j in (i + 1)..<(n - i){ //좌
            x = j
            prevValue = arr[x][y]
            arr[x][y] = curValue
            curValue = prevValue
        }
   
        for j in (i + 1)..<(m - i){ //하
            y = j
            prevValue = arr[x][y]
            arr[x][y] = curValue
            curValue = prevValue
        }
        for j in stride(from: n - i - 2, through: i, by: -1){//우
            x = j
            prevValue = arr[x][y]
            arr[x][y] = curValue
            curValue = prevValue
        }
        for j in stride(from: m - i - 2, through: i, by: -1){ //상
            y = j
            prevValue = arr[x][y]
            arr[x][y] = curValue
            curValue = prevValue
        }

    }
}

for i in 0..<n{
    print(arr[i].map{String($0)}.joined(separator: " "))
}

wapas 2년 전 3

해당 코드의 시간 복잡도는 O(NMR)이고, 말씀해주신대로 다른 언어(C++, Python)은 해당 로직으로 통과가 가능합니다.

Swift로 통과하려면 시간복잡도가 O(NM)인 로직으로 풀어야 합니다.

힌트를 드리자면, 각 그룹을 1차원 배열로 바꿔보는 것 입니다.

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

[Swift] 시간초과의 해결방법이 있을까요?

whdvy3 2년 전 0

wapas 2년 전 3

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말