我在这里编写了这个练习: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");
}
}
请帮助我。提前谢谢你:)
答案 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
位置。