Submission #1332174

#TimeUsernameProblemLanguageResultExecution timeMemory
1332174ChinguunDetecting Molecules (IOI16_molecules)C++20
100 / 100
36 ms6168 KiB
#include "molecules.h"
#include <bits/stdc++.h>
using namespace std;

// #define int long long
#define ff first
#define ss second
#define pb push_back
#define ppb pop_back
#define meta int tm = (tl + tr) / 2, x = i * 2 + 1, y = x + 1

const int N = 2e5 + 7;
const int TN = 4 * N;
// const int oo = 1e18;
const int mod = 1e9 + 7;

typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<pii> vii;

vector<int> find_subset(int l, int u, vector<int> w) {
    int n = w.size();
    vector<pair<long long, int> > v;
    for (int i = 0; i < n; i++)
        v.pb({w[i], i});

    sort(v.begin(), v.end());

    long long sum = 0, tl = 0;
    for (int tr = 0; tr < n; tr++) {
        sum += v[tr].ff;
        while (sum > u) {
            sum -= v[tl].ff;
            tl++;
        }
        if (sum >= l && sum <= u) {
            vi ans;
            for (int i = tl; i <= tr; i++)
                ans.pb(v[i].second);
            return ans;
        }
    }
    return {};
}

// signed main() {
//     int n, l, u;
//     assert(3 == scanf("%d %d %d", &n, &l, &u));
//     vector<int> w(n);
//     for (int i = 0; i < n; i++)
//         assert(1 == scanf("%d", &w[i]));
//     vector<int> result = find_subset(l, u, w);
    
//     cout << result.size() << '\n';
//     for (int i = 0; i < result.size(); i++) {
//         cout << result[i] << ' ';
//     }
//     // printf("%d\n", (int)result.size());
//     // for (int i = 0; i < (int)result.size(); i++)
//     //     printf("%d%c", result[i], " \n"[i == (int)result.size() - 1]);
// }
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...