Как посчитать отдельные элементы в массиве?

Question

Я хочу сосчитать отдельные элементы в следующем массиве:

let b = [
    1, 2, 3,
    [4,5,6],
    [
        [7,8],
        [9,0]
    ]
]

, и мне удалось подсчитать следующий массив:

let a = [
    [1,2,3],
    [4,5],
    [6,7,8,9]
]

с помощью следующего кода:

protocol DeepCountable {
    var deepCount: Int {get}
}

// conditional conformance
extension Array: DeepCountable where Element: DeepCountable {
    var deepCount: Int {
        return self.reduce(0){$0 + $1.deepCount}
    }
}

extension Int: DeepCountable {
    var deepCount: Int { return 1 }
}

print(a.deepCount)      // 9

Как сделать то же самое с массивом b?

print( b.deepCount )

Answer 1

Тип массива b равен [Any]. Any не DeepCountable. Теперь добавьте deepcount свойство в Array

extension Array: DeepCountable {
    var deepCount: Int {
        return self.compactMap({ $0 as? DeepCountable }).reduce(0, { $0 + $1.deepCount })
    }
}

let a = [[1,2,3],[4,5],[6,7,8,9]]
print(a.deepCount)//9
let b = [1, 2, 3,[4,5,6],[[7,8],[9,0]]] as [Any]
print(b.deepCount)//10
let c = [1,2,"a","b"] as [Any]
print(c.deepCount)//2

Answer 2

Вот общее решение, которое может рассчитывать различные типы

extension Array {
    func countType<T>(_ elemType: T.Type) -> Int {
        var count = 0
        for elem in self {            
            if let _ = elem as? T {
                count += 1
            } else if let arr = elem as? [Any] {
                count += arr.countType(elemType)
            }
        }
        return count
    }
}

Использование

let b = [1, 2, 3,[4,5,6],[[7,8],[9,0]]] as [Any]
let c = [1,2,"a","b"] as [Any]

print("Int Count    \(b.countType(Int.self))")
print("Array Count  \(b.countType([Int].self))")
print("Int Count    \(c.countType(Int.self))")
print("String Count \(c.countType(String.self))")
print("Array Count  \(c.countType([Int].self))")

урожайность

Int Count    10  
Array Count  3  
Int Count    2  
String Count 2  
Array Count  0

Answer 3

Следующий код является результатом, который я сделал из перечисленных выше взносов:

protocol DeepCountable {
  var deepCount: Int { get }
}

// now any Array is DeepCountable
extension Array: DeepCountable {
  var deepCount: Int {
    return compactMap{ $0 as? DeepCountable }.reduce(0){ $0 + $1.deepCount }
  }
}

extension Int: DeepCountable {
  var deepCount: Int { return 1 }
}

// test
let a = [[1,2,3],[4,5],[6,7,8,9]]
let b = [1, 2, 3,[4,5,6],[[7,8],[9,0]]] as [Any]
let c = [1,2,"a","b"] as [Any]
let d = [1,2,3,[4,5],[6,7,8,[9, 0]]] as [Any]

print( a.deepCount )  //  9
print( b.deepCount )  // 10
print( c.deepCount )  //  2
print( d.deepCount )  // 10

Этот код должен работать со Swift 5, я пробовал его на Online Swfit Playground , и он отлично работает.

Answer 4

Сначала нужно сгруппировать массив в один, а затем считать этот массив.

let b = [
    1, 2, 3,
    [4,5,6],
    [
        [7,8],
        [9,0]
    ]
]
let flatb = b.flatMap({ $0 })
print(flatb.count)