#include <bits/stdc++.h>
using namespace std;
#define bit(n,i) ((n>>i)&1)
#define all(a) (a).begin(),(a).end()
#define pb push_back
#define ep emplace_back
#define pii pair<int,int>
#define piii pair<int,pii>
#define fi first
#define se second
#define ll long long
#define debug(x) cerr << #x << ' ' << x << '\n'
#define dbg(x) cerr<<#x<<' '<<x<<' '
const int base=31;
const int MOD=1e9+7;
const int N=1e5+5;
void setIO(const string &NAME)
{
if (NAME.size())
{
freopen((NAME+".inp").c_str(),"r",stdin);
freopen((NAME+".out").c_str(),"w",stdout);
}
}
pair<ll,ll>d[N];
bool vis[N];
vector<pii>adj[N];
int travel_plan(int n,int m,int r[][2],int l[],int k,int p[])
{
for (int i=0;i<m;i++)
{
int u=r[i][0],v=r[i][1];
adj[u].pb({v,l[i]});
adj[v].pb({u,l[i]});
}
for (int i=0;i<n;i++) d[i]={1e18,1e18};
priority_queue<pair<ll,int>>pq;
for (int i=0;i<k;i++)
{
pq.push({0,p[i]});
d[p[i]]={0,0};
}
while (pq.size())
{
auto [cost,u]=pq.top();
cost=-cost;
pq.pop();
if (vis[u]) continue;
vis[u]=1;
for (int i=0;i<adj[u].size();i++)
{
int v=adj[u][i].fi,w=adj[u][i].se;
if (cost+w<d[v].se)
{
d[v].se=cost+w;
if (d[v].se<d[v].fi) swap(d[v].fi,d[v].se);
if (d[v].se!=1e18) pq.push({-d[v].se,v});
}
}
}
return d[0].se;
}
int main()
{
ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#ifdef Amon
setIO("Amon");
#endif
/*int n=5,m=4,k=3;
int r[][2]={{0,1},{0,2},{3,2},{2,4}};
int l[]={2,3,1,4};
int p[]={1,3,4};
cout<<travel_plan(n,m,r,l,k,p);*/
return 0;
}
Compilation message
crocodile.cpp: In function 'int travel_plan(int, int, int (*)[2], int*, int, int*)':
crocodile.cpp:50:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
50 | for (int i=0;i<adj[u].size();i++)
| ~^~~~~~~~~~~~~~
crocodile.cpp: In function 'void setIO(const string&)':
crocodile.cpp:21:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
21 | freopen((NAME+".inp").c_str(),"r",stdin);
| ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
crocodile.cpp:22:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
22 | freopen((NAME+".out").c_str(),"w",stdout);
| ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
/usr/bin/ld: /tmp/ccFWyRhV.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccgQbIFZ.o:crocodile.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status
