이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define ii pair<int, int>
#define iii pair<int, ii>
#define ff first
#define ss second
const int INF = 1e18;
signed main(){
ios::sync_with_stdio(0);
cin.tie(0);
int v, e; cin >> v >> e;
int mat[v][v];
for(int i = 0; i < v; i ++)
for(int k = 0; k < v; k ++)
mat[i][k] = INF;
for(int i = 0; i < v; i ++)
mat[i][i] = 0;
for(int i = 0; i < e; i ++){
int a, b, w; cin >> a >> b >> w;
mat[a-1][b-1] = min(mat[a-1][b-1], w);
}
vector<ii> adj[v];
for(int i = 0; i < v; i ++)
for(int j = 0; j < v; j ++)
if(mat[i][j] != INF)
adj[i].emplace_back(j, mat[i][j]);
int dist[v][v][v+1];
memset(dist, -1, sizeof dist);
for(int i = 0; i < v; i ++){
priority_queue<iii, vector<iii>, greater<iii>> pq;
pq.push({0, {i, 0}}); dist[i][i][0] = 0;
while(!pq.empty()){
auto tp = pq.top(); pq.pop();
if(dist[i][tp.ss.ff][tp.ss.ss] != tp.ff || tp.ss.ss == v)
continue;
for(auto& [nxt, wt] : adj[tp.ss.ff]){
if(dist[i][nxt][tp.ss.ss+1] == -1 || dist[i][nxt][tp.ss.ss+1] > wt + tp.ff){
dist[i][nxt][tp.ss.ss+1] = wt + tp.ff;
pq.push({dist[i][nxt][tp.ss.ss+1], {nxt, tp.ss.ss+1}});
}
}
}
for(int j = 0; j < v; j ++)
for(int k = 1; k <= v; k ++)
if(dist[i][j][k-1] != -1)
dist[i][j][k] = min(dist[i][j][k-1], dist[i][j][k]);
}
int K, q; cin >> K >> q;
K = min(K, v);
while(q --){
int ci, di; cin >> ci >> di;
cout << dist[ci-1][di-1][K] << '\n';
}
}
/*
4 7
1 2 1
1 4 10
2 3 1
2 4 5
3 2 2
3 4 1
4 3 2
1 3
1 4
4 2
3 3
*/
컴파일 시 표준 에러 (stderr) 메시지
Main.cpp: In function 'int main()':
Main.cpp:39:23: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
39 | for(auto& [nxt, wt] : adj[tp.ss.ff]){
| ^
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |