# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
963606 | Amr | 앵무새 (IOI11_parrots) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "crocodile.h"
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define F first
#define S second
#define sz size()
const int N = 3e5+2;
const ll inf = 1e18;
ll vis[N];
pair<ll,ll> d[N];
vector<pair<ll,ll> > v[N];
int travel_plan(int N, int M, int R[][2], int L[], int K, int P[])
{
for(int i = 0; i < M; i++)
{
ll x = R[i][0] , y = R[i][1];
ll z = L[i];
v[x].push_back({y,z});
v[y].push_back({x,z});
}
for(int i = 0; i < N; i++) d[i].F = d[i].S = inf;
multiset<pair<ll,ll> > s;
for(int i = 0; i < K; i++)
{
ll x = P[i];
s.insert({0,x});
//s.insert({0,x});
//vis[x] = 1;
d[x] = {0,0};
}
while(s.sz)
{
ll cur = s.begin()->S;
//cout << cur << " " << s.begin()->F << endl;
ll dis1 = d[cur].F , dis2 = d[cur].S;
s.erase(s.begin());
ll dis = dis2;
//vis[cur]++;
//if(vis[i]==1) continue;
for(int i = 0; i < v[cur].sz; i++)
{
ll newn = v[cur][i].F , w = v[cur][i].S;
//if(vis[newn]==2) continue;
if(w+dis<d[newn].S)
{
if(d[newn].F==inf)
{
d[newn].F=w+dis; continue;
}
s.erase({d[newn].S,newn});
d[newn].S = w+dis;
s.insert({d[newn].S,newn});
if(d[newn].S<d[newn].F) swap(d[newn].F,d[newn].S);
}
}
}
return d[0].S;
}