94차 3번 문제풀이

Author: hangyeol

live9/test94/문제3/백한결.py

@@ -1,6 +1,6 @@
 def solution(stones, k):
-    start = 0
-    end = len(stones) - 1
+    start = 1
+    end = max(stones)
     max_friends = 0
 
     while start <= end:
@@ -20,15 +20,12 @@ def checkUnderZero(stones, k, friends):
     count = 0
 
     for stone in stones:
-        if stone - friends <= 0:
+        if stone - friends < 0:
             count += 1
         else:
             count = 0
 
         if count >= k:
             return False
 
-    return True
-
-
-print(checkUnderZero([2, 4, 5, 3, 2, 1, 4, 2, 5, 1], 3, 2))
+    return True