Submission #1288781

#	Time	Username	Problem	Language	Result	Execution time	Memory
1288781		muhammad-ahmad	Detecting Molecules (IOI16_molecules)	C++20	0 / 100	1 ms	336 KiB

molecules

#include <bits/stdc++.h>
#include "molecules.h"
using namespace std;

vector<int> find_subset(int l, int u, vector<int> w) {
	int n = w.size();
	vector<pair<int, int>> X;
	for (int i = 0; i < n; i++){
		X.push_back({w[i], i + 1});
	}
	sort(X.begin(), X.end());
	int a[n + 1], idx[n + 1];
	for (int i = 0; i < n; i++){
		a[i + 1] = X[i].first;
		idx[i + 1] = X[i].second;
	}
	int pref[n + 2] = {}, suf[n + 2] = {};
	
	for (int i = 1; i <= n; i++){
		pref[i] = a[i] + pref[i - 1];
	}
	for (int i = n; i >= 1; i--){
		suf[i] = suf[i + 1] + a[i];
	}
	
	suf[0] = u;
	int R = 0;
	for (int L = 0; L <= n; L++){
		while (R <= n && pref[L] + suf[R]) R++;
		
		if (pref[L] + suf[R] >= l && pref[L] + suf[R] <= u && R > L){
			vector<int> ans;
			for (int i = 1; i <= L; i++) ans.push_back(idx[i] - 1);
			for (int i = R; i <= n; i++) ans.push_back(idx[i] - 1);
			return ans;
		}
	}
	
	return {1};
	
}

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: 0 / 9
• Subtask 2: 0 / 10
• Subtask 3: 0 / 12
• Subtask 4: 0 / 15
• Subtask 5: 0 / 23
• Subtask 6: 0 / 31