百度笔试题——浇花

题目简介：

给出两个喷泉的坐标，并给出一堆花的坐标，求解使得所有的花都被任意一个喷泉浇灌到时，喷泉1的半径的平方和喷泉2的半径的平方之和最小。
题目详情见牛客网

思路：

这题其实思路比较简单，首先求解所有花到喷泉1的距离的平方，然后将其排序，排序后依次选择喷泉1的浇花半径的平方和，再遍历所有未被浇灌到的花到喷泉2的距离的平方和的最大值，求解两者之和的最小值即可。时间复杂度是O(N^2),由于n较小，所以是可行的。
不过比较坑的地方在于，这样做可能无法通过全部数据，因为：

1. int类型的数据可能溢出；
2. 考虑到溢出后也只能过百分之60的数据，因为，少考虑了一种情况，即可能所有的花可能都离喷泉2比较近，喷泉1的半径可能为0。

第二个情况我做题时没有想到，看到很多答案里也没有提，如果有只过了百分之60数据的人，可以看下这一点是否考虑到了。

代码

#include <bits/stdc++.h>
using namespace std;
long long int dis(long long int x1,long long int y1,long long int x2,long long int y2){
    return (x2-x1)*(x2-x1)+(y2-y1)*(y2-y1);
}
long long int mmax(long long int a,long long int b){
    return a>b?a:b;
}
long long int mmin(long long int a,long long int b){
    return a<b?a:b;
}
int main(){
    ios::sync_with_stdio(false);
    int n;
    long long int x1,y1,x2,y2;
    cin>>n;
    cin>>x1>>y1>>x2>>y2;
    vector<pair<long long int,long long int>> flowers;
    while(n--){
        pair<long long int,long long int> flower;
        cin>>flower.first>>flower.second;
        flowers.push_back(flower);
    }
    vector<long long int> disTo1;
    unordered_multimap<long long int,int> map1;
    for(int i=0;i<flowers.size();i++){
        long long int d=dis(x1,y1,flowers[i].first,flowers[i].second);
        map1.insert({d,i});
        disTo1.push_back(d);
    }
    sort(disTo1.begin(),disTo1.end());
    long long int res=1000000000000000000;
    set<int> already;
    long long int tmp;
    for(int i=0;i<disTo1.size();i++){
        tmp=disTo1[i];
        if(tmp>res){
            break;
        }
        auto it=map1.equal_range(disTo1[i]);//返回一个迭代器的pair
        if(it.first!=map1.end()){
            for(auto it1=it.first;it1!=it.second;it1++){
                already.insert(it1->second);
            }
        }
        long long int maxIn2=0;
        for(int i=0;i<disTo1.size();i++){
            if(already.count(i)==0){
                maxIn2=mmax(maxIn2,dis(x2,y2,flowers[i].first,flowers[i].second));
            }
        }
        tmp+=maxIn2;
        res=mmin(res,tmp);
    }
    long long int maxIn2=0;
    for(int i=0;i<disTo1.size();i++){//在此处考虑只使用喷泉2的情况
        maxIn2=mmax(maxIn2,dis(x2,y2,flowers[i].first,flowers[i].second));
    }
    res=mmin(maxIn2,res);
    cout<<res;
    return 0;
}