結果

問題 No.1069 電柱 / Pole (Hard)
ユーザー どらら
提出日時 2020-05-30 00:22:06
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,643 bytes
コンパイル時間 2,236 ms
コンパイル使用メモリ 192,164 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-11-06 12:17:59
合計ジャッジ時間 4,656 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 33 WA * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define REP(i,a,n) for(int i=(a); i<(int)(n); i++)
#define rep(i,n) REP(i,0,n)
#define FOR(it,c) for(__typeof((c).begin()) it=(c).begin(); it!=(c).end(); ++it)
#define ALLOF(c) (c).begin(), (c).end()
typedef long long ll;
typedef unsigned long long ull;

template <class W>
class kShortestPath {
  struct edge {
    int src, dst;
    W w;
    edge(int src, int dst, W w) : src(src), dst(dst), w(w) {}
  };
  struct state {
    int src, dst;
    int plv;
    W wsum;
    state(int src, int dst, int plv, W wsum)
        : src(src), dst(dst), plv(plv), wsum(wsum) {}
    bool operator<(const state& that) const { return wsum > that.wsum; }
  };

  std::vector<std::vector<edge>> G;

 public:
  kShortestPath(int n) : G(n) {}
  void add_edge(int a, int b, W w) { G[a].emplace_back(edge(a, b, w)); }

  std::vector<double> getkthShortestPath(int s, int t, int k) {
    std::vector<std::vector<state>> dist(G.size());
    std::priority_queue<state> que;
    que.push(state(-1, s, -1, 0));
    while (!que.empty()) {
      state e = que.top();
      que.pop();
      if (dist[e.dst].size() >= k) continue;
      dist[e.dst].emplace_back(e);
      for (const edge& f : G[e.dst]) {
        que.push(state(f.src, f.dst, dist[e.dst].size() - 1, f.w + e.wsum));
      }
    }

    if (k > dist[t].size()) return std::vector<double>();

    vector<double> ret;
    rep(i,dist[t].size()){
      //simple path check
      map<int,int> m;
      int lv = i;
      int now = t;
      m[now] = 1;
      bool flg = true;
      while (true) {
        if(now == s && lv == 0) break;
        int plv = dist[now][lv].plv;
        int prev = dist[now][lv].src;
        
        if(m.count(prev) == 1){
          flg = false;
          break;
        }
        m[prev] = 1;
        lv = plv;
        now = prev;
      }
      if(!flg) break;
      ret.push_back(dist[t][i].wsum);
    }
    return ret;
  }
};

int main(){
  int N, M, K;
  cin >> N >> M >> K;
  int X, Y;
  cin >> X >> Y;
  vector<pair<int,int>> v;
  v.emplace_back(0,0);
  rep(i,N){
    int p, q;
    cin >> p >> q;
    v.emplace_back(p,q);
  }

  kShortestPath<double> ksp(N+1);
  rep(i,M){
    int p, q;
    cin >> p >> q;
    
    double px = v[p].first;
    double py = v[p].second;
    double qx = v[q].first;
    double qy = v[q].second;
    double dist = sqrt((px-qx)*(px-qx) + (py-qy)*(py-qy));

    ksp.add_edge(p, q, dist);
    ksp.add_edge(q, p, dist);
  }

  vector<double> ret = ksp.getkthShortestPath(X, Y, K);
  rep(i,K){
    if(i<ret.size()){
      printf("%.12lf\n", ret[i]);
    }else{
      cout << -1 << endl;
    }
  }
  return 0;
}
0