京东第二题用优先级队列实现大顶堆,为啥一直过不了。。

我的思路是用 Java PriorityQueue 实现大顶堆。队列里加一个元素,就从对顶进行对比,主要是怕花时间就没写大顶堆的代码。我自己试了很多例子,有错的,也修改了,就是通不过,各位能帮忙看一下吗?

package exam.jingdong180909;
import java.util.Comparator;
import java.util.Iterator;
import java.util.PriorityQueue;
import java.util.Scanner;
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int total = Integer.parseInt(in.nextLine());
        int count = 0;
        PriorityQueue queue = new PriorityQueue(10, new Comparator() { 
            public int compare(Item o1, Item o2) {
                return o2.compareTo(o1);
            }
        });
        for (Item item : queue) {
            System.out.println("item: " + item.a + " " + item.b + " " + item.c);
        }
        for (int i = 0; i < total; i++) {
            String[] strArr = in.nextLine().split(" ");
            Item temp = new Item(Integer.parseInt(strArr[0]), Integer.parseInt(strArr[1]), Integer.parseInt(strArr[2]));
            Iterator itemIterator = queue.iterator();
            boolean flag = false;
            while (itemIterator.hasNext()) {
                Item item = itemIterator.next();
                if (temp.compareTo(item) == 1 || item.compareTo(temp) == 1) {
//                    System.out.println(item.a + " " + item.b + " " + item.c);
                    flag = true;
                    count++;
                    break;
                }
            }
            if (!flag)
                queue.offer(temp);
        }
        System.out.println(count);
    }
}
class Item implements Comparable {
    public int a;
    public int b;
    public int c;
    public Item(int a, int b, int c) {
        this.a = a;
        this.b = b;
        this.c = c;
    }  
    public int compareTo(Item o) {
        if (a > o.a && b > o.b && c > o.c) {
            return 1;
        } else if (a == o.a && b == o.b && c == o.c) {
            return 0;
        }
        return -1;
    }
}
#京东##笔试题目#
全部评论
这么复杂干嘛,。。直接一个set记录下就好了啊,。。。
点赞 回复 分享
发布于 2018-09-09 22:13
暴力解决就完事了
点赞 回复 分享
发布于 2018-09-09 22:58

相关推荐

点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务