add: 下一个更大元素 III

Author: yi-ge

README.md

@@ -456,6 +456,10 @@ TypeScript / JavaScript 基础算法、数据结构练习，包含 LeetCode 或
 
 ### 数学
 
+- [下一个更大元素 III](src/math/next-greater-element-iii.ts)  [数学, 双指针, 字符串]
+
+  - LeetCode 556. 下一个更大元素 III <https://leetcode.cn/problems/next-greater-element-iii>
+
 - [阶乘后的零](src/math/factorial-trailing-zeroes.js)
 
   - LeetCode 172. 阶乘后的零 <https://leetcode-cn.com/problems/factorial-trailing-zeroes/submissions/>

src/math/next-greater-element-iii.ts

@@ -0,0 +1,29 @@
+// 下一个更大元素 III
+// https://leetcode.cn/problems/next-greater-element-iii
+// 解题思路：参考 31. 下一个排列
+
+const reverse = (nums: string[], begin: number) => {
+  let i = begin, j = nums.length - 1
+  while (i < j) {
+    [nums[i], nums[j]] = [nums[j], nums[i]]
+    i++
+    j--
+  }
+}
+
+export function nextGreaterElement (n: number): number {
+  const numStrArr = [...n.toString()]
+  let i = numStrArr.length - 2
+  while (i >= 0 && numStrArr[i + 1] <= numStrArr[i]) {
+    i--
+  }
+
+  if (i < 0) return -1
+
+  let j = numStrArr.length - 1
+  while (j >= 0 && numStrArr[i] >= numStrArr[j]) { j-- }
+  ;[numStrArr[i], numStrArr[j]] = [numStrArr[j], numStrArr[i]]
+  reverse(numStrArr, i + 1)
+  const res = Number(numStrArr.join(''))
+  return res > 2 ** 31 - 1 ? -1 : res
+}

test/math/next-greater-element-iii.test.ts

@@ -0,0 +1,14 @@
+import { nextGreaterElement } from '../../src/math/next-greater-element-iii'
+
+test('下一个更大元素 III', () => {
+  // 示例 1：
+  // 输入：n = 12
+  // 输出：21
+
+  // 示例 2：
+  // 输入：n = 21
+  // 输出：-1
+
+  expect(nextGreaterElement(12)).toBe(21)
+  expect(nextGreaterElement(21)).toBe(-1)
+})