Python程式:找出修復方程所需的更正次數


假設我們有一個字串 s,它表示一個形如 x+y=z 的方程。我們需要找到需要新增到 s 中的最小數字位數,以便該方程成立。

因此,如果輸入類似於 s = '2+6=7',則輸出將為 2。

我們可以透過插入“1”和“2”將方程轉換為“21+6=27”。因此,所需的更正總數為 2。

為了解決這個問題,我們將遵循以下步驟:

  • 根據“+”字元將 s 分割成部分,將左部分放入 A,將右部分放入 rest

  • 根據“=”字元將 rest 分割成部分,將左部分放入 B,將右部分放入 C

  • 返回 dp(A 的大小 - 1, B 的大小 - 1, C 的大小 - 1, 0)

  • 定義一個函式 dp()。它將接收 i、j、k、carry 作為引數

  • 如果 i <= -1 且 j <= -1 且 k <= -1,則

    • 如果 carry 等於 0,則返回 0,否則返回 1

  • last1 := 如果 i >= 0,則 (A[i]),否則為 0

  • last2 := 如果 j >= 0,則 (B[j]),否則為 0

  • last3 := 如果 k >= 0,則 (C[k]),否則為 0

  • prefix1 := 如果 i >= 0,則 (A[從索引 0 到 i + 1]),否則為 0

  • prefix2 := 如果 j >= 0,則 (B[從索引 0 到 j + 1]),否則為 0

  • prefix3 := 如果 k >= 0,則 (C[從索引 0 到 k + 1]),否則為 0

  • 如果 i <= -1 且 j <= -1,則

    • rhs := prefix3 - carry

    • 如果 rhs <= 0,則

      • 返回 |rhs|

    • 如果 i 等於 -1 或 j 等於 -1,則

      • 返回字串 rhs 的大小

    • 否則,

      • 返回 False

    • 如果 k <= -1,則

      • 返回 str(prefix1 + prefix2 + carry) 的大小

    • ans := 無限大

    • carry2、lhs := 返回將 (carry + last1 + last2) 除以 10 得到的商和餘數

    • 如果 lhs 等於 last3,則

      • ans := dp(i - 1, j - 1, k - 1, carry2)

    • req := last3 - carry - last2

    • extra_zeros := 0 和 -1 - i 的最大值

    • 如果 req < 0,則 carry2 := 1,否則為 0

    • ans := ans 和 1 + extra_zeros + dp(-1、i、j - 1、k - 1、carry2 的最大值) 的最小值

    • req := last3 - carry - last1

    • extra_zeros := 0 和 -1 - j 的最大值

    • 如果 req < 0,則 carry2 := 1,否則為 0

    • ans = ans 和 1 + extra_zeros + dp(i - 1, -1 和 j 的最大值, k - 1, carry2) 的最小值

    • carry2、lhs := 返回將 (last1 + last2 + carry) 除以 10 得到的商和餘數

    • ans := ans 和 1 + dp(i - 1, j - 1, k, carry2) 的最小值

    • 返回 ans

  • 從主方法返回 dp(A 的大小 – 1, B 的大小 – 1, C 的大小 – 1, 0)

示例

讓我們看看下面的實現,以便更好地理解:

即時演示

class Solution:
   def solve(self, s):
      A, rest = s.split("+")
      B, C = rest.split("=")
      def dp(i, j, k, carry):
         if i <= -1 and j <= -1 and k <= -1:
            return 0 if carry == 0 else 1
         last1 = int(A[i]) if i >= 0 else 0
         last2 = int(B[j]) if j >= 0 else 0
         last3 = int(C[k]) if k >= 0 else 0
         prefix1 = int(A[: i + 1]) if i >= 0 else 0
         prefix2 = int(B[: j + 1]) if j >= 0 else 0
         prefix3 = int(C[: k + 1]) if k >= 0 else 0
         if i <= -1 and j <= -1:
            rhs = prefix3 - carry
            if rhs <= 0:
               return abs(rhs)
            if i == -1 or j == -1:
               return len(str(rhs))
            else:
               assert False
         if k <= -1:
            return len(str(prefix1 + prefix2 + carry))
         ans = float("inf")
         carry2, lhs = divmod(carry + last1 + last2, 10)
         if lhs == last3:
            ans = dp(i - 1, j - 1, k - 1, carry2)
         req = last3 - carry - last2
         extra_zeros = max(0, -1 - i)
         carry2 = 1 if req < 0 else 0
         ans = min(ans, 1 + extra_zeros + dp(max(-1, i), j - 1, k - 1, carry2))
         req = last3 - carry - last1
         extra_zeros = max(0, -1 - j)
         carry2 = 1 if req < 0 else 0
         ans = min(ans, 1 + extra_zeros + dp(i - 1, max(-1, j), k - 1, carry2))
         carry2, lhs = divmod(last1 + last2 + carry, 10)
         ans = min(ans, 1 + dp(i - 1, j - 1, k, carry2))
         return ans
      return dp(len(A) - 1, len(B) - 1, len(C) - 1, 0)

ob = Solution()
print (ob.solve('2+6=7'))

輸入

'2+6=7'

輸出

2

更新於: 2020-12-23

152 次瀏覽

開啟你的 職業生涯

透過完成課程獲得認證

立即開始
廣告