Submission #985251

#TimeUsernameProblemLanguageResultExecution timeMemory
985251normankr07Detecting Molecules (IOI16_molecules)C++17
Compilation error
0 ms0 KiB
#define taskname ""
#include "molecules.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;

#ifndef LOCAL
#define cerr \
    if (0)   \
    cerr
#endif

const bool multitest = 0;

int n, l, u;
vector<int> w;

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

void solve()
{
    cin >> n >> l >> u;
    w.assign(n, 0);
    for (auto &v : w)
        cin >> v;
    auto ans = find_subset(l, u, w);
    for (auto v : ans)
    {
        cout << v << ' ';
    }
}

int32_t main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    if (fopen(taskname ".inp", "r"))
    {
        freopen(taskname ".inp", "r", stdin);
        freopen(taskname ".out", "w", stdout);
    }
    int tc = 1;
    if (multitest)
        cin >> tc;
    while (tc--)
    {
        solve();
        cout << '\n';
    }
}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:20:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   20 |     for (int i = 1; i <= w.size(); i++)
      |                     ~~^~~~~~~~~~~
molecules.cpp: In function 'int32_t main()':
molecules.cpp:84:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   84 |         freopen(taskname ".inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~
molecules.cpp:85:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   85 |         freopen(taskname ".out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
/usr/bin/ld: /tmp/ccFfVmRI.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccCOK6RI.o:molecules.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status