查看: 21|回复: 0

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

[复制链接]
  • TA的每日心情
    无聊
    前天 10:16
  • 签到天数: 9 天

    [LV.3]偶尔看看II

    8

    主题

    0

    回帖

    18

    积分

    新手上路

    Rank: 1

    积分
    18
    发表于 3 天前 | 显示全部楼层 |阅读模式

    问题理解

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

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

    算法步骤详解
    • ‌输入处理‌:读取花朵和喷泉坐标
    • ‌距离计算‌:预处理每朵花到两个喷泉的距离平方
    • ‌排序优化‌:按到第一个喷泉的距离排序
    • ‌逆向扫描‌:维护第二个喷泉需要的最大半径
    • ‌结果计算‌:比较所有可能分割方案

    完整代码
    1. #include <iostream>
    2. #include <vector>
    3. #include <algorithm>
    4. using namespace std;

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

    8. int main() {
    9.     int n, x1, y1, x2, y2;
    10.     cin >> n >> x1 >> y1 >> x2 >> y2;
    11.    
    12.     vector<Flower> flowers(n);
    13.     for (int i = 0; i < n; ++i) {
    14.         int x, y;
    15.         cin >> x >> y;
    16.         // 计算到两个喷泉的距离平方(避免浮点运算)
    17.         long long dx1 = x - x1, dy1 = y - y1;
    18.         long long dx2 = x - x2, dy2 = y - y2;
    19.         flowers[i].d1 = dx1*dx1 + dy1*dy1;
    20.         flowers[i].d2 = dx2*dx2 + dy2*dy2;
    21.     }
    22.    
    23.     // 预处理:按d1升序排序
    24.     sort(flowers.begin(), flowers.end(),
    25.         [](const Flower& a, const Flower& b) {
    26.             return a.d1 < b.d1;
    27.         });
    28.    
    29.     // 预处理后缀最大值数组
    30.     vector<long long> suffix_max(n+1, 0);
    31.     for (int i = n-1; i >= 0; --i) {
    32.         suffix_max[i] = max(flowers[i].d2, suffix_max[i+1]);
    33.     }
    34.    
    35.     long long min_sum = suffix_max[0]; // 初始化为全部由喷泉2覆盖的情况
    36.    
    37.     // 遍历所有可能的分割点
    38.     for (int i = 0; i < n; ++i) {
    39.         long long current_sum = flowers[i].d1 + suffix_max[i+1];
    40.         min_sum = min(min_sum, current_sum);
    41.     }
    42.    
    43.     cout << min_sum << endl;
    44.     return 0;
    45. }
    复制代码


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




    回复

    使用道具 举报

    您需要登录后才可以回帖 登录 | 立即注册

    本版积分规则

    快速回复 返回顶部 返回列表