配列を特定要素数ずつの細かい配列に分割する方法

Question

Swift3を使って、配列を特定要素数ずつの細かい配列に分割する方法を教えてください。

let myClasses = ["abc", "def", "ghi", "jkl", "mno"]

// この配列を２要素ずつに分割して３つの別々の配列に分割したい！
// この例ではString配列になっていますが、他のクラスでも使えるような形にしたいです。

どなたかアドバイスいただけますと幸いです。

Answer 1

寝ながらスマホという環境なのでコードは提示できないのですが、Swift3にはそのような機能を持つ関数は用意されていないみたいです。自分で実装する必要があります。

実装の方針としては、まず二次元配列を作成し、そこにmyClasses配列の要素を代入（もしくは追加）していくという実装になると思います。
myClasses配列のインデックスをiとすると、このiを特定要素数で割ったときの商と余りが、二次元配列のインデックスになります。

実装できますでしょうか？

Answer 2

あんまり綺麗じゃないですが

extension Array {
    
    func split(num: Int) -> [[Element]] {
        
        func divide<T>(num: Int, divided: [[T]], source: [T] ) -> ([[T]], [T]) {
            
            if source.isEmpty {
                return (divided, [])
            }
            
            if source.count <= num {
                return (divided + [source], [])
            }
            
            return divide(num: num,
                          divided: divided + [Array<T>(source[0..<num])],
                          source: Array<T>(source.dropFirst(num)))
        }
        
        let (divided, _) = divide(num: num, divided: [], source: self)
        return divided
    }
}

let myClasses = ["abc", "def", "ghi", "jkl", "mno"]

let splited = myClasses.split(num: 2)

---

追記

あまりにもパフォーマンスが悪すぎるので書き直しました。
多分O(N)ぐらいになってます。
複雑になったので説明コメント多めです。

extension Array {
    
    /// 配列を要素数が num の配列の配列にする
    ///
    /// let array = [1, 2, 3, 4, 5]
    /// let splited = array.split(num: 2)
    /// print(splited)
    /// // [[1, 2], [3, 4], [5]]
    func split(num: Int) -> [[Element]] {
        
        /// 配列を半分に分けて前半後半それぞれをもう一度半分に分ける
        /// これを要素数が num 以下になるまで繰り返す
        
        let count = self.count
        
        // 要素数が num 以下の時
        if count <= num {
            
            return [self]
        }
        // 要素数が 2 * num 以下の時
        if count <= num * 2 {
            
            return [Array<Element>(self[0..<num])]
                + [Array<Element>(self[num...])]
        }
        
        // 前半の要素数が num の整数倍になるように分割点を求める
        let splitPoint = Int(count / num / 2) * num
        
        switch splitPoint {
            
        case num:
            // 前半の要素数が num の時
            
            return [Array<Element>(self[0..<num])]
                + Array<Element>(self[num..<count]).split(num: num)
            
        case (num...):
            // 前半の要素数が num 以上の時
            
            return Array<Element>(self[0..<splitPoint]).split(num: num)
                + Array<Element>(self[splitPoint..<count]).split(num: num)
            
        default:
            // ここには到達しない
            
            fatalError("Not reachable")
        }
    }
}

Comments

[MasakiHori]

すいみません。　これO(N^2)ぐらいありますね。　動くけど使い物にならないと思います。　忘れてください

[MasakiHori]

あまりにもひどかったのでパフォーマンスを向上させたものを追記しました。

Answer 3

まずはどのような愚直な方法でもよいので、実現できるコードを書くことが大事ではないでしょうか？配列を使うのでしょうから

(1) 配列に要素を追加するにはどうすればよい？
(2) 配列の要素数を調べるにはどうすればよい？
(3) 要素をひとつづつ処理するにはどうすればよい？

などなどを調べて書いてみると見えてくる気がします。

let a = ["a", "b", "c", "d", "e"]

func splitStringArray(_ a: [String], _ n: Int) -> [[String]] {
  var r: [[String]] = []
  var ia: [String] = []
  for e in a {
    if ia.count >= n {
      r.append(ia)
      ia = []
    }
    ia.append(e)
  }
  if ia.count > 0 {
    r.append(ia)
  }
  return r
}

print(splitStringArray(a, 2))  // [["a", "b"], ["c", "d"], ["e"]]

自分は、最初に挙げた(1)～(3)もうろ覚えというレベルなので、ググって書いてみました。おそらくはswiftに慣れた方ならもっとスマートに書けるのかも知れませんね。

しかし、不格好でもなんでもまずは動くものを書き、swiftの文法やクラスの使い方に慣れていった方がよいと思います。上記のようなものでもいいから書いてみているうちにだんだん苦労なく書けるようになっていくだろうと思います。自分が言語を学ぶ際にはそういうふうにして覚えていきました。最初のころにはずいぶんヘンテコなコードを書いたものですが、自分で色々書いたおかげで多少はましになった気がします。

---

なお、「String限定ではなくて任意の型を扱える」ように関数を書くには「ジェネリクス」というキーワードを調べると書き方がわかる気がします。「swift generics」で検索するとヒットしましたので。しかしそれを書くのは基本的な論理が組めてからでも遅くないと思います。