顺丰科技笔试问题之最多的分值个数问题

顺丰科技笔试题 - 最多的分值个数问题

问题描述

输入n个数，我们按照顺序遍历，可以选择取或者不取，取的话会得到对应的分值，如果分值小于0则退出循环
求：我们遍历这个数组，保证在退出时，能够获得最多的分值个数
示例：

输入：
6
4 -4 1 -3 1 -3
输出:
5
解析：我们选择取第1,3,4,5,6个分值，最多可以取5个

代码实现：

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = Integer.parseInt(in.nextLine());
        String[] split = in.nextLine().split(" ");
        long[] price = new long[n];
        for (int i = 0; i < n; i++) {
            price[i] = Long.parseLong(split[i]);
        }

        //记录累加和
        long preSum = 0;
        //记录结果集
        long count  = 0;
        //定义一个优先队列
        PriorityQueue<Long> queue = new PriorityQueue<>(); //从小到大-5,2,4

        //因为最后一个无论取完是否为负值，都可以取到，故我们只遍历【0，n - 2】
        for (int i = 0; i < n - 1; i++) {
            //如果取完当前的分值，仍然大于等于0，则我们选择取！！！
            if (preSum + price[i] >= 0) {
                ++count;
                preSum += price[i];
                queue.offer(price[i]);
            }else {
                //如果小于0的话，则此时我们判断：
                //如果此时栈为空或者当前元素的值小于等于队首元素，我们直接跳过该元素【不取】
                //否则：说明此时当前元素的值大于队首元素，我们选择取当前元素，放弃取队首元素
                if (queue.isEmpty() || price[i] <= queue.peek()) {
                    continue;
                }else {
                    /*if (preSum + price[i] < 0 && queue.peek() < price[i]) {
                        preSum = preSum - queue.poll();
                        --count;
                    }
                    queue.offer(price[i]);
                    preSum += price[i];
                    ++count;*/
                    //pop队首元素，更新preSum的值
                    preSum = preSum - queue.poll() + price[i];
                    queue.offer(price[i]);
                    //count一加一减，保持不变
                }
            }
        }
        //记得最后count + 1(加上最后一个元素)
        System.out.println(++count);
    }

}