Skip to content
This repository was archived by the owner on Jul 2, 2024. It is now read-only.

Commit 8172601

Browse files
committed
Update sum of strings function
1 parent c3b2ad3 commit 8172601

File tree

1 file changed

+4
-12
lines changed

1 file changed

+4
-12
lines changed

src/calc/func.py

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -143,14 +143,8 @@ def get_sum_of_strings(number_1: str, number_2: str, /) -> str:
143143
"""
144144

145145
# remove leading zeros
146-
for digit in number_1:
147-
if digit != "0":
148-
break
149-
number_1 = number_1[1::]
150-
for digit in number_2:
151-
if digit != "0":
152-
break
153-
number_2 = number_2[1::]
146+
number_1 = number_1.lstrip("0")
147+
number_2 = number_2.lstrip("0")
154148

155149
if not number_1 or not number_2:
156150
return number_1 or number_2 or "0"
@@ -162,10 +156,8 @@ def get_sum_of_strings(number_1: str, number_2: str, /) -> str:
162156
number_1, number_2 = number_1[::-1], number_2[::-1]
163157

164158
# make strings of the same lengths
165-
while len(number_1) < size:
166-
number_1 += "0"
167-
while len(number_2) < size:
168-
number_2 += "0"
159+
number_1 += "0" * (size - len(number_1))
160+
number_2 += "0" * (size - len(number_2))
169161

170162
carry: int = 0
171163
for digit_1, digit_2 in zip(number_1, number_2):

0 commit comments

Comments
 (0)