swift 自定义下标

下标相信大家都很熟悉了,在绝大多数语言中使用下标来读写类似数组或者是字典这样的数据结构的做法,似乎已经是业界标准。在 Swift 中,Array 和 Dictionary 当然也实现了下标读写:

var arr = [1,2,3]
arr[2]            // 3
arr[2] = 4        // arr = [1,2,4]

var dic = ["cat":"meow", "goat":"mie"]
dic["cat"]          // {Some "meow"}
dic["cat"] = "miao" // dic = ["cat":"miao", "goat":"mie"]”

数组的话没有什么好多说的,但是字典需要注意,我们通过下标访问得到的结果是一个 Optional 的值。这是很容易理解的,因为你不能限制下标访问时的输入值,对于数组来说如果越界了只好直接给你脸色让你崩掉,但是对于字典,查询不到是很正常的一件事情。对此,在 Swift 中我们有更好的处理方式,那就是返回 nil 告诉你没有要找的东西。

作为一门代表了先进生产力的语言,Swift 是允许我们自定义下标的。这不仅包含了对自己写的类型进行下标自定义,也包括了对那些已经支持下标访问的类型 (没错就是 Array 和 Dictionay) 进行扩展。我们重点来看看向已有类型添加下标访问的情况吧,比如说 Array。很容易就可以在 Swift 的定义文件 (在 Xcode 中通过 Cmd + 单击任意一个 Swift 内建的类型或者函数就可以访问到) 里,找到 Array 已经支持的下标访问类型:

subscript (index: Int) -> T
subscript (subRange: Range<Int>) -> Slice<T>

共有两种,它们分别接受单个 Int 类型的序号和一个表明范围的 Range<Int>,作为对应,返回值也分别是单个元素和一组对应输入返回的元素。

于是我们发现一个挺郁闷的问题,那就是我们很难一次性取出某几个特定位置的元素,比如在一个数组内,我想取出 index 为 0, 2, 3 的元素的时候,现有的体系就会比较吃力。我们很可能会要去枚举数组,然后在循环里判断是否是我们想要的位置。其实这里有更好的做法,比如说可以实现一个接受数组作为下标输入的读取方法:

extension Array {
    subscript(input: [Int]) -> ArraySlice<Element> {
        get {
            var result = ArraySlice<Element>()
            for i in input {
                assert(i < self.count, "Index out of range")
                result.append(self[i])
            }
            return result
        }

        set {
            for (index,i) in input.enumerated() {
                assert(i < self.count, "Index out of range")
                self[i] = newValue[index]
            }
        }
    }
}

这样,我们的 Array 的灵活性就大大增强了:

var arr = [1,2,3,4,5]
arr[[0,2,3]]            //[1,3,4]
arr[[0,2,3]] = [-1,-3,-4]
arr                     //[-1,2,-3,-4,5]

虽然我们在这里实现了下标为数组的版本,但是我并不推荐使用这样的形式。如果阅读过参数列表一节的读者也许会想为什么在这里我们不使用看起来更优雅的参数列表的方式,也就是 subscript(input: Int...) 的形式。不论从易用性还是可读性上来说,参数列表的形式会更好。但是存在一个问题,那就是在只有一个输入参数的时候参数列表会导致和现有的定义冲突,有兴趣的读者不妨试试看。当然,我们完全可以使用至少两个参数的的参数列表形式来避免这个冲突,即定义形如 subscript(first: Int, second: Int, others: Int...) 的下标方法:

extension Array {
    subscript(first: Int, second: Int, others: Int...) -> ArraySlice<Element> {
        get {
            var result = ArraySlice<Element>()
            result.append(self[first])
            result.append(self[second])
            for i in others {
                assert(i < self.count,"Index out of range")
                result.append(self[i])
            }
            return result
        }
        set {
            var input = [Int]()
            input.append(first)
            input.append(second)
            input.append(contentsOf: others)
            for (index, i) in input.enumerated() {
                assert(i < self.count,"Index out of range")
                self[i] = newValue[index]
            }
        }
    }
}

使用方法如下:

arr[1,2,3,4]               //[2, -3, -4, 5] 
arr[1,2,3,4] = [1,2,3,4]   //[-1, 1, 2, 3, 4] 
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,185评论 6 503
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,652评论 3 393
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 163,524评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,339评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,387评论 6 391
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,287评论 1 301
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,130评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,985评论 0 275
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,420评论 1 313
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,617评论 3 334
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,779评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,477评论 5 345
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,088评论 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,716评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,857评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,876评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,700评论 2 354

推荐阅读更多精彩内容