【Lintcode】1662. Median Index

题目地址：

https://www.lintcode.com/problem/median-index/description

给定一个长度$n$的数组$A$，返回其中位数的下标。如果$A$长度是偶数，则返回其靠左的中位数。

可以将$A$的值连同下标做成一个新的class，然后对新的class的数组进行排序，取出中位数的下标即可。代码如下：

import java.util.Arrays;

public class Solution {
    
    class Pair {
        int val, idx;
        
        public Pair(int val, int idx) {
            this.val = val;
            this.idx = idx;
        }
    }
    
    /**
     * @param A: the array a
     * @return: return the index of median
     */
    public int getAns(int[] A) {
        // write your code here
        if (A == null || A.length == 0) {
            return 0;
        }
        
        Pair[] pairs = new Pair[A.length];
        for (int i = 0; i < A.length; i++) {
            pairs[i] = new Pair(A[i], i);
        }
        
        Arrays.sort(pairs, (p1, p2) -> Integer.compare(p1.val, p2.val));
        
        if (A.length % 2 != 0) {
            return pairs[A.length / 2].idx;
        } else {
            return pairs[A.length / 2 - 1].idx;
        }
    }
}

时间复杂度$O(n\log n)$，空间$O(n)$。