Advertisement
Guest User

Untitled

a guest
May 28th, 2015
237
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.02 KB | None | 0 0
  1. def calc(max_shyness, data_list):
  2. if max_shyness == 0:
  3. return 0
  4. else:
  5. total_count, required_count = 0, 0
  6. for i in range(0, max_shyness+1):
  7. if data_list[i] != 0:
  8. if total_count >= i:
  9. total_count += data_list[i]
  10. else:
  11. diff = i - total_count
  12. required_count += diff
  13. total_count = total_count + diff + data_list[i]
  14. return required_count
  15.  
  16. def calc_required_number(data):
  17. if len(data) == 1:
  18. return -1
  19. if len(data) == 2:
  20. max_shyness, data_list = data[0], data[1]
  21. return calc(int(max_shyness), list(map(int, data_list)))
  22.  
  23. try:
  24. with open('a-large-practice.txt') as datasets:
  25. count = 0
  26. for line in datasets:
  27. res = calc_required_number((line.rstrip()).split(' '))
  28. if res != -1:
  29. print("Case #"+str(count)+": " + str(res))
  30. count += 1
  31. except IOError as err:
  32. print('File error: ' + str(err))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement