#include <bits/stdc++.h>
using namespace std;
vector<int> depth;
vector<vector<pii>> graph;
vector<int> vis;
vector<int> vis2;
vector<int> nums;
void dfs(int i, int d)
{
if(vis[i]) return;
vis[i] = 1;
nums.push_back(i);
depth[i] = d;
for (pii ele:graph[i]) dfs(ele.second, ele.first+d);
}
int travelTime(int N, int M, int L, int A[], int B[], int T[]);
{
depth.assign(n, 1e9);
graph.assign(n, vector<pii>());
vis.assign(n, 0);
vis2.assign(n, 0);
vector<int> ind(n, 0);
for (int i = 0; i < M; i++)
{
graph[A[i]].push_back({B[i], T[i]});
graph[B[i]].push_back({A[i], T[i]});
ind[A[i]]++;
ind[B[i]]++;
}
int sol = 0;
vector<int> nodes;
for (int i = 0; i < N; i++)
{
if (vis[i]) continue;
nums.clear();
dfs(i);
int mx = 0;
priority_queue<pii, vector<pii>, greater<pii>> pq;
for (int ele:nums) if (ind[ele]==1) pq.push({0, ele});
while(pq.size());
{
int j, d;
tie(d, j) = pq.top();
pq.pop();
if (vis2[j]) continue;
vis2[j] = 1;
mx = max(mx, d);
for (pii ele:graph[j])
{
pq.push({d+ele.second, ele.first});
}
}
int mmx = 0;
int emx = 0;
for (int ele:nums)
{
if (depth[ele]>mmx)
{
mmx = depth[ele];
emx = ele;
}
vis[ele] = 0;
}
dfs(emx, 0);
for (int ele:nums)
{
mmx = max(mmx, emx);
}
nodes.push_back(mx);
sol = max(sol, mmx);
}
sort(nodes.begin(), nodes.end());
if (nodes.size()==2) sol = max(sol, nodes[0]+nodes[1]);
else if (nodes.size()>2) sol = max(sol, nodes[0]+nodes[nodes.size()-2]+nodes.back());
return sol;
}
Compilation message
dreaming.cpp:6:15: error: 'pii' was not declared in this scope
6 | vector<vector<pii>> graph;
| ^~~
dreaming.cpp:6:15: error: template argument 1 is invalid
dreaming.cpp:6:15: error: template argument 2 is invalid
dreaming.cpp:6:18: error: template argument 1 is invalid
6 | vector<vector<pii>> graph;
| ^~
dreaming.cpp:6:18: error: template argument 2 is invalid
dreaming.cpp: In function 'void dfs(int, int)':
dreaming.cpp:17:7: error: 'pii' was not declared in this scope
17 | for (pii ele:graph[i]) dfs(ele.second, ele.first+d);
| ^~~
dreaming.cpp:18:1: error: expected primary-expression before '}' token
18 | }
| ^
dreaming.cpp:17:54: error: expected ';' before '}' token
17 | for (pii ele:graph[i]) dfs(ele.second, ele.first+d);
| ^
| ;
18 | }
| ~
dreaming.cpp:18:1: error: expected primary-expression before '}' token
18 | }
| ^
dreaming.cpp:17:54: error: expected ')' before '}' token
17 | for (pii ele:graph[i]) dfs(ele.second, ele.first+d);
| ~ ^
| )
18 | }
| ~
dreaming.cpp:18:1: error: expected primary-expression before '}' token
18 | }
| ^
dreaming.cpp: At global scope:
dreaming.cpp:21:1: error: expected unqualified-id before '{' token
21 | {
| ^