제출 #419263

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
419263		someone	Detecting Molecules (IOI16_molecules)	C++14	100 / 100	55 ms	7636 KiB

molecules

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

struct Molecule {
    ll val, i;
};

std::vector<int> find_subset(int l, int u, std::vector<int> w) {
    std::vector<Molecule> mol;
    int n = w.size();
    ll sum[n+1];
    sum[0] = 0;
    for(int i = 0; i < n; i++)
        mol.push_back({w[i], i});
    std::sort(mol.begin(), mol.end(),
    [](Molecule a, Molecule b) {
        return a.val < b.val;
    });
    for(int i = 0; i < n; i++)
        sum[i+1] = sum[i] + mol[i].val;
    std::vector<int> ans;
    bool hasAnswer = false;
    for(int i = 1; i <= n && !hasAnswer; i++) {
        ll mini = sum[i],
            maxi = sum[n] - sum[n-i];
        if(mini <= u && l <= maxi) {
            for(int deb = 0, fin = i; fin <= n && !hasAnswer; deb++, fin++) {
                int s = sum[fin] - sum[deb];
                if((ll)l <= s && s <= (ll)u) {
                    hasAnswer = true;
                    for(int j = deb; j < fin; j++)
                        ans.push_back(mol[j].i);
                }
            }
        }
    }
    return ans;
}

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