def test_plus_one_v2(self):
self.assertListEqual(plus_one_v2([0]), [1])
self.assertListEqual(plus_one_v2([9]), [1, 0])
self.assertListEqual(plus_one_v2([1, 0, 9]), [1, 1, 0])
self.assertListEqual(plus_one_v2([9, 9, 8, 0, 0, 9]), [9, 9, 8, 0, 1, 0])
self.assertListEqual(plus_one_v2([9, 9, 9, 9]), [1, 0, 0, 0, 0])
def test_plus_one_v2(self):
self.assertListEqual(plus_one_v2([0]), [1])
self.assertListEqual(plus_one_v2([9]), [1, 0])
self.assertListEqual(plus_one_v2([1, 0, 9]), [1, 1, 0])
self.assertListEqual(plus_one_v2([9, 9, 8, 0, 0, 9]),
[9, 9, 8, 0, 1, 0])
self.assertListEqual(plus_one_v2([9, 9, 9, 9]),
[1, 0, 0, 0, 0])
return digits
def plus_one_v3(num_arr):
for idx, digit in reversed(list(enumerate(num_arr))):
num_arr[idx] = (num_arr[idx] + 1) % 10
if num_arr[idx]:
return num_arr
return [1] + num_arr
from algorithms.arrays import plus_one_v1,plus_one_v2
a=[1,2,3]
b=[9,9,9,9]
print("v1=============")
print(plus_one_v1(a))
print(plus_one_v1(b))
print("v2=============")
print(plus_one_v2(a))
print(plus_one_v2(b))
## 박제준 3/18
## 김 서현 5/24