main.py 517 B

123456789101112131415161718192021222324252627
  1. file = open('input.txt','r')
  2. tmp = file.readlines()
  3. data = []
  4. data.append(0)
  5. for t in tmp:
  6. data.append(int(t.replace('\n','')))
  7. data.sort()
  8. onejolt = 1
  9. threejolt = 1
  10. print(data)
  11. for d in range(1,len(data)-1):
  12. if data[d] - data[d-1] == 1:
  13. onejolt += 1
  14. if data[d] - data[d-1] == 3:
  15. threejolt += 1
  16. print('1 jolt: ' + str(onejolt))
  17. print('3 jolt: ' + str(threejolt))
  18. print(str(data[d-1]) + ' ---('+str(data[d] - data[d-1])+')---> '+str(data[d]))
  19. print(onejolt * threejolt)