lintcode中位数下标

给出一个含有 n 个互不相等整数的无序数组，找到其中中位数的下标。下标从 0 开始。
中位数是指这些数排序后最中间的数。
若 n 为偶数，则中位数是数组排序后的第 n/2 个数。
https://www.lintcode.com/problem/median-index/description

class Solution:
    """
    @param a: the array a
    @return: return the index of median
    """
    def getAns(self, a):
        # write your code here 
        b = sorted(a)
        print(b)
        if len(b) % 2 == 0:
            mid_num = b[len(b) // 2 -1]
        else:
            mid_num = b[len(b) // 2 ]
        print(mid_num)
        mid_index = 0
        for i in a:
            if i == mid_num:
                return mid_index
            mid_index += 1
        return mid_index