Submission #719980

#	Time	Username	Problem	Language	Result	Execution time	Memory
719980		vinnipuh01	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	105 ms	14928 KiB

molecules

#include "molecules.h"
#include <bits/stdc++.h>
//#define int long long

using namespace std;

vector <int> v;

vector<int> find_subset(int l , int u , vector<int> w) {
	set <pair<int, int> > st, s;
	long long sum = 0;
	vector <pair<int, int> > ww;
	for ( int i = 0; i < w.size(); i ++ )
		ww.push_back( { w[  i], i } );
	sort( ww.begin(), ww.end() );
	if ( ww.front().first > u ) {
		vector <int> v;
		return v;
	}
	else if ( ww.front().first <= u && ww.front().first >= l ) {
		vector <int> v;
		v.push_back( ww.front().second );
		return v;
	}
	for ( int i = 0; i < ww.size(); i ++ ) {
		if ( sum + ww[ i ].first <= u )
			st.insert( { ww[ i ].first, ww[ i ].second } ), sum = sum + ww[ i ].first;
		else
			s.insert( { ww[ i ].first, ww[ i ].second } );
	}
	while ( sum < l ) {
		if ( !s.size() )
			break;
		sum = sum + s.rbegin()->first - st.begin()->first;
		st.erase( st.begin() );
		st.insert( *s.rbegin() );
		s.erase( --s.end() );
	}
	vector <int> v;
	v.clear();
	if ( sum < l )
		return v;
	for ( auto i : st )
		v.push_back( i.second );
	sort( v.begin(), v.end() );
    return v;
}

//main () {
//	int l, r;
//	cin >> l >> r;
//	int n;
//	cin >> n;
//	int num;
//	for ( int i = 1; i <= n; i ++ ) {
//		cin >> num;
//		v.push_back( num );
//	}
//	v = find_subset( l, r, v );
//	if ( v.size() )
//		cout << "YES\n";
//	else
//		cout << "NO\n";
//}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:13:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   13 |  for ( int i = 0; i < w.size(); i ++ )
      |                   ~~^~~~~~~~~~
molecules.cpp:25:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   25 |  for ( int i = 0; i < ww.size(); i ++ ) {
      |                   ~~^~~~~~~~~~~

• Subtask 1: 9 / 9
• Subtask 2: 10 / 10
• Subtask 3: 12 / 12
• Subtask 4: 15 / 15
• Subtask 5: 23 / 23
• Subtask 6: 31 / 31