Skip to content

Commit 4e023e4

Browse files
authored
Update arranging-coins.py
1 parent 3697b09 commit 4e023e4

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

Python/arranging-coins.py

+8-6
Original file line numberDiff line numberDiff line change
@@ -21,12 +21,14 @@ def arrangeCoins(self, n):
2121
:type n: int
2222
:rtype: int
2323
"""
24+
def check(mid, n):
25+
return mid*(mid+1) <= 2*n
26+
2427
left, right = 1, n
2528
while left <= right:
26-
mid = left + (right - left) / 2
27-
if 2 * n < mid * (mid+1):
28-
right = mid - 1
29+
mid = left + (right-left)//2
30+
if not check(mid, n):
31+
right = mid-1
2932
else:
30-
left = mid + 1
31-
return left - 1
32-
33+
left = mid+1
34+
return right

0 commit comments

Comments
 (0)