競技プログラミング
Swift

CodilityのTrainingを解いてみた その2

More than 1 year has passed since last update.

問題 OddOccurrencesInArray

  • 使用言語:Swift3

ソースコード

public func solution2(_ A : inout [Int], _ K : Int) -> [Int] {
        let aNum = A.count
        if aNum == 0 {
            return A
        }

        let validK = K % aNum
        var ansArray = [Int]()
        if validK == 0 {
            return A
        }

        for i in aNum - validK...aNum - 1  {
            ansArray.append(A[i])
        }
        for i in 0...aNum - validK - 1 {
            ansArray.append(A[i])
        }
        return ansArray
    }