제출 #987278

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
987278		normankr07	Detecting Molecules (IOI16_molecules)	C++17	100 / 100	43 ms	6340 KiB

molecules

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

std::vector<int> find_subset(int l, int u, std::vector<int> w)
{
    // for (int i = 1; i <= n; i++)
    // {
    //     if (l <= w[i - 1] && w[i - 1] <= u)
    //     {
    //         return vector<int>(i);
    //     }
    // }
    int n = w.size();
    vector<pair<int, int>> ww;
    ll sum = 0;
    for (int i = 0; i < n; i++)
    {
        sum += w[i];
        ww.push_back(make_pair(w[i], i));
    }
    sort(ww.begin(), ww.end());
    vector<int> id;
    for (int i = n - 1, j = n; i < j && i >= -1;)
    {
        while (sum > u && i >= 0)
        {
            sum -= ww[i].first;
            i--;
        }
        if (l <= sum && sum <= u)
        {
            for (int p = 0; p <= i; p++)
                id.push_back(ww[p].second);
            for (int p = j; p < n; p++)
                id.push_back(ww[p].second);
            return id;
        }
        j--;
        sum += ww[j].first;
    }
    return vector<int>();
}

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