Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def window_result(arr, size):
- num_inc = 0
- count = 0
- for elem in arr:
- if elem == 'i' or elem == 'e':
- count += 1
- else:
- num_inc += (count*(count+1))/2
- count = 0
- num_inc += (count*(count+1))/2
- num_dec = 0
- count = 0
- for elem in arr:
- if elem == 'd' or elem == 'e':
- count += 1
- else:
- num_dec += (count*(count+1))/2
- count = 0
- num_dec += (count*(count+1))/2
- return num_inc-num_dec
- conf = raw_input().strip().split()
- num_days, window_size = int(conf[0]), int(conf[1])
- data = [int(i) for i in raw_input().strip().split()]
- begin = 0
- end = window_size - 1
- while end < num_days:
- diff_array = []
- for i in xrange(begin + 1, end + 1):
- if data[i] > data[i - 1]:
- diff_array.append('i')
- elif data[i] == data[i - 1]:
- diff_array.append('e')
- else:
- diff_array.append('d')
- print window_result(diff_array, window_size-1)
- begin += 1
- end += 1
Add Comment
Please, Sign In to add comment