算法：购房之旅

题目

小强有n个朋友，每个朋友有一定数量的金币，现在他们要购买房子，一共有m个房子，每个房子有两个参数：舒适度和价格，当一个人的金币大于等于一个房子的价格时，才可以购买房子，且要满足以下条件：1.一个人至多购买一个房子。2.一个房子至多被一个人购买。现在小强想知道n个朋友购买的房子的舒适度之和最大可能是多少？

输入描述

第一行两个整数n,m 接下来一行n个数，第i个整数x表示第i个人的金币x, 1<=x<=10^9 接下来m行每行两个整数表示每个房子的舒适度a和价格b，1<=a,b<=10^9,1<=n,m<=2*10^5

输出描述

输出一个数表示最大可能的舒适度之和

示例 1

输入

2 2
2 2
2 2
2 2

输出

4

说明

每个朋友都会买一个舒适度为2的房子

思路与代码

这个问题可以抽象为一个贪心算法问题。我们可以按照以下步骤来解决：

1. 排序：

   • 首先，将朋友们拥有的金币数量从小到大排序。
   • 然后，将房子按价格从低到高排序。如果价格相同，则按舒适度从高到低排序。

2. 分配房子：

   • 用一个最大堆（优先队列）来记录当前能够购买的房子，并始终选择舒适度最高的房子进行分配。
   • 遍历每一个朋友，对于每个朋友，根据其金币数量，找到所有能购买的房子（即价格小于等于朋友金币的房子），然后将这些房子的舒适度加入最大堆。
   • 每次从最大堆中取出舒适度最高的房子，累加到总舒适度，并标记该房子为已购买。

3. 结果输出：

   • 最后输出累加的最大舒适度和。

C++实现

#include <iostream>
#include <set>
#include <vector>
#include <algorithm>
#include <queue>

using namespace std;

// 定义房子结构
struct House {
long long comfort;
long long price;
};

// 比较函数，用于将房子按价格从高到低排序
bool compareHouse(const House &a, const House &b) {
    return a.comfort > b.comfort;
}

int main() {
     int n = 2, m = 2;

    vector<int> friends(n);
    vector<House> houses(m);

    // 输入朋友的金币数
    for (int i = 0; i < n; ++i) {
        friends[i] = 2;
    }

    // 输入房子的舒适度和价格
    for (int i = 0; i < m; ++i) {
        houses[i].comfort = 2;
        houses[i].price = 2;
    }

    // 最大堆，用于选择舒适度最高的房子
    priority_queue<int> maxComfortHeap;
    int houseIndex = 0;
    long long maxComfortSum = 0;

    // 遍历每个朋友，选择可以购买的房子
    for (int i = 0; i < n; ++i) {
        // 朋友i金币可以购买的所有房子都加入堆中
        while (houseIndex < m && houses[houseIndex].price <= friends[i]) {
            maxComfortHeap.push(houses[houseIndex].comfort);
            ++houseIndex;
        }
        // 从堆中选择舒适度最高的房子
        if (!maxComfortHeap.empty()) {
            maxComfortSum += maxComfortHeap.top();
            maxComfortHeap.pop();
        }
    }

    // 输出结果
    cout << maxComfortSum << endl;

    return 0;
}

代码讲解

• 排序部分：朋友和房子分别根据金币和价格进行排序，这样可以确保我们按顺序考虑每个朋友能买到的最便宜的房子。
• 最大堆：使用优先队列维护当前能购买的房子的最大舒适度，每次从中取出舒适度最高的房子分配给当前朋友。
• 时间复杂度：排序的时间复杂度是O(n log n + m log m)，堆操作的时间复杂度是O(n log m)，总的时间复杂度为O(n log n + m log m)，可以处理题目给出的数据范围。