使用Quicksort算法在未排序的数组中查找中位数

时间:2015-10-26 14:07:46

标签: java algorithm quicksort

我在这里编写了这个练习:https://www.hackerrank.com/challenges/find-median。 我使用的是快速排序算法,但不是用于排序数组。我用它来分割阵列并找到药剂。但我的代码仍然没有完全正常工作,至少在网站上进行样本测试。

7
0 1 2 4 6 5 3

我试图寻找我的问题,但我无法找到它们。 在某些情况下,我的代码返回true结果。 例如:

7
0 6 3 5 2 4 1

这是我的代码:

/**
 * file FindMedian.java
 */
import java.util.Scanner;
class Number{
    int n;

    public int getN() {
        return n;
    }

    public void setN(int n) {
        this.n = n;
    }
}
public class FindMedian {
    public static int partition(int[] a, int low, int high, int n){
        int i=low+1, j= high;
        while (true){
            while(a[i]<a[low]){
                i++;
                if (i==high) break;
            }
            while (a[j]>a[low]){
                j--;
                if (j==low) break;
            }
            if (i>=j)
                break;
            swap(a, i, j);
            i++;
            j--;
        }
        swap(a, low, j);
        return j;
    }
    public static void progress(int[] a, int low, int high,int n, Number i){
        if (low>=high)
            return;
        int j= partition(a, low, high, n);

        if (j==n){
            i.setN(j);
            return;
        }
        else
        if (j>n)
            progress(a, low, j-1, n, i);
        else if (j<n)
            progress(a, j+1, high, n, i);
    }
    public static void swap(int[] a, int i, int j){
        int temp= a[i];
        a[i]= a[j];
        a[j]= temp;
    }

    public static void main(String[] args){
        Scanner sc= new Scanner(System.in);
        int a[]= new int[sc.nextInt()];
        int n= (a.length-1)/2;
        for (int index=0; index< a.length; index++)
            a[index]= sc.nextInt();

        Number i= new Number();
        i.setN(-1);

        progress(a, 0, a.length-1, n, i);

        if (i.getN()!=-1)
            System.out.println(a[i.getN()]);
        else System.out.println("Can't find");
    }
}

请帮助我。提前谢谢你:)

1 个答案:

答案 0 :(得分:2)

我认为你需要获得枢轴的第k个位置并通过执行此“int k = j-low + 1”来交叉检查它是否位于与中位数相同的位置。从阵列中的变量low开始。例如,第二个索引处的值将是数组中第三个最小的元素。

同样对于第二次递归调用,由于我们知道中位数位于枢轴的右侧(位于第k位置),我们期望结果位于右侧的第(nk)位置子阵列

 public static int progress(int[] a, int low, int high,int n){
        if (low==high)
            return a[low];
        //partition array and return index of pivot
        int j= partition(a, low, high, n);

        //find the kth position of the pivot 
        int k=j-low+1;
        //if the kth position of the pivot is the same as the required ith smallest int return pivot.
        if (k==n){
            return a[j];
        }
        else
        if (n<k)
           return progress(a, low, j-1, n, i);
        else if (n>k)
            return progress(a, j+1, high, n-k, i);
    }

另一个错误的是你的主要方法中的这一行:

int n= (a.length-1)/2; 

应该更改为int n=(a.length+1)/2,因为具有奇数个元素的数组的中位数位于(N+1)/2点(N = a.length)。例如,对于具有7个元素,中位数预计在(7+1)/2=4th  位置。