Submission #1319221

#	Time	Username	Problem	Language	Result	Execution time	Memory
1319221		tkm_algorithms	Detecting Molecules (IOI16_molecules)	C++20	100 / 100	36 ms	6156 KiB

molecules

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

using namespace std;
using ll = long long;
//#define int ll
using P = pair<ll, int>;
#define all(x) x.begin(), x.end()
#define rep(i, l, n) for (int i = l; i < (n); ++i)
#define sz(x) (int)x.size()
const char nl = '\n';
const int mod = 1e9+7;

vector<int> find_subset(int l, int u, vector<int> w) {
	vector<P> nw; int n = sz(w);
	rep(i, 0, n)nw.push_back(P(w[i], i));
	sort(all(nw));
	
	ll sm = 0, r = -1;
	rep(i, 0, n) {
		if (r < i)
			r = i, sm = nw[i].first;
		while (r < n-1 && sm < l)sm += nw[++r].first;
		if (l <= sm && sm <= u) {
			vector<int> res;
			rep(j, i, r+1)res.push_back(nw[j].second);
			return res;
		}
		
		sm -= nw[i].first;
	}
	
	return {};
}

Compilation message (stderr)

molecules.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~
molecules_c.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~

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