# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
299153 | azatega | Detecting Molecules (IOI16_molecules) | C++11 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "molecules.h"
#include <vector>
#include <algorithm>
#include <iostream>
using namespace std;
#define ll long long
vector<int> find_subset(int l, int u, vector<int> w) {
vector<pair<ll, int>> new_w;
vector<int> bl;
int n = w.size();
ll tot_sum = 0;
for(int i = 0; i < n; i++){
tot_sum += w[i];
new_w.push_back({w[i], i});
}
sort(new_w.begin(), new_w.end());
ll currsum = 0;
int tot_sum_idx = 0;
for(int i = 0; i < n; i++){
currsum += new_w[i].first;
if(currsum > u)
break;
while(tot_sum_idx <= i)
tot_sum -= new_w[tot_sum_idx++].first;
while(tot_sum + currsum > u)
tot_sum -= new_w[tot_sum_idx++].first;
if(tot_sum + currsum >= l){
vector<int> res;
for(int x = 0; x <= i; x++)
res.push_back(new_w[x].second);
for(int x = tot_sum_idx; x < n; x++)
res.push_back(new_w[x].second);
return res;
}
}
return bl;
}
int main(){
int n, l, u;
vector<int> niz;
cin >> n >> l >> u;
niz.resize(n);
for(int i = 0; i < n; i++)
cin >> niz[i];
vector<int> rs = find_subset(l, u, niz);
for(int i = 0; i < rs.size(); i++)
cout << rs[i] << endl;
}