Submission #445567

#	Time	Username	Problem	Language	Result	Execution time	Memory
445567		hgmhc	Detecting Molecules (IOI16_molecules)	C++17	31 / 100	200 ms	716 KiB

molecules

#include "molecules.h"
#include <bits/stdc++.h>
#define REP(i,a,b) for (auto i = (a); i <= (b); ++i)
#define PER(i,a,b) for (auto i = (b); i >= (a); --i)
#define ALL(x) (x).begin(), (x).end()
using namespace std;
using ii = pair<int,int>;
const int N = 20'0000;
bool dp[1003];
vector<int> v[1003];
vector<ii> mol;

std::vector<int> find_subset(int l, int u, std::vector<int> W) {
    REP(i,0,size(W)-1) mol.push_back({W[i],i});
    dp[0] = true;
    for (auto [w,i] : mol) {
        PER(x,0,1000) {
            if (x+w > 1000 || !dp[x]) continue;
            dp[x+w] = true;
            v[x+w].assign(ALL(v[x]));
            v[x+w].push_back(i);
            if (l <= x+w && x+w <= u) {
                sort(ALL(v[x+w]));
                return v[x+w];
            }
        }
    }
	return std::vector<int>(0);
}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:3:41: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    3 | #define REP(i,a,b) for (auto i = (a); i <= (b); ++i)
      |                                         ^
molecules.cpp:14:5: note: in expansion of macro 'REP'
   14 |     REP(i,0,size(W)-1) mol.push_back({W[i],i});
      |     ^~~

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