소스 검색

Simplify loop

Taddeus Kroes 6 년 전
부모
커밋
923e261de3
1개의 변경된 파일2개의 추가작업 그리고 4개의 파일을 삭제
  1. 2 4
      2019/01_fuel.py

+ 2 - 4
2019/01_fuel.py

@@ -5,12 +5,10 @@ def fuel(mass):
     return mass // 3 - 2
 
 def fuelrec(mass):
-    f = fuel(mass)
-    total = 0
+    total = f = 0
     while f >= 0:
         total += f
-        mass = f
-        f = fuel(mass)
+        mass = f = fuel(mass)
     return total
 
 masses = [int(line) for line in sys.stdin]