結果

問題 No.1545 [Cherry 2nd Tune N] Anthem
ユーザー midri1784midri1784
提出日時 2021-06-11 22:19:46
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 207 ms / 3,000 ms
コード長 3,397 bytes
コンパイル時間 2,408 ms
コンパイル使用メモリ 220,196 KB
実行使用メモリ 34,224 KB
最終ジャッジ日時 2024-05-08 18:12:21
合計ジャッジ時間 12,990 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 67 ms
16,024 KB
testcase_05 AC 189 ms
28,036 KB
testcase_06 AC 65 ms
14,452 KB
testcase_07 AC 78 ms
16,896 KB
testcase_08 AC 136 ms
16,732 KB
testcase_09 AC 66 ms
10,752 KB
testcase_10 AC 73 ms
14,208 KB
testcase_11 AC 96 ms
17,956 KB
testcase_12 AC 176 ms
25,932 KB
testcase_13 AC 53 ms
11,540 KB
testcase_14 AC 64 ms
15,704 KB
testcase_15 AC 22 ms
6,624 KB
testcase_16 AC 87 ms
20,216 KB
testcase_17 AC 39 ms
7,936 KB
testcase_18 AC 207 ms
23,692 KB
testcase_19 AC 86 ms
15,292 KB
testcase_20 AC 35 ms
7,424 KB
testcase_21 AC 122 ms
25,684 KB
testcase_22 AC 120 ms
22,020 KB
testcase_23 AC 119 ms
15,676 KB
testcase_24 AC 6 ms
5,376 KB
testcase_25 AC 3 ms
5,376 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 5 ms
5,376 KB
testcase_29 AC 6 ms
5,376 KB
testcase_30 AC 4 ms
5,376 KB
testcase_31 AC 4 ms
5,376 KB
testcase_32 AC 3 ms
5,376 KB
testcase_33 AC 4 ms
5,376 KB
testcase_34 AC 4 ms
5,376 KB
testcase_35 AC 8 ms
5,376 KB
testcase_36 AC 7 ms
5,376 KB
testcase_37 AC 2 ms
5,376 KB
testcase_38 AC 4 ms
5,376 KB
testcase_39 AC 2 ms
5,376 KB
testcase_40 AC 3 ms
5,376 KB
testcase_41 AC 2 ms
5,376 KB
testcase_42 AC 3 ms
5,376 KB
testcase_43 AC 5 ms
5,376 KB
testcase_44 AC 4 ms
5,376 KB
testcase_45 AC 4 ms
5,488 KB
testcase_46 AC 4 ms
5,376 KB
testcase_47 AC 7 ms
5,504 KB
testcase_48 AC 12 ms
8,960 KB
testcase_49 AC 9 ms
6,528 KB
testcase_50 AC 13 ms
8,704 KB
testcase_51 AC 3 ms
5,376 KB
testcase_52 AC 16 ms
8,576 KB
testcase_53 AC 9 ms
6,656 KB
testcase_54 AC 3 ms
5,376 KB
testcase_55 AC 12 ms
7,936 KB
testcase_56 AC 5 ms
5,376 KB
testcase_57 AC 9 ms
6,528 KB
testcase_58 AC 9 ms
6,528 KB
testcase_59 AC 3 ms
5,376 KB
testcase_60 AC 5 ms
5,376 KB
testcase_61 AC 4 ms
5,376 KB
testcase_62 AC 8 ms
5,888 KB
testcase_63 AC 4 ms
5,376 KB
testcase_64 AC 114 ms
24,876 KB
testcase_65 AC 3 ms
5,376 KB
testcase_66 AC 85 ms
34,224 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

typedef long long i64;
typedef unsigned long long ui64;
typedef vector<i64> vi;
typedef vector<vi> vvi;
typedef pair<i64, i64> pi;

#define pb push_back
#define sz(a) i64((a).size())
#define all(c) (c).begin(), (c).end()
#define REP(s, e, i) for(i=(s); i < (e); ++i)
inline void RI(i64 &i) {scanf("%lld", &(i));}
inline void RVI(vi &v) { for(i64 i=0;i<sz(v);++i) { RI(v[i]); } }
inline void RVVI(vvi &vv) { for(i64 i=0;i<sz(vv);++i) { RVI(vv[i]); } }
inline void WI(const i64 &i) {printf("%lld\n", i);}
inline void WVI(const vi &v, char sep=' ') { for(i64 i=0;i<sz(v);++i) { if(i != 0){ printf("%c", sep); } printf("%lld", v[i]);} printf("\n"); }
inline void WS(const string &s) { printf("%s\n", s.c_str()); }
inline void WB(bool b, const string &yes, const string &no) { if(b){ WS(yes);} else { WS(no);} }
inline void YESNO(bool b) { WB(b, "YES", "NO"); }
inline void YesNo(bool b) { WB(b, "Yes", "No"); }
#define BUF_LENGTH 1000000
inline void RS(string &s) {static char buf[BUF_LENGTH]; scanf("%s", buf); s = buf;}
template<typename T> inline bool IN(T &S, const typename T::key_type &key) {
  return S.find(key) != S.end();
}
template<typename T> inline bool ON(const T &b, i64 idx) {
  return ((T(1) << idx) & b) != 0;
}

int main(int argc, char *argv[]) {
  i64 i, j, k;
  i64 N, S, T, K; cin >> N >> S >> T >> K;
  --S; --T; 
  vi X(N); RVI(X);
  vector<vector<pi>> edges(N);
  vector<vector<pi>> r_edges(N);
  i64 M; cin >> M;
  REP(0, M, i) {
    i64 a, b, y; RI(a); RI(b); RI(y); --a; --b; 
    edges[a].pb({b, y});
    r_edges[b].pb({a, y});
  }

  auto update = [](const pi &a, const pi &b) {
    if(b.first == -1) {
      return a;
    }
    if(a.first == -1) {
      return b;
    }
    if(b.first < a.first) {
      return b;
    }
    return a;
  };

  // BFS part
  vector<vector<pi>> TS(K, vector<pi>(N, {-1, -1}));
  TS[0][S] = {X[S], -1};
  REP(0, K-1, k) {
    REP(0, N, i) {
      if(TS[k][i].first >= 0) {
        for(auto &e: edges[i]) {
          i64 n = e.first, w = e.second;
          TS[k+1][n] = update(TS[k+1][n], {TS[k][i].first + w + X[n], i});
        }
      }
    }
  }

  // dijkstra part
  vector<pi> D(N, pi(-1, -1));
  using ppi = pair<pi, i64>;
  priority_queue<ppi, vector<ppi>, greater<ppi>> pq;

  D[T] = {0, -1};
  pq.push({D[T], T});
  while(!pq.empty()) {
    auto c = pq.top(); pq.pop();
    i64 cur = c.second;
    pi d = c.first;
    if(d != D[cur]) {
      continue;
    }
    for(auto &e: r_edges[cur]) {
      i64 n = e.first, w = e.second;
      i64 dn = d.first + X[cur] + w;
      if(D[n].first == -1 || dn < D[n].first) {
        D[n] = {dn, cur};
        pq.push({{dn, cur}, n});
      }
    }
  }

  i64 idx = -1, m = -1;
  REP(0, N, i) {
    if(TS[K-1][i].first != -1 && D[i].first != -1) {
      i64 candidate = TS[K-1][i].first + D[i].first;
      if(m == -1 || candidate < m) {
        m = candidate;
        idx = i;
      }
    }
  }

  if(idx == -1) {
    WS("Impossible");
  }
  else {
    vi ans;
    k = K-1;
    i64 prev = idx;
    while(k >= 0) {
      ans.push_back(prev+1);
      prev = TS[k][prev].second;
      --k;
    }
    reverse(all(ans));
    i64 next = D[idx].second;
    while(next != -1) {
      ans.push_back(next+1);
      next = D[next].second;
    }
    //cerr << idx+1 << endl;
    WS("Possible");
    WI(m);
    WI(sz(ans));
    WVI(ans);
  }

  return 0;
}
0