1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
#!/usr/bin/env python3
def target_index(*elements, target):
''' Given an array of integers and a target element, return the list of
indices in the sorted array where the element is same as the given target
element.
>>> target_index(1, 5, 3, 2, 4, 2, target=2)
[1, 2]
>>> target_index(1, 2, 4, 3, 5, target = 6)
[]
>>> target_index(5, 3, 2, 4, 2, 1, target = 4)
[4]
'''
return [i for i, elem in enumerate(sorted(elements)) if elem == target]
if __name__ == "__main__":
import doctest
doctest.testmod()
|
