# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
971001 | KasymK | Detecting Molecules (IOI16_molecules) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "bits/stdc++.h"
using namespace std;
vector<int> find_subset(int l, int u, vector<int> w){
int n = (int)w.size();
vector<pair<int, int>> v;
for(int i = 0; i < n; ++i)
v.push_back({i, w[i]});
auto cmp = [&](pair<int, int> a, pair<int, int> b) -> bool {
if(a.second != b.second)
return a.second > b.second;
return a.first > b.first;
};
sort(v.begin(), v.end(), cmp);
long long answer = 0;
vector<int> ans;
for(int i = 0, j = 0; j < n; ++j){
answer += v[j].second;
while(answer > u)
answer -= v[i++].second;
if(answer >= l){
for(int k = i; k <= j; ++k)
ans.push_back(v[k].first);
return ans;
}
}
return vector<int> {};
}
int main(){
int n, l, u;
scanf("%d%d%d", &n, &l, &u);
vector<int> v(n);
for(int &i : v)
scanf("%d", &i);
vector<int> ans = find_subset(l, u, v);
for(int i : ans)
printf("%d ", i);
puts("");
return 0;
}