swift 对于ループの中でや继续突破を使用しているのを，过滤に书き换える

Posted 2021-05-19

let array = [0, 2, 0, 2, 2, 1, 2, 2]

var map = [Int]()
for i in array {
	if i == 2 {
		map.append(i)
	} else if i == 1 {
		break
	} else {
		continue
	}
}
print(map) // [2, 2, 2]

let array = [0, 2, 0, 2, 2, 1, 2, 2]

// flatMap({$0})を入れないと、コンパイルエラー
// AnySequenceとArrayで同じprefix()メソッドがあり、どちらを使用するのかが曖昧のため、コンパイルエラーになる
let map = array.prefix(while: {$0 != 1}).flatMap({$0}).filter( {$0 == 2} )
print(map) // [2, 2, 2]

// filterの中で何かして、結果に反映するかの判断するには、このようにすることができる
// つまり、continueを使用していたところは、return falseにすることができる
let map = array.prefix(while: {$0 != 1}).flatMap({$0}).filter( {
	if $0 == 2 {
		return true
	} else {
		return false
	}
} )
print(map) // [2, 2, 2]