Submission #394891

#	Time	Username	Problem	Language	Result	Execution time	Memory
394891		suren	Swapping Cities (APIO20_swap)	C++14	0 / 100	2080 ms	95636 KiB

swap

#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[2000045];
pair < int , pair< int, int > > p[2000045];
int parent[2000045];
int sz[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 );
}
int getMinimumFuelCapacity(int X, int Y) {
	map < int, int > mp;
	int i;
	flag = false;
	mp.clear();
	for( i = 1; i <= n; i ++ ){
			parent[i] = i; 
	}
	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 ){
			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].S.F ].size() == 3 ){
			if( parent[p[i].S.F] == parent[X] || parent[p[i].S.F] == parent[Y] )
				flag = true;
		}
		if( vc[ p[i].S.S ].size() == 3 ){
			if( parent[p[i].S.S] == parent[X] || parent[p[i].S.S] == parent[Y] )
				flag = true;
		}
		if( mp.size() == i+1 && mp[X] == mp[Y] && mp[X] == 1 && parent[x] == parent[y] ){
			return p[i].first;
		}
		if( mp[X] == mp[Y] && mp[X] == 1 && flag == true && parent[x] == parent[y] )
			return p[i].first;
	}
	return -1;
}

Compilation message (stderr)

swap.cpp: In function 'int getMinimumFuelCapacity(int, int)':
swap.cpp:58:17: warning: comparison of integer expressions of different signedness: 'std::map<int, int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   58 |   if( mp.size() == i+1 && mp[X] == mp[Y] && mp[X] == 1 && parent[x] == parent[y] ){
      |       ~~~~~~~~~~^~~~~~
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] );
      |  ~~~~~~~~~~^~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 6
• Subtask 2: 0 / 7
• Subtask 3: 0 / 17
• Subtask 4: 0 / 20
• Subtask 5: 0 / 23
• Subtask 6: 0 / 27