Swift 中枚举的基本用法

关注

原创 2022-03-15 18:00:09 3.8k 阅读 AI 写同款· GEO 优化 ›

【经典算法】全排列问题(暴搜枚举) 题目简述给定一个整数n，将数字1~n排成一排，按照字典序将所有的排列输出。输入格式共一行，包含一个整数n。输出格式按字典序输出所有排列方案，每个方案占一行。数据范围 1≤n≤9 输入样例： 3 输出样例： 1 2 3 1 3 2 2 1 3 2 3 1 3 1 2 3 2 1 暴力枚举上图是递归树。我们要枚举，递归都可以画一棵树。当我们到最后一层时候，即该排列所有位都枚举完，填上数字了，就输出并返回。否则我们要枚举该排列的枚举的位上应该填什么数。经典的回溯算法，暴力枚举。时间阅读详情

一、枚举语法

//枚举的语法:名称需要首字母大写
enum Direction {
    case north
    case south
    case west
    case east
}
//多个成员值可以出现在同一行中，要用逗号隔开
enum Direction2 {
    case north, south, west, east
}

注意：与 OC 不同的是，Swift 中的枚举在创建时不分配默认整数值。也就是说 north、south、west、east 不等于隐式的 0、1、2 和 3

二、使用 Switch 语句来匹配枚举值

enum Direction {
    case north, south, west, east
}
//使用 Switch 语句来匹配每一个单独的枚举值
let directionToHead = Direction.south
switch directionToHead {
case .north:
    print("north")
case .south:
    print("south")
case .west:
    print("west")
case .east:
    print("east")
}

三、遍历枚举的 case

可以通过在枚举名字后面写 : CaseIterable 来允许枚举被遍历。Swift 会暴露一个包含对应枚举类型所有情况的集合名为 allCases。

//在枚举名字: 后面写 CaseIterable,Swift 会暴露 allCase 集合
enum Direction: CaseIterable {
    case north, south, west, east
}
//遍历枚举
for dir in Direction.allCases {
    print(dir)
}
print(Direction.allCases.count)   //获取枚举所有情况的个数

四、关联值

可以定义 Swift 枚举来存储任意给定类型的关联值，如果需要的话不同枚举成员关联值的类型可以不同

//关联值
enum BarCode {
    case upc(Int, Int, Int, Int)
    case qrCode(String)
}
var productBarCode = BarCode.upc(1, 2, 3, 4)
print(productBarCode)    //打印: upc(1, 2, 3, 4)
productBarCode = .qrCode("ABSDEFGHIJK")
print(productBarCode)    //打印: qrCode("ABSDEFGHIJK")

//可以在 case 后面写 let 或者 在括号里面 写 let 定义相匹配的变量
switch productBarCode {
case .upc(let first, let second, let third, let fourth):
    print("UPC: \(first), \(second), \(third), \(fourth)")
case let .qrCode(productCode):
    print("QR code: \(productCode)")
}

五、原始值

枚举成员可以用相同类型的默认值预先填充(称为原始值)

注意：如果声明的时候不指定枚举类型就没有rawValue属性(包括关联值)

//枚举也可以有原始值
enum ControlCharacter: Character {  //如果没有声明枚举的类型,是没有 rawValue 属性可以访问的
    case tab = "\t"
    case linefeed = "\n"
    case carrigeReturn = "\r"
}
print(ControlCharacter.linefeed.rawValue)  //输出结果是换行了

六、预设原始值

在操作存储整数或字符串原始值枚举的时候，不必显式地给每一个成员都分配一个原始值。当没有分配时，Swift 将会自动为你分配值

//预设原始值:预设west等于5,那么后面的值依次往后面加1
enum point: Int {
    case west = 5
    case north   // 6
    case east    // 7
    case south   // 8
}
print(point.south.rawValue)   //输出结果为:8

enum Point2: Int {
    case west
    case north
    case east = 5
    case south
}
//值得注意的是: 如果是给 east 赋值为 5, west 的值是从 0, north 的值是 1
print(Point2.south.rawValue)   // 6
print(Point2.west.rawValue)    // 0
print(Point2.north.rawValue)   // 1
//预设原始值: 必须要定义枚举的类型
enum point: String {
    case west
    case north
    case east
    case south
}
print(point.south.rawValue)   //输出结果为:south,此时可以不写 rawValue,它们输出的值是一样的

七、从原始值初始化

如果用原始值类型来定义一个枚举，那么枚举就会自动收到一个可以接受原始值类型的值初始化器(叫做 rawValue的形式参数)然后返回一个枚举成员或者 nil 。你可以使用这个初始化器来尝试创建一个枚举的新实例

//预设原始值:预设west等于5,那么后面的值依次往后面加1
enum point: Int {
    case west = 5
    case north   // 6
    case east    // 7
    case south   // 8
}
print(point.south.rawValue)   //输出结果为:8

//也可以使用rawValue来创建或者初始化一个枚举值,最终可以获取枚举的具体值
let direction2 = point(rawValue: 9)
print(direction2 ?? 0)   //如果direction2没有值的话,默认就是0

八、递归枚举

递归枚举是拥有另一个枚举作为枚举成员关联值的枚举。当编译器操作递归枚举时必须插入间接寻址层。可以在声明枚举成员之前使用 indirect关键字来明确它是递归的

//如说表达式(5 + 4) * 2 在乘法右侧有一个数但有其他表达式在乘法的左侧
indirect enum Expression {
    case number(Int)
    case add(Expression, Expression)
    case multiply(Expression, Expression)
}
let four = Expression.number(4)
let five = Expression.number(5)
let sum = Expression.add(four, five)
print(four)
print(sum)
let multiply = Expression.multiply(sum, Expression.number(2))
print(multiply)
//在函数中根据switch语句获取匹配的case条件
func eval(_ expression: Expression) -> Int {
    //使用值绑定的方式
    switch expression {
    case .number(let value):
        return value
    case .add(let first, let second):
        return eval(first) + eval(second)
    case .multiply(let first, let second):
        return eval(first) * eval(second)
    }
}
print(eval(multiply))   // 18