codeforces374-div2C Journey

Problem - C - Codeforces

Codeforces. Programming competitions and contests, programming community

町$u$から町$v$に行くには$t$かかる.$T$までに最大でいくつの町を回ることが出来るか.状態を($i$番目の町, $j$個既に町を訪れた)としてdijkstra.最短時間を求めているので,既に訪れた町が重複することはない.costが$T$を超える場合は遷移させない.後は経路復元して出力した.

Code

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#include <iostream>
#include <sstream>
#include <vector>
#include <string>
#include <cstring>
#include <algorithm>
#include <queue>
#include <map>
#include <set>

#define REP(i,k,n) for(int i=k;i<n;i++)
#define rep(i,n) for(int i=0;i<n;i++)
#define each(it,v) for(__typeof((v).begin()) it=(v).begin();it!=(v).end();it++)
#define INF 1<<30
#define mp make_pair

using namespace std;
typedef long long ll;
typedef pair<int,int> P;
typedef pair<ll, P> IP;

struct edge {
  int from,to;
  int cost;

  edge(int t,int c) : to(t),cost(c) {}
  edge(int f,int t,int c) : from(f),to(t),cost(c) {}

  bool operator<(const edge &e) const {
      return cost < e.cost;
  }
};

int n, m;
vector<edge> G[5005];
ll t;
int d[5005][5005], prev[5005][5005];

void dijkstra(int s, int n) {
  priority_queue<IP, vector<IP>, greater<IP> > que;

  rep(i, 5005) rep(j, 5005) {
      d[i][j] = INF;
      prev[i][j] = -1;
  }
  
  d[s][1] = 0;

  que.push(mp(0, mp(s, 1)));
  
  while(que.size()) {
      IP p = que.top(); que.pop();

      int cos = p.first;
      int v = p.second.first;
      int cnt = p.second.second;
      if(d[v][cnt] < cos) continue;
  
      rep(i, G[v].size()) {
          edge e = G[v][i];
          ll nc = ll(d[v][cnt]) + ll(e.cost);
          if(nc > t) continue;

          if(d[e.to][cnt+1] > nc) {
              d[e.to][cnt+1] = nc;
              prev[e.to][cnt+1] = v;
              que.push(mp(d[e.to][cnt+1], mp(e.to, cnt+1)));
          }
      }
  }
}

vector<int> get_path(int t, int cnt) {
  vector<int> path;
  for(;t!=-1;t=prev[t][cnt], cnt--) path.push_back(t);
  reverse(path.begin(),path.end());
  return path;
}

int main() {
  cin >> n >> m >> t;

  rep(i, m) {
      int a, b;
      ll c;
      cin >> a >> b >> c;

      a--; b--;
      G[a].push_back(edge(b, c));
      // G[b].push_back(edge(a, c));
  }

  dijkstra(0, n * n);

  for(int i = n; i > 0; i--) {
      if(d[n-1][i] == INF) continue;
      cout << i << endl;

      vector<int> ret = get_path(n-1, i);
      rep(j, ret.size()) {
          if(j) cout << " ";
          cout << ret[j] + 1;
      }
      cout << endl;
      break;
  }

  return 0;
}

最初に$WA$を出したのでlong longに全て直して提出したが$[5000][5000]$なのでメモリが無理.遷移の比較の時にllにして上げれば良いので要らないことに気づく.間違っている理由が全く分からずずっと考えていたが,有向グラフなことを見落としていて,逆の辺を消したら通った(完).