Submission #262200

#	Time	Username	Problem	Language	Result	Execution time	Memory
262200		stoyan_malinin	Detecting Molecules (IOI16_molecules)	C++14	46 / 100	1083 ms	2560 KiB

molecules

#include "molecules.h"
//#include "grader.cpp"

#include <bitset>
#include <cstring>

using namespace std;

const int MAXN = 5e5 + 5;

int dp[MAXN];

void filterVector(vector <int> &v, int lim)
{
    vector <int> newVal;
    for(int x: v)
    {
        if(x<=lim) newVal.push_back(x);
    }

    v = newVal;
}

vector <int> recoverAnswer(int x, vector <int> &w)
{
    vector <int> answer;

    while(x!=0)
    {
        answer.push_back(dp[x]);
        x = x - w[ dp[x] ];
    }

    return answer;
}

vector <int> solveSingle(int l, int u, vector <int> &w)
{
    for(int i = 0;i<w.size();i++)
    {
        if(w[i]>=l && w[i]<=u) return vector <int> {i};
    }

    return vector <int> {};
}

vector <int> specialCaseSolutions(int l, int u, vector <int> &w)
{
    vector <int> answer = {};

    answer = solveSingle(l, u, w);
    if(answer.empty()==false) return answer;
}

vector<int> find_subset(int l, int u, vector <int> w)
{
    memset(dp, -1, sizeof(dp));
    filterVector(w, u);

    vector <int> specialCase = specialCaseSolutions(l, u, w);
    if(specialCase.empty()==false) return specialCase;

    dp[0] = -2;
    for(int i = 0;i<w.size();i++)
    {
        for(int j = u-w[i];j>=max(l-w[i], 0);j--)
        {
            if(dp[j]!=-1 && dp[ j + w[i] ]==-1)
            {
                dp[ j + w[i] ] = i;
                return recoverAnswer(j+w[i], w);
            }
        }
        for(int j = l-w[i]-1;j>=0;j--)
        {
            if(dp[j]!=-1 && dp[ j + w[i] ]==-1) dp[ j + w[i] ] = i;
        }
    }

    return vector <int> {};
}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> solveSingle(int, int, std::vector<int>&)':
molecules.cpp:39:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   39 |     for(int i = 0;i<w.size();i++)
      |                   ~^~~~~~~~~
molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:64:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   64 |     for(int i = 0;i<w.size();i++)
      |                   ~^~~~~~~~~
molecules.cpp: In function 'std::vector<int> specialCaseSolutions(int, int, std::vector<int>&)':
molecules.cpp:53:1: warning: control reaches end of non-void function [-Wreturn-type]
   53 | }
      | ^

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