Guest User

Untitled

a guest
Jul 22nd, 2018
82
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.02 KB | None | 0 0
  1. def window_result(arr, size):
  2. num_inc = 0
  3. count = 0
  4. for elem in arr:
  5. if elem == 'i' or elem == 'e':
  6. count += 1
  7. else:
  8. num_inc += (count*(count+1))/2
  9. count = 0
  10. num_inc += (count*(count+1))/2
  11. num_dec = 0
  12. count = 0
  13. for elem in arr:
  14. if elem == 'd' or elem == 'e':
  15. count += 1
  16. else:
  17. num_dec += (count*(count+1))/2
  18. count = 0
  19. num_dec += (count*(count+1))/2
  20.  
  21. return num_inc-num_dec
  22.  
  23. conf = raw_input().strip().split()
  24. num_days, window_size = int(conf[0]), int(conf[1])
  25. data = [int(i) for i in raw_input().strip().split()]
  26.  
  27. begin = 0
  28. end = window_size - 1
  29.  
  30. while end < num_days:
  31. diff_array = []
  32. for i in xrange(begin + 1, end + 1):
  33. if data[i] > data[i - 1]:
  34. diff_array.append('i')
  35. elif data[i] == data[i - 1]:
  36. diff_array.append('e')
  37. else:
  38. diff_array.append('d')
  39. print window_result(diff_array, window_size-1)
  40. begin += 1
  41. end += 1
Add Comment
Please, Sign In to add comment