#include "dreaming.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;
#define S second
#define F first
#define all(x) (x).begin(),(x).end()
#define sz size()
#define Yes cout << "YES" << endl
#define No cout << "NO" << endl
#define pb(x) push_back(x);
#define endl '\n'
#define fast ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
const int N=3e5+7;
ll INF=INT_MAX,mod=1e9+7;
int TT=1;
ll power(ll x, unsigned int y)
{
ll res = 1;
x = x; // % mod;
if (x == 0) return 0;
while (y > 0)
{
if (y & 1) res = (res*x) ; // % mod;
y = y>>1;
x = (x*x) ; // % mod;
}
return res;
}
ll curans = -1e18;
ll inf = 1e18;
ll vis[N];
pair<ll,ll> dp[N];
ll ans[N];
ll mn = 1e18;
vector<pair<ll,ll> > v[N];
void dfs(ll x, ll par)
{
if(vis[x]) return;
vis[x] = 1;
dp[x] = {0,-inf};
vector<ll> vv;
for(int i = 0; i < v[x].sz; i++)
{
ll newn = v[x][i].F;
if(newn==par) continue;
dfs(newn,x);
vv.push_back(dp[newn].F+v[x][i].S);
}
sort(all(vv)); reverse(all(vv));
if(vv.sz) dp[x].F = vv[0];
if(vv.sz>1)dp[x].S = vv[1];
}
void dfs2(ll x, ll par, ll w)
{
if(vis[x]) return;
vis[x] = 1;
ans[x] = 0;
if(par!=-1)
{
ll wh = dp[par].F;
if(dp[x].F+w==wh) wh = dp[par].S;
ans[x] = w+max(wh,ans[par]);
mn = min(mn,max(dp[x].F,ans[x]));
}
else
{
curans = max(curans,dp[x].F+dp[x].S);
mn = min(mn,dp[x].F);
}
for(int i = 0; i < v[x].sz; i++)
{
ll newn = v[x][i].F;
if(newn==par) continue;
dfs2(newn,x,v[x][i].S);
}
}
int travelTime(int N,int M,int L, int A[],int B[],int T[])
{
for(int i = 0; i < M; i++)
{
ll x = A[i], y = B[i], z = T[i];
v[x].push_back({y,z});
v[y].push_back({x,z});
}
for(int i = 0; i < N; i++)
{
if(!vis[i]) dfs(i,-1);
}
//for(int i = 0; i < N; i++) cout << dp[i].F << " " << dp[i].S << endl; cout << endl;
for(int i =0; i < N; i++) vis[i] = 0;
vector<ll> curv;
for(int i = 0; i < N; i++)
{
if(!vis[i])
{
mn = 1e18;
dfs2(i,-1,0);
curv.push_back(mn);
//cout << i << " " << mn << endl;
}
}
sort(all(curv)); reverse(all(curv));
if(curv.sz==1) return curv[0];
if(curv.sz==2) return max(curans,curv[0] + curv[1] + L);
else return max(max(curv[0]+curv[1]+L,curv[1]+curv[2]+2*L),curans);
}
Compilation message
dreaming.cpp: In function 'void dfs(ll, ll)':
dreaming.cpp:45:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
45 | for(int i = 0; i < v[x].sz; i++)
| ^
dreaming.cpp: In function 'void dfs2(ll, ll, ll)':
dreaming.cpp:74:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
74 | for(int i = 0; i < v[x].sz; i++)
| ^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
46 ms |
24212 KB |
Output is correct |
2 |
Correct |
50 ms |
23836 KB |
Output is correct |
3 |
Correct |
28 ms |
22608 KB |
Output is correct |
4 |
Correct |
10 ms |
12380 KB |
Output is correct |
5 |
Correct |
8 ms |
11612 KB |
Output is correct |
6 |
Correct |
12 ms |
13404 KB |
Output is correct |
7 |
Correct |
2 ms |
10588 KB |
Output is correct |
8 |
Correct |
29 ms |
17748 KB |
Output is correct |
9 |
Correct |
28 ms |
20820 KB |
Output is correct |
10 |
Correct |
3 ms |
10588 KB |
Output is correct |
11 |
Correct |
41 ms |
21080 KB |
Output is correct |
12 |
Correct |
52 ms |
22832 KB |
Output is correct |
13 |
Correct |
2 ms |
10588 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
10584 KB |
Output is correct |
2 |
Correct |
2 ms |
10584 KB |
Output is correct |
3 |
Correct |
3 ms |
10588 KB |
Output is correct |
4 |
Correct |
2 ms |
10588 KB |
Output is correct |
5 |
Correct |
2 ms |
10588 KB |
Output is correct |
6 |
Correct |
2 ms |
10584 KB |
Output is correct |
7 |
Correct |
3 ms |
10584 KB |
Output is correct |
8 |
Correct |
3 ms |
10584 KB |
Output is correct |
9 |
Incorrect |
2 ms |
10588 KB |
Output isn't correct |
10 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
46 ms |
24212 KB |
Output is correct |
2 |
Correct |
50 ms |
23836 KB |
Output is correct |
3 |
Correct |
28 ms |
22608 KB |
Output is correct |
4 |
Correct |
10 ms |
12380 KB |
Output is correct |
5 |
Correct |
8 ms |
11612 KB |
Output is correct |
6 |
Correct |
12 ms |
13404 KB |
Output is correct |
7 |
Correct |
2 ms |
10588 KB |
Output is correct |
8 |
Correct |
29 ms |
17748 KB |
Output is correct |
9 |
Correct |
28 ms |
20820 KB |
Output is correct |
10 |
Correct |
3 ms |
10588 KB |
Output is correct |
11 |
Correct |
41 ms |
21080 KB |
Output is correct |
12 |
Correct |
52 ms |
22832 KB |
Output is correct |
13 |
Correct |
2 ms |
10588 KB |
Output is correct |
14 |
Correct |
2 ms |
10584 KB |
Output is correct |
15 |
Correct |
2 ms |
10584 KB |
Output is correct |
16 |
Correct |
3 ms |
10588 KB |
Output is correct |
17 |
Correct |
2 ms |
10588 KB |
Output is correct |
18 |
Correct |
2 ms |
10588 KB |
Output is correct |
19 |
Correct |
2 ms |
10584 KB |
Output is correct |
20 |
Correct |
3 ms |
10584 KB |
Output is correct |
21 |
Correct |
3 ms |
10584 KB |
Output is correct |
22 |
Incorrect |
2 ms |
10588 KB |
Output isn't correct |
23 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
18 ms |
16072 KB |
Output is correct |
2 |
Correct |
19 ms |
16040 KB |
Output is correct |
3 |
Correct |
18 ms |
16092 KB |
Output is correct |
4 |
Correct |
18 ms |
16092 KB |
Output is correct |
5 |
Correct |
19 ms |
16196 KB |
Output is correct |
6 |
Correct |
19 ms |
16856 KB |
Output is correct |
7 |
Correct |
18 ms |
16348 KB |
Output is correct |
8 |
Correct |
19 ms |
16096 KB |
Output is correct |
9 |
Correct |
19 ms |
16128 KB |
Output is correct |
10 |
Correct |
26 ms |
16272 KB |
Output is correct |
11 |
Correct |
2 ms |
10584 KB |
Output is correct |
12 |
Correct |
5 ms |
14812 KB |
Output is correct |
13 |
Correct |
6 ms |
14548 KB |
Output is correct |
14 |
Correct |
7 ms |
14548 KB |
Output is correct |
15 |
Correct |
6 ms |
14548 KB |
Output is correct |
16 |
Correct |
5 ms |
14548 KB |
Output is correct |
17 |
Correct |
7 ms |
14548 KB |
Output is correct |
18 |
Correct |
6 ms |
14544 KB |
Output is correct |
19 |
Correct |
7 ms |
14548 KB |
Output is correct |
20 |
Correct |
2 ms |
10584 KB |
Output is correct |
21 |
Correct |
2 ms |
10588 KB |
Output is correct |
22 |
Correct |
3 ms |
10588 KB |
Output is correct |
23 |
Correct |
6 ms |
14548 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
10584 KB |
Output is correct |
2 |
Correct |
2 ms |
10584 KB |
Output is correct |
3 |
Correct |
3 ms |
10588 KB |
Output is correct |
4 |
Correct |
2 ms |
10588 KB |
Output is correct |
5 |
Correct |
2 ms |
10588 KB |
Output is correct |
6 |
Correct |
2 ms |
10584 KB |
Output is correct |
7 |
Correct |
3 ms |
10584 KB |
Output is correct |
8 |
Correct |
3 ms |
10584 KB |
Output is correct |
9 |
Incorrect |
2 ms |
10588 KB |
Output isn't correct |
10 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
46 ms |
24212 KB |
Output is correct |
2 |
Correct |
50 ms |
23836 KB |
Output is correct |
3 |
Correct |
28 ms |
22608 KB |
Output is correct |
4 |
Correct |
10 ms |
12380 KB |
Output is correct |
5 |
Correct |
8 ms |
11612 KB |
Output is correct |
6 |
Correct |
12 ms |
13404 KB |
Output is correct |
7 |
Correct |
2 ms |
10588 KB |
Output is correct |
8 |
Correct |
29 ms |
17748 KB |
Output is correct |
9 |
Correct |
28 ms |
20820 KB |
Output is correct |
10 |
Correct |
3 ms |
10588 KB |
Output is correct |
11 |
Correct |
41 ms |
21080 KB |
Output is correct |
12 |
Correct |
52 ms |
22832 KB |
Output is correct |
13 |
Correct |
2 ms |
10588 KB |
Output is correct |
14 |
Correct |
2 ms |
10584 KB |
Output is correct |
15 |
Correct |
2 ms |
10584 KB |
Output is correct |
16 |
Correct |
3 ms |
10588 KB |
Output is correct |
17 |
Correct |
2 ms |
10588 KB |
Output is correct |
18 |
Correct |
2 ms |
10588 KB |
Output is correct |
19 |
Correct |
2 ms |
10584 KB |
Output is correct |
20 |
Correct |
3 ms |
10584 KB |
Output is correct |
21 |
Correct |
3 ms |
10584 KB |
Output is correct |
22 |
Incorrect |
2 ms |
10588 KB |
Output isn't correct |
23 |
Halted |
0 ms |
0 KB |
- |