# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
849630 | Bach21 | Detecting Molecules (IOI16_molecules) | C++14 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
#include "molecules.h"
using namespace std;
#define int long long
vector<int> find_subset(int l, int u, vector<int> w)
{
pair<int,int> w1[w.size()+1];
for (int i=0;i<w.size();i++)
{
w1[i].first=w[i];
w1[i].second=i+1;
}
sort(w1,w1+w.size(),greater<pair<int,int>>());
int l1=0;
int r=w.size()-1;
int tmp=0;
while (r>l1)
{
if (tmp+w1[l1].first<=l)
{
tmp+=w1[l1].first;
l1++;
} else if (tmp+w1[l1].first>=l && tmp+w1[l1].first<=u)
{
tmp+=w1[l1].first;
l1++;
} else if (tmp+w1[l1].first>=u)
{
while (tmp<l)
{
tmp+=w1[r].first;
r--;
}
while (tmp>u)
{
tmp-=w1[l1].first;
l1--;
}
}
if (tmp<=u && tmp>=l) break;
}
if (tmp<u || tmp>l) return {};
vector<int> res;
for (int i=0;i<=l1;i++)
{
res.push_back(w1[i].second);
}
for (int i=r;i<w.size();i++)
{
res.push_back(w1[i].second);
}
return res;
}