소스 검색

Use stdin instead of fileinput

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

+ 2 - 2
2019/01_fuel.py

@@ -1,6 +1,5 @@
 #!/usr/bin/env python3
-import fileinput
-masses = [int(line) for line in fileinput.input()]
+import sys
 
 def fuel(mass):
     return int(mass / 3) - 2
@@ -14,5 +13,6 @@ def fuelrec(mass):
         f = fuel(mass)
     return total
 
+masses = [int(line) for line in sys.stdin]
 print(sum(map(fuel, masses)))
 print(sum(map(fuelrec, masses)))