p64の練習問題(4.1)

4.1 sum関数のコードを書いてみる

答えを見ずに書いたらこんな感じ

func sum(array: [Int]) -> Int {
    var arr = array
    
    if arr.count < 2 {
        return arr[0]
    }
    return arr[ arr.count - 1 ] + sum(array: Array(arr.dropLast(1)))
    
}

print(sum(array: [1,2,3,4]))

解答を参考にするともう少しスマートにかけたかな？

def sum(list):
   if list == []:
      return 0
   return list[0] + sum(list[1:])

func sum(array: [Int]) -> Int {
    var arr = array
    guard arr.isEmpty else {
        return arr[0] + sum(array: Array(arr.dropFirst(1)))
    }
    
    return 0
}

print(sum(array: [1,2,3,4]))

詰まったころ

下記のように書くと

func sum(array: [Int]) -> Int {
    var arr = array
    
    if arr.count < 2 {
        return arr[0]
    }
    return arr[ arr.count - 1 ] + sum(array: arr.dropLast(1))
    
}

print(sum(array: [1,2,3,4]))

エラーメッセージが出た

main.swift:7:50: error: cannot convert value of type 'ArraySlice<Int>' to expected argument type '[Int]'
    return arr[ arr.count - 1 ] + sum(array: arr.dropLast(1))

公式ドキュメントを参照するとdropLastはSubstringをかえすらしい。arrayを返すのだと思ってた。

func dropLast(_ k: Int) -> Substring

対策

Array(arr.dropLast(1))とかいて型を変換した

4.2 リスト内の要素の個数を数える再帰関数を書いてみる

func count(array: [Int]) -> Int {
    var arr = array
    
    guard arr.isEmpty else {
        return 1 + count(array: Array(arr.dropFirst(1)))
    }
    return 0
}

print(count(array:  [1,2,3,4,7]))

4.3 リストに含まれている数字のうち、最も大きな数字を見つけ出す

これも答えを見ずに考えたらこんな感じ　ちょっと汚い

func largeNumber(array: [Int]) -> Int {
    var arr = array
    
    if arr.count < 2 {
        return arr[0]
    }
    
    if arr[ arr.count - 1 ] == largeNumber(array: Array(arr.dropLast(1))) {
        return arr[ arr.count - 1 ]
    } else if arr[ arr.count - 1 ] < largeNumber(array: Array(arr.dropLast(1))) {
        return largeNumber(array: Array(arr.dropLast(1)))
    } else {
        return arr[ arr.count - 1 ]
    }
}

print("largeNumber(array: [1,2])",largeNumber(array: [1,9,6]))

教本の答えはこんな感じ

def max(list):
   if len(list) == 2:
      return list[0] if list[0] > list[1] else list[1]
   sub_max = max(list[1:])
   return list[0] if list[0] > sub_max else sub_max

Swiftで書いてみるとこんな感じ
唐突にarr.count == 2って書いてあったらちょっとわかりにくいかも・・・・

func largeNumber(array: [Int]) -> Int {
    var arr = array
    
    if arr.count == 2 {
        return arr[0] > arr[1] ? arr[0] : arr[1]
    }
    var subMax = largeNumber(array:Array(arr.dropLast(1)))
    return arr[0] > subMax ? arr[0] : subMax
}

print("largeNumber(array: [1,2])",largeNumber(array: [1,9,6]))

参考文献

Swift の Array やら ArraySlice やらポインタの変換まとめ

Apple docment popLast()

Apple docment dropLast(_:)