Valid PAN format. Swift 3. HackerRank

SwiftRegularExpressionExample.txt

import Foundation

// read the integer n
let n = Int(readLine()!)!

var arr = Array(repeating: "", count: n)
for index in 0...n-1 {
    arr[index] = String(readLine()!)
}

func isPANValid(code : String) -> Bool {
    let regex = try! NSRegularExpression(pattern: "^[A-Z]{5}[0-9]{4}[A-Z]{1}$")
    return regex.numberOfMatches(in: code, range: NSRange(location:0, length: code.characters.count)) == 1
}

for string in arr {
    if isPANValid(code: string) {
        print("YES")
    } else {
        print("NO")
    }
}