Submission #265608

#TimeUsernameProblemLanguageResultExecution timeMemory
265608cjoaDetecting Molecules (IOI16_molecules)C++11
31 / 100
87 ms65536 KiB
#include "molecules.h"

#include <iostream>
#include <vector>
#include <bitset>
#include <cassert>

using namespace std;

typedef bitset<500001> BS;

typedef vector<int> VI;
typedef vector<VI> VVI;

std::vector<int> find_subset(int l, int u, std::vector<int> w) {
   const int N = w.size();
// assert(N <= 100 && u <= 1000);

   vector<BS> can(1);
   can.reserve(N+1);
   can[0] = true;
   for (int n = 0; n < N; ++n) {
   // cerr << "* n:" << n << "  w:" << w[n] << endl;
   // cerr << can[n] << endl;
   // cerr << (can[n] << w[n]) << endl;
      can.emplace_back( can[n] | (can[n] << w[n]) );
   }

   for (int sum = l; sum <= u; ++sum) {
      if (can[N][sum]) {
         vector<int> res;
         for (int s = sum, n = N-1; s != 0; --n) {
            assert(n >= 0);
            if (s - w[n] >= 0 && can[n][s - w[n]]) {
               res.push_back(n);
               s -= w[n];
            }
         }
         return res;
      }
   }

   return {};
}
#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...