結果

問題 No.1069 電柱 / Pole (Hard)
ユーザー HaarHaar
提出日時 2020-06-04 23:00:05
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 6,074 bytes
コンパイル時間 3,277 ms
コンパイル使用メモリ 250,396 KB
実行使用メモリ 8,832 KB
最終ジャッジ日時 2024-11-30 18:21:57
合計ジャッジ時間 6,129 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,820 KB
testcase_03 AC 2 ms
6,816 KB
testcase_04 AC 146 ms
6,820 KB
testcase_05 WA -
testcase_06 AC 9 ms
6,816 KB
testcase_07 AC 10 ms
6,820 KB
testcase_08 WA -
testcase_09 WA -
testcase_10 AC 6 ms
6,816 KB
testcase_11 AC 2 ms
6,816 KB
testcase_12 AC 3 ms
6,816 KB
testcase_13 AC 5 ms
6,820 KB
testcase_14 AC 5 ms
6,820 KB
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 AC 8 ms
6,816 KB
testcase_20 WA -
testcase_21 AC 5 ms
6,816 KB
testcase_22 AC 3 ms
6,820 KB
testcase_23 AC 13 ms
6,816 KB
testcase_24 WA -
testcase_25 AC 9 ms
6,816 KB
testcase_26 AC 8 ms
6,816 KB
testcase_27 AC 7 ms
6,816 KB
testcase_28 AC 8 ms
6,816 KB
testcase_29 AC 3 ms
6,816 KB
testcase_30 AC 2 ms
6,820 KB
testcase_31 AC 2 ms
6,820 KB
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 AC 2 ms
6,816 KB
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
testcase_40 WA -
testcase_41 WA -
testcase_42 WA -
testcase_43 AC 3 ms
6,816 KB
testcase_44 WA -
testcase_45 WA -
testcase_46 WA -
testcase_47 WA -
testcase_48 AC 7 ms
6,816 KB
testcase_49 AC 6 ms
6,820 KB
testcase_50 AC 3 ms
6,816 KB
testcase_51 WA -
testcase_52 AC 3 ms
6,820 KB
testcase_53 AC 7 ms
6,816 KB
testcase_54 WA -
testcase_55 WA -
testcase_56 AC 2 ms
6,820 KB
testcase_57 WA -
testcase_58 AC 2 ms
6,816 KB
testcase_59 WA -
testcase_60 WA -
testcase_61 WA -
testcase_62 AC 2 ms
6,816 KB
testcase_63 AC 2 ms
6,820 KB
testcase_64 WA -
testcase_65 AC 2 ms
6,820 KB
testcase_66 WA -
testcase_67 AC 2 ms
6,820 KB
testcase_68 WA -
testcase_69 AC 6 ms
6,820 KB
testcase_70 AC 5 ms
6,816 KB
testcase_71 AC 7 ms
6,816 KB
testcase_72 AC 4 ms
6,820 KB
testcase_73 AC 5 ms
6,820 KB
testcase_74 AC 6 ms
6,816 KB
testcase_75 AC 6 ms
6,820 KB
testcase_76 WA -
testcase_77 WA -
testcase_78 WA -
testcase_79 AC 6 ms
6,816 KB
testcase_80 WA -
testcase_81 AC 5 ms
6,816 KB
testcase_82 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

#ifdef DEBUG
#include <Mylib/Debug/debug.cpp>
#else
#define dump(...)
#endif

/**
 * @title グラフ用テンプレート
 * @docs graph_template.md
 */
template <typename Cost = int> class Edge{
public:
  int from,to;
  Cost cost;
  Edge() {}
  Edge(int to, Cost cost): to(to), cost(cost){}
  Edge(int from, int to, Cost cost): from(from), to(to), cost(cost){}
};

template <typename T> using Graph = std::vector<std::vector<Edge<T>>>;
template <typename T> using Tree = std::vector<std::vector<Edge<T>>>;

template <typename T, typename C> void add_edge(C &g, int from, int to, T w = 1){
  g[from].emplace_back(from, to, w);
}

template <typename T, typename C> void add_undirected(C &g, int a, int b, T w = 1){
  add_edge<T, C>(g, a, b, w);
  add_edge<T, C>(g, b, a, w);
}




/**
 * @docs input_tuple_vector.md
 */
template <typename T, size_t ... I>
void input_tuple_vector_init(T &val, int N, std::index_sequence<I...>){
  (void)std::initializer_list<int>{
    (void(std::get<I>(val).resize(N)), 0)...
      };
}

template <typename T, size_t ... I>
void input_tuple_vector_helper(T &val, int i, std::index_sequence<I...>){
  (void)std::initializer_list<int>{
    (void(std::cin >> std::get<I>(val)[i]), 0)...
      };
}

template <typename ... Args>
auto input_tuple_vector(int N){
  std::tuple<std::vector<Args>...> ret;

  input_tuple_vector_init(ret, N, std::make_index_sequence<sizeof...(Args)>());
  for(int i = 0; i < N; ++i){
    input_tuple_vector_helper(ret, i, std::make_index_sequence<sizeof...(Args)>());
  }

  return ret;
}


template <typename T>
struct YenAlgorithm{

  Graph<T> g;
  int s, t, K, N;

  std::vector<std::vector<bool>> valid;

  using Path = std::pair<T, std::vector<int>>;

  std::vector<std::optional<Path>> result;
  std::priority_queue<Path, std::vector<Path>, std::greater<Path>> stock;
  
  YenAlgorithm(Graph<T> g, int s, int t, int K): g(g), s(s), t(t), K(K), N(g.size()){run();}

  void run(){
    valid.resize(N);
    for(int i = 0; i < N; ++i){
      valid[i].assign(g[i].size(), true);
    }

    result.resize(K);

    for(int i = 0; i < K; ++i){
      if(i == 0){
        auto res = shortest_path(s);
        if(res){
          stock.push(*res);
        }
      }else{
        std::vector<int> prev_path;

        {
          int cur = s;
          for(auto u : result[i-1].value().second){
            prev_path.push_back(cur);
            cur = g[cur][u].to;
          }
          prev_path.push_back(t);
          dump(prev_path);
        }

        std::vector<bool> check(i, true);

        for(int k = 0; k < (int)prev_path.size() - 1; ++k){
          const int u = prev_path[k];
          
          for(int j = 0; j < i; ++j){
            if(check[j]){
              valid[prev_path[k]][(*result[j]).second[k]] = false;
            }
          }

          dump(check, valid[prev_path[k]]);

          auto res = shortest_path(u);
          dump(res);
          if(res){

            auto [c, p] = *res;

            std::vector<int> temp;
            for(int j = 0; j < k; ++j){
              int v = (*result[i-1]).second[j];

              c += g[prev_path[j]][v].cost;
              temp.push_back(v);
            }

            temp.insert(temp.end(), p.begin(), p.end());
            //std::swap(temp, p);
            
            stock.emplace(c, temp);
          }
          

          for(int j = 0; j < i; ++j){
            if(check[j]){
              valid[prev_path[k]][(*result[j]).second[k]] = true;
            }
          }

          for(int j = 0; j < i; ++j){
            if(check[j]){
              if(prev_path[k+1] != g[prev_path[k]][(*result[j]).second[k]].to){
                check[j] = false;
              }
            }
          }
        }
      }

      std::cerr << "\n";

      if(stock.empty()) break;

      result[i] = stock.top(); stock.pop();
    }
  }


  std::optional<std::pair<T, std::vector<int>>> shortest_path(int from){
    std::vector<bool> visited(N, false);
    std::vector<std::optional<T>> dist(N);
    std::vector<std::pair<int, int>> restore(N);

    std::priority_queue<
      std::pair<T, int>,
      std::vector<std::pair<T, int>>,
      std::greater<std::pair<T, int>>> pq;

    dist[from] = 0;
    pq.emplace(0, from);

    while(not pq.empty()){
      auto [d, i] = pq.top(); pq.pop();

      if(visited[i]) continue;
      visited[i] = true;

      for(int k = 0; k < (int)g[i].size(); ++k){
        if(not valid[i][k]) continue;
        auto &e = g[i][k];

        if(not dist[e.to]){
          dist[e.to] = d + e.cost;
          pq.emplace(*dist[e.to], e.to);
          restore[e.to] = std::make_pair(i, k);
        }else{
          if(*dist[e.to] > d + e.cost){
            dist[e.to] = d + e.cost;
            if(not visited[e.to]) pq.emplace(*dist[e.to], e.to);
            restore[e.to] = std::make_pair(i, k);
          }
        }
      }
    }
    
    if(dist[t]){
      std::vector<int> p;

      int cur = t;
      while(cur != from){
        auto [i, j] = restore[cur];
        p.push_back(j);
        cur = i;
      }
      
      std::reverse(p.begin(), p.end());

      return {std::make_pair(*dist[t], p)};
    }else{
      return {}; 
    }
  }
};




int main(){
  int N, M, K; std::cin >> N >> M >> K;
  int X, Y; std::cin >> X >> Y;
  --X, --Y;

  auto [p, q] = input_tuple_vector<double, double>(N);

  Graph<double> g(N);
  for(int i = 0; i < M; ++i){
    int P, Q; std::cin >> P >> Q;
    --P, --Q;

    double dx = p[P] - p[Q];
    double dy = q[P] - q[Q];

    double L = std::sqrt(dx * dx + dy * dy);

    add_undirected(g, P, Q, L);
  }


  auto res = YenAlgorithm(g, X, Y, K).result;

  /*
  std::cerr << "\n";
  
  for(auto a : res){
    std::vector<int> p;
    int cur = X;
    for(auto u : a->second){
      p.push_back(cur);
      cur = g[cur][u].to;
    }
    p.push_back(Y);
    dump(p);
  }

  dump(res);*/

  for(auto x : res){
    if(not x){
      std::cout << -1 << "\n";
    }else{
      std::cout << std::fixed << std::setprecision(12) << x->first << "\n";
    }
  }

  return 0;
}
0