이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5+45;
int n,m,k;
vector <pair<int,ll>> g[N];
bool ex[N];
ll dp1[N];
void dfstr(int x,int par){
vector <ll> svi;
for(auto f : g[x]){
int u = f.first;
if(u == par) continue;
ll w = f.second;
dfstr(u,x);
svi.push_back(dp1[u]+w);
}
if(svi.empty()) return;
assert(svi.size() >= 2);
sort(svi.begin(),svi.end());
dp1[x] = svi[1];
}
int zadrvo(){
dfstr(1,0);
return dp1[1];
}
ll d1[N],d2[N];
int travel_plan(int br,int M,int R[][2],int L[],int K,int P[]){
n = br;
m = M;
k = K;
for(int i = 0; i < m; i++){
g[R[i][0]+1].push_back({R[i][1]+1,L[i]});
g[R[i][1]+1].push_back({R[i][0]+1,L[i]});
}
for(int i = 0; i < k; i++){
ex[P[i]+1] = 1;
}
//return zadrvo();
for(int i = 1; i <= n; i++) d1[i] = d2[i] = 1e18;
priority_queue <pair<pair<ll,ll>,int>> pq;
for(int i = 1; i <= n; i++){
if(ex[i]){
pq.push({{0,0},i});
d1[i] = d2[i] = 0;
}
}
while(!pq.empty()){
ll drugi = -pq.top().first.first;
ll prvi = -pq.top().first.second;
int u = pq.top().second;
pq.pop();
for(auto f : g[u]){
int v = f.first,cost = f.second;
if(drugi+cost < d1[v]){
d2[v] = d1[v];
d1[v] = drugi+cost;
if(d2[v] != 1e18) pq.push({{-d2[v],-d1[v]},v});
}
else if(drugi+cost < d2[v]){
d2[v] = drugi+cost;
if(d2[v] != 1e18) pq.push({{-d2[v],-d1[v]},v});
}
}
}
return d2[1];
}
/*int grane[10*N][2],tezine[10*N],izlazi[N];
int main(){
ios_base::sync_with_stdio(false);
cin.tie(0);
int br1,br2,br3;
cin >> br1 >> br2 >> br3;
for(int i = 0; i < br2; i++){
cin >> grane[i][0] >> grane[i][1] >> tezine[i];
}
for(int i = 0; i < br3; i++) cin >> izlazi[i];
cout << travel_plan(br1,br2,grane,tezine,br3,izlazi) << endl;
}*/
컴파일 시 표준 에러 (stderr) 메시지
crocodile.cpp: In function 'int travel_plan(int, int, int (*)[2], int*, int, int*)':
crocodile.cpp:58:12: warning: unused variable 'prvi' [-Wunused-variable]
58 | ll prvi = -pq.top().first.second;
| ^~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |