查找排序数组中元素的索引

Question

Given a code that sorts an array of integers, how do i search and find the index of each element in the sorted array.

def countSort(arr):
    output = [0 for i in range(256)]

    count = [0 for i in range(256)]

    result = ["" for _ in arr]

    # Store count of each integers
    for i in arr:
        count[ord(i)] += 1

    # Change count[i] so that count[i] now contains actual
    # position of this integers in output array
    for i in range(256):
        count[i] += count[i - 1]

    for i in range(len(arr)):
        output[count[ord(arr[i])] - 1] = arr[i]
        count[ord(arr[i])] -= 1

    for i in range(len(arr)):
        result[i] = output[i]
    return result

driver code

    arr = input("Enter numbers: ")
    result = countSort(arr)
    print("Sorted array is %s" % ("".join(result)))

This is what i tried to do but it has errors

      print(input("Search for Element:"))
      for k in result:
          if k not in result:
             print("Element not found", k)
          else:
             print("Index of Element is ", result.index(k))

Answer 1

Below is a code that let the user insert a number to search.

The code return the index of the number if it can be found in the sorted list.

Note : The code below uses python builtin sort and does not make use of your sorting code.

NUMBERS = [4, 6, 2, 12, 56, 34, 90]

sorted_numbers = sorted(NUMBERS)

number_to_search = input("Search for Element:")

try:
    idx = sorted_numbers.index(number_to_search)
    print('The index of {} is {}.'.format(number_to_search, idx))
except ValueError:
    print('Cant find the number {}.'.format(number_to_search))