首發於微信公衆號《前端成長記》,寫於 2019.12.15javascript
本文記錄刷題過程當中的整個思考過程,以供參考。主要內容涵蓋:前端
題目地址java
給定一個二叉樹,判斷它是不是高度平衡的二叉樹。node
本題中,一棵高度平衡二叉樹定義爲:算法
一個二叉樹每一個節點 的左右兩個子樹的高度差的絕對值不超過1。微信
示例 1:優化
給定二叉樹 [3,9,20,null,null,15,7]
3d
3 / \ 9 20 / \ 15 7
返回 true
。code
示例 2:blog
給定二叉樹 [1,2,2,3,3,null,null,4,4]
1 / \ 2 2 / \ 3 3 / \ 4 4
返回 false
。
咱們上一期作過經過遍歷求二叉樹的最大深度的題目,這題最粗暴的一個方案就是計算出每一個子樹的最大深度作高度判斷,很明顯,這個效率低下。咱們能夠經過改爲自底而上的方案,當中間過程不符合,則能夠跳出計算。
Ⅰ.計算子樹最大深度作判斷
代碼:
/** * @param {TreeNode} root * @return {boolean} */ var isBalanced = function(root) { if (root === null) return true function maxDepth (node) { if (node === null) return 0 const l = maxDepth(node.left) const r = maxDepth(node.right) return Math.max(l, r) + 1 } return Math.abs(maxDepth(root.left) - maxDepth(root.right)) <= 1 && isBalanced(root.left) && isBalanced(root.right) };
結果:
O(n^2)
Ⅱ.自底而上
代碼:
/** * @param {TreeNode} root * @return {boolean} */ var isBalanced = function(root) { function maxDepth (node) { if (node === null) return 0 const l = maxDepth(node.left) if (l === -1) return -1 const r = maxDepth(node.right) if (r === -1) return -1 return Math.abs(l - r) <= 1 ? Math.max(l, r) + 1 : -1 } return maxDepth(root) !== -1 };
結果:
O(n)
思路基本上都是這兩種,未發現方向不一樣的解法。
這裏很明顯,你們都是用深度遍從來解決問題,計算子樹深度會發現,有不少重複運算,因此不妨試試自底而上的方式,直接在計算高度過程當中就返回,也能夠叫作「提早阻斷」。因此,這道題建議是使用自底而上的方式來做答。
給定一個二叉樹,找出其最小深度。
最小深度是從根節點到最近葉子節點的最短路徑上的節點數量。
說明: 葉子節點是指沒有子節點的節點。
示例:
給定二叉樹 [3,9,20,null,null,15,7],
3 / \ 9 20 / \ 15 7
返回它的最小深度 2.
這道題很明顯自頂而下就能夠了,判斷每一個節點的子節點是否存在,不存在,則該路徑爲最短路徑。若是存在,就按深度的方式比較最小值。整體上來講,也能夠用以前求最大深度的幾種方式來做答。
Ⅰ.遞歸
代碼:
/** * @param {TreeNode} root * @return {number} */ var minDepth = function(root) { if (root === null) return 0 if (root.left === null && root.right === null) return 1 let res = Infinity if(root.left !== null) { res = Math.min(minDepth(root.left), res) } if(root.right !== null) { res = Math.min(minDepth(root.right), res) } return res + 1 };
結果:
O(n)
Ⅱ.利用棧迭代
代碼:
/** * @param {TreeNode} root * @return {number} */ var minDepth = function(root) { if (root === null) return 0 if (root.left === null && root.right === null) return 1 // 棧 let s = [{ node: root, dep: 1 }] let dep = Infinity while(s.length) { // 先進後出 var cur = s.pop() if (cur.node !== null) { let curDep = cur.dep if (cur.node.left === null && cur.node.right === null) { dep = Math.min(dep, curDep) } if (cur.node.left !== null) s.push({node: cur.node.left, dep: curDep + 1}) if (cur.node.right !== null) s.push({node: cur.node.right, dep: curDep + 1}) } } return dep };
結果:
O(n)
Ⅲ.利用隊列
代碼:
/** * @param {TreeNode} root * @return {number} */ var minDepth = function(root) { if (root === null) return 0 if (root.left === null && root.right === null) return 1 // 隊列 let s = [{ node: root, dep: 1 }] let dep = 0 while(s.length) { // 先進先出 var cur = s.shift() var node = cur.node dep = cur.dep if (node.left === null && node.right === null) break; if (node.left !== null) s.push({node: node.left, dep: dep + 1}) if (node.right !== null) s.push({node: node.right, dep: dep + 1}) } return dep };
結果:
O(n)
整體上而言分紅深度優先和廣度優先,最基本的就是遞歸和迭代了。沒有發現二叉樹相關題目的一些新奇解法。
很明顯能夠看出遞歸和利用棧迭代是深度優先,利用隊列是廣度優先。這裏自頂而下比較合適,只要找到葉子節點,直接就是最小深度了,能夠省去很多運算。
給定一個二叉樹和一個目標和,判斷該樹中是否存在根節點到葉子節點的路徑,這條路徑上全部節點值相加等於目標和。
說明: 葉子節點是指沒有子節點的節點。
示例:
給定以下二叉樹,以及目標和 sum = 22
,
5 / \ 4 8 / / \ 11 13 4 / \ \ 7 2 1
返回 true
, 由於存在目標和爲 22 的根節點到葉子節點的路徑 5->4->11->2
。
這道題個人想法是由於要找到葉子節點,因此深度優先更爲合適,這裏就使用前文的兩種方法:
Ⅰ.遞歸
代碼:
/** * @param {TreeNode} root * @param {number} sum * @return {boolean} */ var hasPathSum = function(root, sum) { if (root === null) return false // 剩餘須要的值 sum -= root.val if (root.left === null && root.right === null) { return sum === 0 } else { return hasPathSum(root.left, sum) || hasPathSum(root.right, sum) } };
結果:
O(n)
Ⅱ.迭代
代碼:
/** * @param {TreeNode} root * @param {number} sum * @return {boolean} */ var hasPathSum = function(root, sum) { if (root === null) return false // 棧 let stack = [{ node: root, remain: sum - root.val }] while(stack.length) { // 先進後出 var cur = stack.pop() var node = cur.node if (node.left === null && node.right === null && cur.remain === 0) return true if (node.left !== null) { stack.push({ node: node.left, remain: cur.remain - node.left.val }) } if (node.right !== null) { stack.push({ node: node.right, remain: cur.remain - node.right.val }) } } return false };
結果:
O(n)
這裏看到一個方案是採用後序遍歷,路徑長度由以前的棧改爲變量保存,可是這個在我看來沒有中序遍歷合適,感興趣的能夠 點此查閱 。另外仍是有選擇使用廣度優先,利用隊列來解的,這裏也算一個不一樣思路,就當作補充吧。
Ⅰ.利用隊列
代碼:
/** * @param {TreeNode} root * @param {number} sum * @return {boolean} */ var hasPathSum = function(root, sum) { if (root === null) return false // 隊列 let q = [{ node: root, sum: root.val }] while(q.length) { // 當前層元素的個數 for(let i = 0; i < q.length; i++) { let cur = q.shift() let node = cur.node if (node.left === null && node.right === null && cur.sum === sum) return true if (node.left !== null) { q.push({ node: node.left, sum: cur.sum + node.left.val}) } if (node.right !== null) { q.push({ node: node.right, sum: cur.sum + node.right.val}) } } } return false };
結果:
O(n)
給定一個非負整數 numRows,生成楊輝三角的前 numRows 行。
在楊輝三角中,每一個數是它左上方和右上方的數的和。
示例:
輸入: 5 輸出: [ [1], [1,1], [1,2,1], [1,3,3,1], [1,4,6,4,1] ]
這道題最笨的方案就是雙重循環,首尾爲1,其餘位爲 S(l)[n] = S(l-1)[n-1] + S(l-1)[n]
。固然這裏很明顯也能夠當作一個動態規劃問題來解答。
這裏有個坑,給的是索引,不是第 n 行
Ⅰ.動態規劃
代碼:
/** * @param {number} numRows * @return {number[][]} */ var generate = function(numRows) { let res = [] for(let i = 0; i < numRows; i++) { // 全部默認都填了1,能夠節省很多運算 res.push(new Array(i+1).fill(1)) // 第三行開始才須要修改 for(j = 1; j < i; j++) { res[i][j] = res[i-1][j] + res[i-1][j-1] } } return res };
結果:
O(n^2)
這裏看到兩個不一樣方向的,一個是遞歸,由於這題在遞歸卡片中,一個是二項式定理。
Ⅰ.遞歸
代碼:
/** * @param {number} numRows * @return {number[][]} */ var generate = function (numRows) { let res = [] function sub(row, numRows, arr) { let temp = [] if (row < numRows) { for (let i = 0; i <= row; i++) { if (row === 0) { temp.push(1) } else { let left = i - 1 >= 0 ? arr[row - 1][i - 1] : 0 let right = i < arr[row - 1].length ? arr[row - 1][i] : 0 temp.push(left + right) } } arr.push(temp) sub(++row, numRows, arr) return arr } else { return arr } } return sub(0, numRows, res) };
結果:
O(n^2)
Ⅱ.二項式定理
優點在於能夠直接計算第n行,用二項式定理公式計算。 (a+b)^n
一共有n+1項,每一項的係數對應楊輝三角的第 n 行。第 r 項的係數等於 組合數 C(n,r)
。
代碼:
/** * @param {number} numRows * @return {number[][]} */ var generate = function(numRows) { var res = []; /** * 組合數 * @param n * @param r * @returns {number} * @constructor */ function C(n, r) { if(n == 0) return 1; return F(n) / F(r) / F(n - r); } /** * 階乘 * @param n * @returns {number} * @constructor */ function F(n) { var s = 1; for(var i = 1;i <= n;i++) { s *= i; } return s; } for (var i = 0;i < numRows;i++){ res[i] = []; for (var j = 0;j < i + 1;j++){ res[i].push(C(i, j)); } } return res; };
結果:
O(n^2)
對於數學敏感的開發者,很容易就想到使用二項式定理。可是在我看來,找到了一個計算規則,就很容易想到使用動態規劃來解決問題,我也推薦使用動態規劃來生成楊輝三角。
給定一個非負索引 k,其中 k ≤ 33,返回楊輝三角的第 k 行。
在楊輝三角中,每一個數是它左上方和右上方的數的和。
示例:
輸入: 3 輸出: [1,3,3,1]
進階:
你能夠優化你的算法到 O(k) 空間複雜度嗎?
上面從他人解法中發現了二項式定理能夠直接求第 n 行。另外咱們也能夠發現個規律,第幾行實際上就有幾個數,且首尾爲1。固然也能夠使用動態規劃來做答。
Ⅰ.動態規劃
代碼:
/** * @param {number} rowIndex * @return {number[]} */ var getRow = function(rowIndex) { // rowIndex 是索引,0至關於第1行 if (rowIndex === 0) return [1] let res = [] for(let i = 0; i < rowIndex + 1; i++) { let temp = new Array(i+1).fill(1) // 第三行開始才須要修改 for(let j = 1; j < i; j++) { temp[j] = res[j - 1] + res[j] } res = temp } return res };
結果:
O(n^2)
Ⅱ.二項式定理
代碼:
/** * @param {number} rowIndex * @return {number[]} */ var getRow = function(rowIndex) { /** * 組合數 * @param n * @param r * @returns {number} * @constructor */ function C(n, r) { if(n == 0) return 1; return F(n) / F(r) / F(n - r); } /** * 階乘 * @param n * @returns {number} * @constructor */ function F(n) { var s = 1; for(var i = 1;i <= n;i++) { s *= i; } return s; } let res = [] // 由於是經過上一項計算,因此第1項的 n 爲0 for (var i = 0;i < rowIndex + 1;i++){ res.push(C(rowIndex, i)); } return res; };
結果:
O(n)
由於發現每行的對稱性,因此也能夠求一半後反轉複製便可。
Ⅰ.反轉複製
代碼:
/** * @param {number} rowIndex * @return {number[]} */ var getRow = function(rowIndex) { // rowIndex 是索引,0至關於第1行 if (rowIndex === 0) return [1] let res = [] for(let i = 0; i < rowIndex + 1; i++) { let temp = new Array(i+1).fill(1) // 第三行開始才須要修改 const mid = i >>> 1 for(let j = 1; j < i; j++) { if (j > mid) { temp[j] = temp[i - j] } else { temp[j] = res[j - 1] + res[j] } } res = temp } return res };
結果:
O(n^2)
其實更像一個數學問題,不斷地找出規律來節省運算,真是「學好數理化,走遍天下都不怕」。
(完)
本文爲原創文章,可能會更新知識點及修正錯誤,所以轉載請保留原出處,方便溯源,避免陳舊錯誤知識的誤導,同時有更好的閱讀體驗
若是能給您帶去些許幫助,歡迎 ⭐️star 或 ✏️ fork (轉載請註明出處:https://chenjiahao.xyz)