每日OJ题_牛客_小红的口罩_堆+贪心_C++_Java

目录

牛客_小红的口罩_堆+贪心

题目解析

C++代码

Java代码

---

牛客_小红的口罩_堆+贪心

小红的口罩

描述：
        疫情来了，小红网购了 n个口罩。众所周知，戴口罩是很不舒服的。小红每个口罩戴一天的初始不舒适度为 ai​。
        小红有时候会将口罩重复使用（注：这是非常不卫生的！），每次重复使用时，该口罩的不舒适度会翻倍！小红想知道，自己在不舒适度总和不超过 kkk 的情况下，最多能用现有的口罩度过多少天？

输入描述：

第一行输入两个正整数 n 和 k ，分别代表口罩的总数、以及小红最多能忍受的不舒适度总和。
第二行输入 nn个正整数 ai​ ，用空格隔开。分别代表每个口罩初始的不舒适度。
1≤n≤10^5,1≤ai,k≤10^9

输出描述：

一个整数，代表小红最多能度过的天数。

---

题目解析

        首先很容易想到一个朴素做法，每次 O(n)寻找最小值，然后操作。这样的复杂度是大约 O(n2)的，无法通过。

        然后观察该做法，发现这个 O(n)寻找最小值非常累赘。采取小根堆每次寻找最小值、然后删除最小值、同时加入原最小值翻倍后的值，复杂度均为 O(1)。大概答案最多是 n 天，于是总复杂度 O(n)。

C++代码

#include <functional>
#include <iostream>
#include <queue>
#include <vector>
using namespace std;

int main()
{
    int n = 0, k = 0;
    cin >> n >> k;
    vector<int> arr(n);
    priority_queue<int, vector<int>, greater<int>> heap;
    for(int i = 0; i < n; ++i)
    {
        cin >> arr[i];
        heap.push(arr[i]);
    }
    int sum = 0, res = 0;
    while(heap.size() && sum + heap.top() <= k)
    {
        int x = heap.top();
        heap.pop();
        sum += x;
        heap.push(x * 2);
        ++res;
    }
    cout << res << endl;
    return 0;
}

Java代码

import java.util.*;
public class Main
{
    public static void main(String[] args)
    {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt(), k = in.nextInt();
        PriorityQueue<Integer> heap = new PriorityQueue<>();
        for(int i = 0; i < n; i++)
        {
            int x = in.nextInt();
            heap.add(x);
        }

        int sum = 0, count = 0;
        while(true)
        {
            int t = heap.poll();
            sum += t;
            count++;
            heap.add(t * 2);
            if(sum > k)
            {
                System.out.println(count - 1);
                break;
            }
        }
    }
}