• + 0 comments

    public static int luckBalance(int k, List> contests) { PriorityQueue que = new PriorityQueue<>(); // min-heap int sum = 0;

    for (int i = 0; i < contests.size(); i++) {
        int l = contests.get(i).get(0);
        int imp = contests.get(i).get(1);
        if (imp == 1) {
            que.add(l);
        } else {
            sum += l; 
        }
    }
    while (que.size() > k) {
        sum -= que.poll(); 
    }
    while (!que.isEmpty()) {
        sum += que.poll();
    }
    return sum;
    
    }