結果

問題 No.1069 電柱 / Pole (Hard)
コンテスト
ユーザー hotman78
提出日時 2020-06-05 03:19:58
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 6,185 bytes
コンパイル時間 27,834 ms
コンパイル使用メモリ 456,004 KB
最終ジャッジ日時 2025-01-10 21:32:54
ジャッジサーバーID
（参考情報） 		judge5 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 49 WA * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("Ofast")
#pragma GCC target("avx2")
#include<bits/stdc++.h>
#include <xmmintrin.h>
#include <immintrin.h>
using namespace::std;
__attribute__((constructor))void init(){cin.tie(0);ios::sync_with_stdio(false);cout<<fixed<<setprecision(15);}
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/tree_policy.hpp>
#include <ext/pb_ds/priority_queue.hpp>
#include<ext/pb_ds/tag_and_trait.hpp>
// #include <boost/multiprecision/cpp_dec_float.hpp>
// #include <boost/multiprecision/cpp_int.hpp>
// namespace mp = boost::multiprecision;
// typedef mp::number<mp::cpp_dec_float<0>> cdouble;
// typedef mp::cpp_int cint;
template<typename T>using pbds=__gnu_pbds::tree<T,__gnu_pbds::null_type,less<T>,__gnu_pbds::rb_tree_tag,__gnu_pbds::tree_order_statistics_node_update>;
template<typename T>using pbds_map=__gnu_pbds::tree<T,T,less<T>,__gnu_pbds::rb_tree_tag,__gnu_pbds::tree_order_statistics_node_update>;
template<typename T,typename E>using hash_map=__gnu_pbds::gp_hash_table<T,E>;
template<typename T>using pqueue =__gnu_pbds::priority_queue<T, greater<T>,__gnu_pbds::rc_binomial_heap_tag>;
typedef long long lint;
#define INF (1LL<<60)
#define IINF (1<<30)
#define LINF (9223372036854775807LL)
#define EPS (1e-10)
#define endl ('\n')
//#define MOD 1000000007LL
#define MOD 998244353LL
typedef vector<lint> vec;
typedef vector<vector<lint>> mat;
typedef vector<vector<vector<lint>>> mat3;
typedef vector<string> svec;
typedef vector<vector<string>> smat;
template<typename T>inline void numout(T t){bool f=0;for(auto i:t){cout<<(f?" ":"")<<i<INF/2?i:"INF";f=1;}cout<<endl;}
template<typename T>inline void numout2(T t){for(auto i:t)numout(i);}
template<typename T>inline void output(T t){bool f=0;for(auto i:t){cout<<(f?" ":"")<<i;f=1;}cout<<endl;}
template<typename T>inline void output2(T t){for(auto i:t)output(i);}
template<typename T>inline void _output(T t){bool f=0;for(lint i=0;i<t.size();i++){cout<<f?"":" "<<t[i];f=1;}cout<<endl;}
template<typename T>inline void _output2(T t){for(lint i=0;i<t.size();i++)output(t[i]);}
#define rep(i,n) for(lint i=0;i<lint(n);++i)
#define repi(i,a,b) for(lint i=lint(a);i<(lint)(b);++i)
#define rrep(i,n) for(lint i=lint(n)-1;i>=0;--i)
#define rrepi(i,a,b) for(lint i=lint(b)-1;i>=lint(a);--i)
#define irep(i) for(lint i=0;;++i)
#define all(n) begin(n),end(n)
#define dist(a,b,c,d) sqrt(pow(a-c,2)+pow(b-d,2))
inline lint gcd(lint A,lint B){return B?gcd(B,A%B):A;}
inline lint lcm(lint A,lint B){return A/gcd(A,B)*B;}
// inline cint cgcd(cint A,cint B){return B?cgcd(B,A%B):A;}
// inline cint clcm(cint A,cint B){return A/cgcd(A,B)*B;}
bool chmin(auto& s,const auto& t){bool res=s>t;s=min(s,t);return res;}
bool chmax(auto& s,const auto& t){bool res=s<t;s=max(s,t);return res;}
const vector<lint> dx={1,0,-1,0,1,1,-1,-1};
const vector<lint> dy={0,1,0,-1,1,-1,1,-1};
#define SUM(v) accumulate(all(v),0LL)
auto call=[](auto f,auto... args){return f(f,args...);};

template<typename T>
struct heap{
    struct node{
        node* ch[2]={0,0};
        int s;
        T val;
        int from,to;
        node(T val,int from,int to):s(1),val(val),from(from),to(to){}
    };
    using np=node*;
    np root=0;
    heap(np t=0):root(t){}
    np meld(np a,np b){
        if(!b)return a?new node(*a):0;
        if(!a)return b?new node(*b):0;
        a=new node(*a);b=new node(*b);
        if(a->val>b->val)swap(a,b);
        a->ch[1]=meld(a->ch[1],b);
        if(!a->ch[0]||a->ch[0]->s<a->ch[1]->s)swap(a->ch[0],a->ch[1]);
        a->s=(a->ch[1]?a->ch[1]->s:0)+1;
        return a;
    }
    heap meld(heap b){
        return heap(meld(root,b.root));
    }
    heap insert(T x,int from,int to){
        return heap(meld(root,new node(x,from,to)));
    }
    heap pop(){
        return heap(meld(root->ch[0],root->ch[1]));
    }
    T top(){
        return root?root->val:T(-1);
    }
    bool empty(){
        return !root;
    }
};

int main(){
    lint n,m,k;
    cin>>n>>m>>k;
    vector<pair<lint,lint>>v(n);
    lint x,y;
    cin>>x>>y;
    x--;y--;
    mat g(n);
    rep(i,n){
        cin>>v[i].first>>v[i].second;
    }
    rep(i,m){
        lint s,t;
        cin>>s>>t;
        s--;t--;
        g[s].push_back(t);
        g[t].push_back(s);
    }
    vector<double>p(n,INF);
    vector<lint>par(n,-1);
    vector<set<lint>> ch(n);
    priority_queue<pair<double,lint>,vector<pair<double,lint>>,greater<pair<double,lint>>>que;
    que.emplace(0,y);
    p[y]=0;
    while(!que.empty()){
        auto [t,n]=que.top();
        que.pop();
        for(auto e:g[n]){
            if(chmin(p[e],t+dist(v[n].first,v[n].second,v[e].first,v[e].second))){
                par[e]=n;
                que.emplace(p[e],e);
            }
        }
    }
    rep(i,n){
        if(par[i]!=-1)ch[par[i]].emplace(i);
    }
    vector<heap<double>>h(n);
    rep(i,n)h[i]=heap<double>();
    queue<lint>qq;
    qq.emplace(y);
    auto d=[&](lint from,lint to){
        return dist(v[from].first,v[from].second,v[to].first,v[to].second)-p[from]+p[to];
    };
    while(!qq.empty()){
        auto n=qq.front();
        qq.pop();
        if(par[n]!=-1)h[n]=h[n].meld(h[par[n]]);
        for(auto e:g[n]){
            if(e!=par[n]&&!ch[n].count(e))h[n]=h[n].insert(d(n,e),n,e);
            if(ch[n].count(e))qq.emplace(e);
        }
    }
    auto comp=[](auto s,auto t){return get<0>(s)>get<0>(t);};
    priority_queue<tuple<double,heap<double>::np,set<lint>>,vector<tuple<double,heap<double>::np,set<lint>>>,decltype(comp)>qqq(comp);
    heap<double>s;
    s=s.insert(p[x],-1,x);
    qqq.emplace(p[x],s.root,set<lint>());
    vector<double>ans;
    while(!qqq.empty()){
        auto [t,now,s]=qqq.top();
        qqq.pop();
        ans.push_back(t);
        if(int(ans.size())==k)break;
        if(now->ch[0]){
            qqq.emplace(t+now->ch[0]->val-now->val,now->ch[0],s);
        }
        if(now->ch[1]){
            qqq.emplace(t+now->ch[1]->val-now->val,now->ch[1],s);
        }
        if(!s.count(now->to)&&h[now->to].root){
            s.insert(now->to);
            qqq.emplace(t+h[now->to].root->val,h[now->to].root,s);
        }
    }
    rep(i,k){
        if(i<(int)ans.size())cout<<ans[i]<<endl;
        else cout<<-1<<endl;
    }
}
0