LeetCode - #119 杨辉三角 II

105 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第29天,点击查看活动详情

前言

本题由于没有合适答案为以往遗留问题,最近有时间将以往遗留问题一一完善。

我们社区陆续会将顾毅(Netflix 增长黑客,《iOS 面试之道》作者,ACE 职业健身教练。)的 Swift 算法题题解整理为文字版以方便大家学习与阅读。

LeetCode 算法到目前我们已经更新到 116 期,我们会保持更新时间和进度(周一、周三、周五早上 9:00 发布),每期的内容不多,我们希望大家可以在上班路上阅读,长久积累会有很大提升。

不积跬步,无以至千里;不积小流,无以成江海,Swift社区 伴你前行。如果大家有建议和意见欢迎在文末留言,我们会尽力满足大家的需求。

难度水平:中等

1. 描述

给定一个非负索引 rowIndex,返回「杨辉三角」的第 rowIndex 行。

在「杨辉三角」中,每个数是它左上方和右上方的数的和。

2. 示例

示例 1

输入: rowIndex = 3
输出: [1,3,3,1]

示例 2

输入: rowIndex = 0
输出: [1]

示例 3

输入: rowIndex = 1
输出: [1,1]

提示:

  • 0 <= rowIndex <= 33

3. 答案

题解 1

class Solution {
   func getRow(_ rowIndex: Int) -> [Int] {
       //滚动数组
       var arr:Array = Array(repeating: 1, count: (rowIndex+1))    
       for i in 0..<rowIndex
       {
           // 第一个元素 不需要计算 所以j从 1 开始 
           // i+1 代表最后一个元素:  < (i+1) 表示最后一个元素不用计算
           for j in (1 ..< (i+1)).reversed() 
           {
               // 从后往前计算, 防止覆盖
               arr[j] =  arr[j] + arr[j - 1]
           }        
       }
        return arr
   }
}

题解 2

class Solution {
   func getRow(_ rowIndex: Int) -> [Int] {
       var memo: [[Int]] = [[1], [1, 1]]
       if rowIndex <= 1 { return memo[rowIndex] }
       return calculateRows(memo, rowIndex)
   }

   func calculateRows (_ memo: [[Int]], _ rowIndex: Int) -> [Int] {
       if memo.count-1 == rowIndex { return memo[rowIndex] }
       var dp = memo
       var prevArr = dp[dp.count-1]
       var arr = Array(repeating: 0, count: prevArr.count+1)
       arr[0] = 1
       arr[arr.count-1] = 1

       for i in 1..<arr.count-1 {
           arr[i] = prevArr[i-1]+prevArr[i]
       }

       dp.append(arr)
       return calculateRows(dp, rowIndex)
   }
}

点击前往 LeetCode 练习

关于我们

我们是由 Swift 爱好者共同维护,我们会分享以 Swift 实战、SwiftUI、Swift 基础为核心的技术内容,也整理收集优秀的学习资料。