From 22db43d69c1110a61c9db70e604461d3878082d4 Mon Sep 17 00:00:00 2001 From: hollowcrust <72879387+hollowcrust@users.noreply.github.com> Date: Thu, 12 Oct 2023 01:31:14 +0800 Subject: [PATCH 1/3] Update doctests for power_sum.py --- backtracking/power_sum.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/backtracking/power_sum.py b/backtracking/power_sum.py index fcf1429f8570..155bd114eeae 100644 --- a/backtracking/power_sum.py +++ b/backtracking/power_sum.py @@ -8,7 +8,6 @@ from math import pow - def backtrack( needed_sum: int, power: int, @@ -23,8 +22,8 @@ def backtrack( (0, 3) >>> backtrack(100, 3, 1, 0, 0) (0, 1) - >>> backtrack(800, 2, 1, 0, 0) - (0, 561) + >>> backtrack(500, 2, 1, 0, 0) + (0, 109) >>> backtrack(1000, 10, 1, 0, 0) (0, 0) >>> backtrack(400, 2, 1, 0, 0) @@ -61,8 +60,8 @@ def solve(needed_sum: int, power: int) -> int: 3 >>> solve(100, 3) 1 - >>> solve(800, 2) - 561 + >>> solve(500, 2) + 109 >>> solve(1000, 10) 0 >>> solve(400, 2) From bb9ec4a451dc8670600b463327e65afa182dd644 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 11 Oct 2023 17:40:03 +0000 Subject: [PATCH 2/3] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- backtracking/power_sum.py | 1 + 1 file changed, 1 insertion(+) diff --git a/backtracking/power_sum.py b/backtracking/power_sum.py index 155bd114eeae..8fc3d545ee19 100644 --- a/backtracking/power_sum.py +++ b/backtracking/power_sum.py @@ -8,6 +8,7 @@ from math import pow + def backtrack( needed_sum: int, power: int, From 4e336e6dd1760ed7e97d47314a10c172cc4b12b2 Mon Sep 17 00:00:00 2001 From: hollowcrust <72879387+hollowcrust@users.noreply.github.com> Date: Thu, 12 Oct 2023 01:47:08 +0800 Subject: [PATCH 3/3] Move the import command into backtrack() --- backtracking/power_sum.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backtracking/power_sum.py b/backtracking/power_sum.py index 8fc3d545ee19..23e2b13c6c70 100644 --- a/backtracking/power_sum.py +++ b/backtracking/power_sum.py @@ -6,8 +6,6 @@ The only solution is 2^2+3^2. Constraints: 1<=X<=1000, 2<=N<=10. """ -from math import pow - def backtrack( needed_sum: int, @@ -32,6 +30,8 @@ def backtrack( >>> backtrack(50, 1, 1, 0, 0) (0, 3658) """ + from math import pow + if current_sum == needed_sum: # If the sum of the powers is equal to needed_sum, then we have a solution. solutions_count += 1