Submission #126788

#	Time	Username	Problem	Language	Result	Execution time	Memory
126788		losmi247	Detecting Molecules (IOI16_molecules)	C++14	31 / 100	17 ms	14940 KiB

molecules

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair <int,int> p;
const int N = 2e5+45;

int n,a[N];
p go[105][10002];
p dp[105][10002];
vector <int> sol;

vector <int> find_subset(int l,int u,vector <int> w){
    n = w.size();
    for(int i = 0; i < n; i++){
        a[i+1] = w[i];
    }
    dp[0][0] = p(1,0);
    for(int i = 1; i <= n; i++){
        dp[i][0] = p(1,0);
        for(int j = 1; j <= 10002; j++){
            if(dp[i-1][j].first){
                dp[i][j] = p(1,0);
                go[i][j] = p(i-1,j);
            }
            if(j >= a[i] && dp[i-1][j-a[i]].first){
                dp[i][j] = p(1,1);
                go[i][j] = p(i-1,j-a[i]);
            }
            //cout << i << " " << j << " " << dp[i][j].first << endl;
        }
    }
    p poc = p(0,0);
    for(int j = l; j <= u; j++){
        if(dp[n][j].first){
            poc = p(n,j);
            break;
        }
    }
    while(poc.first){
        if(dp[poc.first][poc.second].second){
            sol.push_back(poc.first-1);
        }
        poc = go[poc.first][poc.second];
    }
    return sol;
}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:21:27: warning: iteration 10001 invokes undefined behavior [-Waggressive-loop-optimizations]
             if(dp[i-1][j].first){
                ~~~~~~~~~~~^~~~~
molecules.cpp:20:26: note: within this loop
         for(int j = 1; j <= 10002; j++){
                        ~~^~~~~~~~

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