牛客25461 双喷泉浇花难题 如何用算法快速找到最优解

39 阅读2分钟

截图未命名.jpg

问题理解

我们需要为两个喷泉确定最小的工作半径,使得:

  1. 花坛中所有花都被浇灌到
  2. 喷泉工作半径的平方和(r1² + r2²)最小

关键思路

  1. ‌距离计算‌:对每朵花计算到两个喷泉的欧几里得距离平方
  2. ‌排序处理‌:按到第一个喷泉的距离排序
  3. ‌分割点搜索‌:从后往前寻找最优分割点,使得:
    • 前i朵花由第一个喷泉覆盖
    • 剩余花朵由第二个喷泉覆盖
  4. ‌特殊情况处理‌:考虑全部由一个喷泉覆盖的情况

完整代码

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

struct Flower {
    long long d1, d2; // 到两个喷泉的距离平方
};

int main() {
    int n, x1, y1, x2, y2;
    cin >> n >> x1 >> y1 >> x2 >> y2;
    
    vector<Flower> flowers(n);
    for (int i = 0; i < n; ++i) {
        int x, y;
        cin >> x >> y;
        // 计算到两个喷泉的距离平方(避免浮点运算)
        long long dx1 = x - x1, dy1 = y - y1;
        long long dx2 = x - x2, dy2 = y - y2;
        flowers[i].d1 = dx1*dx1 + dy1*dy1;
        flowers[i].d2 = dx2*dx2 + dy2*dy2;
    }
    
    // 预处理:按d1升序排序
    sort(flowers.begin(), flowers.end(), 
        [](const Flower& a, const Flower& b) {
            return a.d1 < b.d1;
        });
    
    // 预处理后缀最大值数组
    vector<long long> suffix_max(n+1, 0);
    for (int i = n-1; i >= 0; --i) {
        suffix_max[i] = max(flowers[i].d2, suffix_max[i+1]);
    }
    
    long long min_sum = suffix_max[0]; // 初始化为全部由喷泉2覆盖的情况
    
    // 遍历所有可能的分割点
    for (int i = 0; i < n; ++i) {
        long long current_sum = flowers[i].d1 + suffix_max[i+1];
        min_sum = min(min_sum, current_sum);
    }
    
    cout << min_sum << endl;
    return 0;
}

原文: 牛客25461 双喷泉浇花难题 如何用算法找到最优解 从几何到优化的完美解法