深圳某互联网证券面试题:双指针的使用

背景

面试深圳某互联网证券公司,算法题是给一个固定时间区间的股票价格,找出最佳买卖时机,使得收益最大,其中一天只能买卖操作一次。这里最佳方案是使用双指针方法。在学习字符串和数组的一些算法题时,发现常用的双指针方法可以解决很多问题,并且可以有效的降低时间复杂度。最早接触双指针方法是链表的快慢指针,用来判断一个单项链表是否有环。

这篇文章将接触到的经典双指针法用在一些字符串和数组的算法做一个简单的总结。

介绍

双指针有两种方案:

  • 前后指针,一个是数组的起始位置,一个是数组的结束位置
  • 快慢指针,一个快指针进行遍历,一个慢指针在某些特殊条件下移动

前后指针迭代

前后指针的一般步骤如下:

  1. 指定一个往右遍历的指针leftIndex和一个向左遍历的指针rightIndex
  2. 判断leftIndexrightIndex对应的元素关系(根据题目)
  3. 根据条件来更新leftIndexrightIndex
  4. 结束条件是leftIndex == rightIndex

实例1:升序数组两个数的和等于目标数

给定一个已按照升序排列的有序数组,找到两个数使得他们相加之后等于目标数。

函数应该返回这两个下标值index1和index2,其中index1必须小于index2

示例:

输入:numbers = [2, 5, 7, 11, 15], target = 9

输出:[1, 2]

注意:返回的下标值index1和index2不是从零开始

分析

题目给出数组是升序数组,所以我们可以使用前面介绍的前后指针遍历方法。

  • leftIndex的元素值和rightIndex的元素和相等与目标值时,代表找到了两个索引。

  • 如果小于目标值,则leftIndex的指针右移。

  • 如果大于目标值,则rightIndex的指针左移。

  • 如果leftIndex等于rightIndex,则没有找到。

如下图所示:

双指针_01.png
代码
// swift实现
func twoSum(_ numbers: [Int], _ target: Int) -> [Int] {
    // 如果数组是空,则直接返回[-1]
    if numbers.isEmpty {
        return [-1]
    }
    
    var left = numbers.startIndex
    var right = numbers.endIndex - 1
    while left < right {
        let sum = numbers[left] + numbers[right]
        if sum == target {
            return [left + 1, right + 1]    // 找到直接返回
        } else if sum > target {
            right -= 1
        } else {
            left += 1
        }
    }
    
    return [-1]
}
复杂度

时间复杂度:O(N)

空间复杂度:O(1)

快慢指针遍历

另外一种常见的方法就是使用快慢指针遍历,一般步骤如下:

  1. 定义一个慢指针slowIndex和快指针quickIndex都指向数组的第一个位置
  2. quickIndex每次移动一步,而slowIndex在满足某个特定条件的时候移动一步
  3. 当前quickIndex结束之后,slownIndex指向的就是新数组的最后一位

实例1

给定一个含有 n 个正整数的数组和一个正整数 s ,找出该数组中满足其和 ≥ s 的长度最小的子数组,并返回其长度如果不存在符合条件的子数组,返回 0。

示例:

输入:s = 7, nums = [2, 3, 1, 2, 4, 3]

输出:2

解释:子数组 [4,3] 是该条件下的长度最小的子数组。

分析

定义两个指针slowIndexquickIndex,快指针每移动一步,则计算慢指针到快指针之和sum,并记下子数组的长度minLength

如果sum的值大于或者等于目标值sslowIndex移动一步,继续计算sum的和,更新minLength

如果sum小于目标值s,则quickIndex继续移动

如下图所示:

双指针_02.png
代码
// swift实现
func minSubArrayLen(_ s: Int, _ nums: [Int]) -> Int {
  // 数组如果是空直接返回为0
    if nums.isEmpty {
        return 0
    }
        
    var slow = nums.startIndex
    var minLength = Int.max
        
    var sum = 0
    for i in 0..<nums.count {
        sum = sum + nums[i]
        while sum >= s {
            minLength = min(minLength, i - slow + 1) // 找到最小的长度
            sum = sum - nums[slow]
            slow += 1
        }
    }
        
    return (minLength == Int.max ? 0 : minLength)
}
复杂度

时间复杂度:O(N)

空间复杂度:O(1)

练习

习题1:翻转字符串

编写一个函数,其作用是将输入的字符串反转过来。输入字符串以字符数组 char[] 的形式给出。

不要给另外的数组分配额外的空间,你必须原地修改输入数组**、使用 O(1) 的额外空间解决这一问题。

你可以假设数组中的所有字符都是 ASCII码表中的可打印字符。

代码

// swift实现
func reverseString(_ s: inout [Character]) {
    var leftIndex = s.startIndex
    var rightIndex = s.endIndex - 1
    while leftIndex < rightIndex {
        let tempCharacter = s[leftIndex]
        s[leftIndex] = s[rightIndex]
        s[rightIndex] = tempCharacter
            
        leftIndex += 1
        rightIndex -= 1
    }
}

习题2:最大连续1的个数

给定一个二进制数组, 计算其中最大连续1的个数。

示例 1:

输入: [1,1,0,1,1,1]
输出: 3
解释: 开头的两位和最后的三位都是连续1,所以最大连续1的个数是 3.

注意:

  • 输入的数组只包含 01
  • 输入数组的长度是正整数,且不超过 10,000。

代码

func findMaxConsecutiveOnes(_ nums: [Int]) -> Int {
  var ret = 0
  var count = 0
  for num in nums {
    if num != 1 {
      ret = max(count, ret)
      count = 0
    } else {
      count += 1
    }
  }
  
  return max(count, ret)
}

买卖股票的最佳时机

回到开头,发现原来这个题目也是刷题网站(leetcode)上的。

给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。

如果你最多只允许完成一笔交易(即买入和卖出一支股票一次),设计一个算法来计算你所能获取的最大利润。

注意:你不能在买入股票前卖出股票。

示例 1:

输入: [7,1,5,3,6,4]
输出: 5
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格;同时,你不能在买入前卖出股票。

示例 2:

输入: [7,6,4,3,1]]

输出: 0

解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。

代码

func maxProfit(_ prices: [Int]) -> Int {
    var slowIndex = prices.startIndex
    var maxDiff = Int.min
    for i in 0..<prices.count { // 快指针一个一个遍历
        let diff = prices[i] - prices[slowIndex]    // 收益
        if diff > 0 {
          maxDiff = max(maxDiff, diff)  // 最大收益
        } else {
            slowIndex = i   // 移动慢指针到i
        }
    }

    return max(maxDiff, 0)
}

总结

双指针方法,可以解决大部分数组和字符串的问题。深刻了解原理,对解这方面的题目还是一个比较重要的思考方向。

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