-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathPeak Index in a Mountain Array.py
52 lines (38 loc) · 1.19 KB
/
Peak Index in a Mountain 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
41
42
43
44
45
46
47
48
49
50
51
52
'''
Let's call an array A a mountain if the following properties hold:
A.length >= 3
There exists some 0 < i < A.length - 1 such that A[0] < A[1] < ... A[i-1] < A[i] > A[i+1] > ... > A[A.length - 1]
Given an array that is definitely a mountain, return any i such that A[0] < A[1] < ... A[i-1] < A[i] > A[i+1] > ... > A[A.length - 1].
Example 1:
Input: [0,1,0]
Output: 1
Example 2:
Input: [0,2,1,0]
Output: 1
Note:
3 <= A.length <= 10000
0 <= A[i] <= 10^6
A is a mountain, as defined above.
'''
class Solution(object):
def peakIndexInMountainArray(self, A):
"""
:type A: List[int]
:rtype: int
"""
left = [False for i in xrange(len(A))]
for i in xrange(1, len(A)):
if A[i-1] < A[i]:
left[i] = True
else:
break
right = [False for i in xrange(len(A))]
for i in reversed(xrange(0, len(A) - 1)):
if A[i] > A[i+1]:
right[i] = True
else:
break
for i in xrange(1, len(A) - 1):
if left[i] and right[i]:
return i
return -1