#include <bits/stdc++.h>
#include "crocodile.h"
using namespace std;
template<typename T1, typename T2>istream& operator>>(istream& in, pair<T1, T2>& a) { return in >> a.first >> a.second; }
template<typename T1, typename T2>ostream& operator<<(ostream& out, pair<T1, T2> a) { return out << a.first << " " << a.second;}
template<typename T> void print(T t) { cout << t <<' '; }
template<typename T, typename... Args> void print(T t, Args... args) { print(t);print(args...); }
string operator*=(string& s, int cnt) { string t = s;for (size_t i = 1; i < cnt; i++)s += t;return s; }
string operator*(string s, int cnt) { return s *= cnt; }
#define int long long
#define pb push_back
#define ppb pop_back
#define pf push_front
#define ppf pop_front
#define all(x) (x).begin(),(x).end()
#define allr(x) (x).rbegin(),(x).rend()
#define uniq(v) (v).erase(unique(all(v)),(v).end())
#define len(x) (int)((x).size())
#define elif else if
#define add insert
#define append push_back
#define pop pop_back
#define str string
#define in :
#define fr first
#define sc second
#define pii pair<int,int>
#define vi vector<int>
#define vii vector<pii>
#define mi map<int,int>
#define mii map<pii,int>
#define rep(i,a,b) for(int i=a;i<b;i++)
#define rrep(i,a,b) for(int i=a;i>b;i--)
#define el '\n'
#define printl(arg) cout << arg << endl
// #define print(arg) cout << arg
#define inputa(arg) for (auto& e : arg) cin >> e
#define printa(arg) for (auto& e : arg) print(e);
#define printr(arg) { printl(arg);return; }
#define printd(arg) printf("%0.15lf\n", arg)
const int mod=1e9+7;
const int INF=1e18;
const int MAX_N=2e5+2;
int n,m,k,x,y,z,t,q,counter;
// vector<vector<vector<int>>> dp(101 , vector<vector<int>>(101, vector<int>(101)));
// vector<vector<int>>dp(5001, vi(5001)),ndp(5001,vi(5001));
int travel_plan(int n,int m,int r[][2],int l[],int k,int p[]){
vector<vector<vector<int>>> a(n);
rep(i,0,m){
int x=r[i][0], y=r[i][1],z=l[i];
a[x].append({y,z});
a[y].append({x,z});
}
vi v(n,0);
vector<vector<int>> dis(n,vector<int>(2,INF));
set<pii> dq;
rep(i,0,k){
int t=p[i];
dis[t]={0,0};
dq.add({0,t});
}
while(len(dq)){
pii tt=*dq.begin();
int par=tt.sc;
if (v[par]) continue;
v[par]=1;
for(auto& i in a[par]){
int d=i[1],child=i[0];
if(d+dis[par][1]<dis[child][1]){
if(d+dis[par][1]<=dis[child][0]){
dis[child]={d+dis[par][1],dis[child][0]};
}else{
dis[child]={dis[child][0],d+dis[par][1]};
}
dq.add({dis[child][1],child});
}
}
}
return dis[0][1];
}
Compilation message
crocodile.cpp: In function 'std::string operator*=(std::string&, int)':
crocodile.cpp:11:75: warning: comparison of integer expressions of different signedness: 'size_t' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
11 | string operator*=(string& s, int cnt) { string t = s;for (size_t i = 1; i < cnt; i++)s += t;return s; }
| ~~^~~~~
/usr/bin/ld: /tmp/cch8eJdi.o: in function `main':
grader.cpp:(.text.startup+0x36): undefined reference to `travel_plan(int, int, int (*) [2], int*, int, int*)'
collect2: error: ld returned 1 exit status