每日一练——贪心算法01

Problem：

设有一条边远山区的道路AB，沿着道路AB分布着n所房子，这些房子到A的距离
分别是d1，d2,.....,dn(di<d2<…<dn).为了给所有房子的用户提供移动电话服务，需要在这条道路上设置一些基站.为了保证通信质量，每所房子应该位于距离某个基站的4km范围之内.设计一个算法找到基站的位置，并且使得基站总数达到最少。

思路：

首先在离A距离最近的房子建一个基站，如果离这房子最近的房子距离大于4km，则在该房子这建一个基站，反之则跳过该房子，下一个房子是否离这房子距离大于4km。在我的代码中，首先输入用例，我用了一个结构体House来定义房子，每个结构体的属性有distance这一属性，然后通过sort进行排序，按照距离从小到大排序，用一个for循环，如果当前房子到A的距离大于前一个基站到A的距离加上4km，那么在当前房子处设置一个新的基站，更新上一个基站的位置为当前房子，基站数量加1。循环结束后此时count为最少基站总数。

Sample Input：

请输入房子的数量：5
请输入每个房子到A的距离：3 7 10 12 18

Sample Output：

基站位置分别是：7 12 18
基站总数最少为：3

代码：

#include <iostream> 
#include <vector>   
#include <algorithm> 

using namespace std; 

// 定义房子结构体，包含距离A的距离属性
struct House {
    int distance;
};

// 自定义比较函数，按照距离从小到大排序
bool compare(const House& a, const House& b) {
    return a.distance < b.distance;
}

int main() {
    int n; // 房子的数量
    cout << "请输入房子的数量：";
    cin >> n; // 输入房子的数量
    vector<House> houses(n); // 存储房子信息的元素
    cout << "请输入每个房子到A的距离：";
    for (int i = 0; i < n; i++) {
        cin >> houses[i].distance; // 输入每个房子到A的距离
    }

    sort(houses.begin(), houses.end(), compare); // 对房子按照距离A的距离进行排序

    int count = 0; // 初始化基站数量为0
    int last_station = 0; // 初始化上一个基站的位置为A
    cout << "基站位置分别为：";
    for (int i = 0; i < n; i++) {
        if (houses[i].distance - last_station > 4) { // 如果当前房子到A的距离大于前一个基站到A的距离加上4km，那么在当前房子处设置一个新的基站，更新上一个基站的位置为当前房子，基站数量加1
            count++; // 基站数量加1
            cout << houses[i].distance << " "; // 输出当前位置和下一个基站的位置
            last_station = houses[i].distance; // 更新上一个基站的位置为当前房子
        }    
    }

    cout <<"\n基站总数最少为："<< count<< endl; // 输出基站最少数量

    return 0; 
}