結果

問題 No.2179 Planet Traveler
コンテスト
ユーザー HIcoder
提出日時 2023-07-12 09:11:11
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,819 bytes
コンパイル時間 887 ms
コンパイル使用メモリ 109,200 KB
最終ジャッジ日時 2025-02-15 10:00:17
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 23 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<set>
#include<algorithm>
#include<vector>
#include<string>
#include<set>
#include<map>
#include<numeric>
#include<queue>
#include<cmath>
using namespace std;
typedef long long ll;
const ll INF=1LL<<40;
typedef pair<int,int> P;
typedef pair<int,P> PP;
const ll MOD=998244353;
const double PI=acos(-1);

/*
// abs(√a - √b) <= √cかどうかの判定
bool f(ll a,ll b,ll c){


    //ll t=(a*a + b*b + c*c)-(2*a*b + 2*b*c + 2*c*a);
    //t<=0

    ll t=(a*a + b*b + c*c)/(2*a*b +2*b*c + 2*c*a);

    return t<=1;
}
*/

/*
bool f(ll a,ll b,ll c){   
    double t=abs(sqrt(a) - sqrt(b));

    return t<=sqrt(c);
}
*/
/*
bool f(ll a,ll b,ll c){   
    //ll t=(a+b-c)/(4*a*b)*(a+b-c);

     ll t=(a+b-c)*(a+b-c)/(4*a*b);


    ll u=abs(a+b-c);

    if(u>(INF/u)) return false;

    return t<=1;
}
*/

int main(){

    int N;
    cin>>N;
    vector<ll> X(N),Y(N),T(N);
    for(int i=0;i<N;i++){
        cin>>X[i]>>Y[i]>>T[i];
    }


    auto f=[&](int i,int j){
        if(T[i]==T[j]){ 
            return (X[i]-X[j])*(X[i]-X[j])+(Y[i]-Y[j])*(Y[i]-Y[j]);
        }else{
            //T[i]!=T[j]

            ll Ri=X[i]*X[i]+Y[i]*Y[i];
            ll Rj=X[j]*X[j]+Y[j]*Y[j];

            ll ub=1LL<<30;
            ll lb=-1;

            while(ub-lb>1){
                ll mid=(ub+lb)/2;
                ll c=mid*mid;

                if(4*Ri*Rj<c){
                    ub=mid;
                }else{
                    lb=mid;
                }
            }
            ll r=lb;

            //return R1s+R2s - (ll)sqrt(4*R1s*R2s);
            return Ri+Rj - r;
        }

    };

    ll ub=INF,lb=-1;
    //ubのときは必ずたどり着ける
    while(ub>lb+1){

        ll mid=(ub+lb)/2;

        vector<vector<int>> G(N);

        for(int i=0;i<N;i++){
            for(int j=i+1;j<N;j++){
                
                if(f(i,j)<=mid){
                    //iとjに辺が張れる
                    G[i].push_back(j);
                    G[j].push_back(i);
                    
                }

            }
        }
        /*
        for(int i=0;i<N;i++){
            cout<<"i="<<i<<endl;
            for(int v:G[i]){
                cout<<"v="<<v<<endl;
            }
        }
        */


        //bfs
        vector<ll> dp(N,INF);
        dp[0]=0;//惑星0がスタート
        queue<int> que;
        que.push(0);

        while(!que.empty()){

            int now=que.front();
            que.pop();
            for(int to:G[now]){
                if(dp[to]==INF){
                    dp[to]=dp[now]+1;
                    que.push(to);
                }
            }
        }

        if(dp[N-1]!=INF){
            //midで到達できる
            ub=mid;
            
        }else{
            lb=mid;
            
        }

    }

    cout<<ub<<endl;

}
0