swift swift中的点头统治系统

Posted 2021-05-19

//tree rule system

enum node {
    case a
    case b
    case c
    case d
    case e
    case f
}

//returns an array of next possible nodes
func getNextNodeRule(input:node) -> [node] {
    switch input {
    case .a:
        return [node.b, node.c, node.d]
    case .b:
        return [node.e, node.f, node.a]
    case .c:
        return [node.b, node.c, node.d]
    case .d:
        return [node.b]
    case .e:
        return [node.e, node.f]
    case .f:
        return [node.a, node.b, node.c]
    }
}