Swift - 让你的[数组]溜得飞起来

由于Swfit强大的扩展特性,搞得现在很少自定义过自己的类型。一般都是直接XXX+extension.swift 文件了。然后几乎直接点语法...其实这样也简洁我们的代码,相当于把具体的实现逻辑交给那个接口,返回自己想要的东西就足够了。
下面就让我们小小扩展下数组 让它飞一下 ︿( ̄︶ ̄)︿
前面的分解数组语法参考objc的一篇博客
一些小扩展命名参考Haskell语言
虫子来自Haskell趣味玩法
extension Array {
    
    ///SELF -> 头 + [尾巴部分]
    var decomposeHead: (head: Element, tail: [Element])? {
        return (count > 0) ? (self[0], Array(self[1..<count])) : nil
    }
    
    ///SELF -> [头部分] + 尾巴
    var decomposeTail: (initt: [Element], last:Element)? {
        return (count > 0 ) ? (Array(self[0..<count-1]), self[self.count-1]) : nil
    }
    
    ///返回首个元素 Swift本身已实现方法有 .first
    var head: Element? {
        return decomposeHead?.head
    }
    
    //返回首个元素 Swift本身已实现方法有 .last
    var last: Element? {
        return decomposeTail?.last
    }
    
    //返回数组的尾部,即除掉头部之后的部分
    var tail: [Element]? {
        return decomposeHead?.tail
    }
    
    ///返回数组的前部,即除掉最后一个元素
    var initt: [Element]? {
        return decomposeTail?.initt
    }

    ///返回随机元素
    var randomElement: Element? {
        guard count > 0 else { return nil }
        return self[Int(arc4random_uniform(UInt32(count)))]
    }        

    ///数组 -> 转字典 按照原序号当键值,这里默认键为整数类型
    var transIntoDict: [Int: Element]? {
        
        var listDict = [Int: Element]()
        self.enumerate().forEach {
            listDict[$0] = $1
        }
        return listDict
    }
    
    ///判断是否存在该元素 Swift本身已实现方法有 contains
    func elem<T : Equatable>(elem: T) -> Bool {
        let filtered = self.filter {$0 as? T == elem}
        return filtered.count > 0
    }
    
    ///判断元素的总数
    func elemNum<T : Equatable>(elem: T) -> Int {
        let filtered = self.filter {$0 as? T == elem}
        return filtered.count
    }

    /// 随意转移一个元素到头部 如果存在相同 同时转移全部到前面
    func customHeads<T : Equatable>(elem: T) -> [Element] {

        let head = self.filter { $0 as? T == elem }
        let tail =  self.filter { $0 as? T != elem }
        return head + tail
    }
    
    /// 随意转移一个元素到头部 如果存在相同 只转移遇到最先那个
    func customHead<T : Equatable>(elem: T) -> [Element] {
        
        var isHasElem = false
        var newArr = [Element]()
        self.forEach { currElem in
            if (currElem as? T) == elem && isHasElem == false {
                isHasElem = true
                newArr.insert(currElem, atIndex: 0)
            } else {
                newArr.append(currElem)
            }
        }
        return newArr
    }
    
    /*
      * |--⬆️值类型------华丽丽分界线-----⬇️引用类型--|
    */
    
    //take 返回前面几个元素-如果大于数组即返回全部
    mutating func take(index: Int) -> [Element]? {
        guard self.count >= index else {
            return self
        }
        self = Array(self[0..<index])
        return self
    }
    
    ///drop 删除前面几个元素-如果大于数组个数-默认全部删除
    mutating func drop(index: Int) -> [Element]? {
        guard self.count >= index else {
            self = []
            return self
        }
        self = Array(self[index..<self.count])
        return self
    }

    /*
     * 打乱数组
    */
    mutating func shuffle() {
        for i in (0..<self.count).reverse() {
            let ix1 = i
            let ix2 = Int(arc4random_uniform(UInt32(i+1)))
            (self[ix1], self[ix2]) =  (self[ix2], self[ix1])
        }
    }
}

使用例子:

var list = [1,2,3,4,5,6]

list.head              // = 1
list.last             // = 6
list.tail            // = [2, 3, 4, 5, 6]
list.initt          // = [1, 2, 3, 4, 5]
list.transIntoDict // = [4: 5, 5: 6, 2: 3, 0: 1, 1: 2, 3: 4]

list.elem(1) // = true
list.elem(8) // = false

list.take(5) // -> [1, 2, 3, 4, 5]
list.drop(2) // -> [3, 4, 5]

其他基本特性参考哥这篇:Swift 类型 - 数组

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

推荐阅读更多精彩内容

  • 发现 关注 消息 iOS 第三方库、插件、知名博客总结 作者大灰狼的小绵羊哥哥关注 2017.06.26 09:4...
    肇东周阅读 12,068评论 4 62
  • 有很多年没有这种认真的观看一个长纪录片的体验了。我更愿意把它叫做“纪录片”,没有戏剧效果、没有那么多好听的概念,纯...
    阳光左侧阅读 269评论 2 2
  • 最近有朋友问我初学者该如何学习彩铅? 我的建议是从你自己喜欢的,简单的小物体开始画,练熟了再画复杂的。其实自己接触...
    多恩美术工作室阅读 3,754评论 4 74
  • 20170328linux的shell和内核 对于linux来说,shell就是命令行,或者桌面版的窗口命令行。 ...
    田旭1阅读 1,236评论 0 0
  • 爱的是这四合院屋檐下围成的一方天空。云从晨时的千颗珍珠模样,渐变为团团圆圆的棉絮,再接着是层层片片的鱼鳞,到了中午...
    达瓦枫溪阅读 209评论 0 0