NSArray и NSMutableArray - это классы NSFoundation, ими можно пользоваться в Swift'e , но это не очень хорошая практика
Лучше все же использовать чисто свифтовские вещи
https://developer.apple.com/library/mac/documentat...
Модифицировать примерно так же, как Вы написали
