#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mk make_pair
#define F first
#define S second
int n, m, flag = true;
vector < int > u, v, w;
vector < int > vc[200045];
pair < int , pair< int, int > > p[200045];
int sz[200045];
int parent[200045];
int get( ll x ){
if( parent[x] == x )
return x;
parent[x] = get( parent[x] );
}
void init(int N, int M,
std::vector<int> U, std::vector<int> V, std::vector<int> W) {
int i;
n = N;m = M;u = U;v = V;w = W;
for( i = 0; i < m; i ++ ){
p[i].second.F=u[i];
p[i].second.S=v[i];
p[i].first=w[i];
}
sort( p, p + m );
for( i = 1; i <= n; i ++ ){
sz[i] = 1; parent[i] = i;
}
}
bool vis[200045];
int getMinimumFuelCapacity(int X, int Y) {
map < int, int > mp;
int i;
flag = false;
mp.clear();
for( i = 0; i < n; i ++ )
vc[i].clear();
for( i = 0; i < m; i ++ ){
int x = get( p[i].second.F );
int y = get( p[i].second.S );
if( x != y ){
if( sz[x] < sz[y] )
swap( x, y );
sz[x] += sz[y];
parent[y] = x;
}
vc[ p[i].second.F ].pb( p[i].second.S );
vc[ p[i].second.S ].pb( p[i].second.F );
mp[p[i].second.F] = 1;
mp[ p[i].second.S ] = 1;
if( vc[ p[i].second.F ].size() == 3 || vc[p[i].S.S].size() == 3 ){
flag = true;
}
if( mp.size() == i+1 && mp[X] == mp[Y] && mp[X] == 1 ){
return p[i].first;
}
if( mp[X] == mp[Y] && mp[X] == 1 && flag == true )
return p[i].first;
}
return -1;
}
Compilation message
swap.cpp: In function 'int getMinimumFuelCapacity(int, int)':
swap.cpp:57:17: warning: comparison of integer expressions of different signedness: 'std::map<int, int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
57 | if( mp.size() == i+1 && mp[X] == mp[Y] && mp[X] == 1 ){
| ~~~~~~~~~~^~~~~~
swap.cpp: In function 'int get(long long int)':
swap.cpp:17:12: warning: control reaches end of non-void function [-Wreturn-type]
17 | parent[x] = get( parent[x] );
| ~~~~~~~~~~^~~~~~~~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
3 ms |
4944 KB |
Output is correct |
| 2 |
Runtime error |
8 ms |
9932 KB |
Execution killed with signal 11 |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
3 ms |
4944 KB |
Output is correct |
| 2 |
Runtime error |
8 ms |
9932 KB |
Execution killed with signal 11 |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
3 ms |
4944 KB |
Output is correct |
| 2 |
Runtime error |
8 ms |
9932 KB |
Execution killed with signal 11 |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
3 ms |
4944 KB |
Output is correct |
| 2 |
Runtime error |
8 ms |
9932 KB |
Execution killed with signal 11 |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
3 ms |
4944 KB |
Output is correct |
| 2 |
Runtime error |
8 ms |
9932 KB |
Execution killed with signal 11 |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
3 ms |
4944 KB |
Output is correct |
| 2 |
Runtime error |
8 ms |
9932 KB |
Execution killed with signal 11 |
