forked from adnanaziz/EPIJudge
-
Notifications
You must be signed in to change notification settings - Fork 0
/
alternating_array.py
40 lines (30 loc) · 1.03 KB
/
alternating_array.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
from test_framework.test_failure_exception import TestFailureException
from test_framework.test_utils import enable_timer_hook
def rearrange(A):
# Implement this placeholder.
return
def check_answer(A):
for i in range(len(A)):
if i % 2:
if A[i] < A[i - 1]:
raise TestFailureException('')
if i + 1 < len(A):
if A[i] < A[i + 1]:
raise TestFailureException('')
else:
if i > 0:
if A[i - 1] < A[i]:
raise TestFailureException('')
if i + 1 < len(A):
if A[i + 1] < A[i]:
raise TestFailureException('')
@enable_timer_hook
def rearrange_wrapper(timer, A):
timer.start()
rearrange(A)
timer.stop()
check_answer(A)
from test_framework import test_utils_generic_main, test_utils
if __name__ == '__main__':
test_utils_generic_main.generic_test_main('alternating_array.tsv',
rearrange_wrapper)