算法访谈街道中央分隔带挑战:Java
我正试图解决interviewstreet的中值挑战。我在这里看到了一个类似的问题:interviewstreet median challenge但我想找出我的方法有什么问题。我使用二进制搜索和排序的数组列表来找出每个点的中值。只有第1、第3和第10次测试通过,其余测试均因回答错误而失败。 问题:http://pastebin.com/1QhbiB2U 代码如下:
/**
* @param args
*/
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
long N = in.nextLong();
List<Long> list = new ArrayList<Long>();
for(int i=0; i<N; i++){
String op = in.next();
long number = in.nextLong();
performOperation(op, number, list);
}
}
private static void performOperation(String op, long number, List<Long> list) {
int index = Collections.binarySearch(list, number);
if(op.equalsIgnoreCase("r")){
if(index < 0){
System.out.println("Wrong!");//Doesn't exist
return;
}else{
list.remove(index);//Remove any one occurence
}
}else{
if(index < 0){
list.add(-index-1, number);//Add in sorted list
}else{
list.add(index, number);//Add where the same number exists, should still be sorted.
}
}
if(list.size() == 0){
System.out.println("Wrong!");
}else if(list.size()%2 == 0){
double median = (list.get(list.size()/2) + list.get(list.size()/2 - 1))/2.0;
if(median == Math.ceil(median))
System.out.println((long)median);
else
System.out.println(median);
}else{
System.out.println(list.get((list.size()-1)/2));
}
}
# 1 楼答案
我认为问题在于附加程序中的双打输出。 我通过问题验证了该程序的输入:
给出:
这种改变适用于上述情况(尽管不是很好):
请注意ArrayList。带索引的add是O(n)