Submission #50293

#	Time	Username	Problem	Language	Result	Execution time	Memory
50293		faishol27	Detecting Molecules (IOI16_molecules)	C++14	100 / 100	88 ms	4564 KiB

molecules

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

typedef long long ll;
typedef pair<int,int> ii;

#define PUB push_back
#define fi first
#define se second

bool cmp(ii a, ii b){
    if(a.fi == b.fi) return a.se < b.se;
	return a.fi < b.fi;
}

std::vector<int> find_subset(int l, int u, std::vector<int> w) {
    int N = w.size(),
        ki = -1,
        ka = 0;
    ll  sum = 0;
    vector<int> ans;
    vector<ii>data;

    for(int i=0;i<w.size();i++) data.PUB({w[i],i});

    sort(data.begin(), data.end(), cmp);
    
	while(ki!=ka && ka < N && ki < N){
        if(l <= sum+data[ka].fi && sum+data[ka].fi <= u){
            sum += data[ka].fi;
            break;
        }else if(sum+data[ka].fi < l){
            sum += data[ka].fi;
            ka++;
        }else if(sum+data[ka].fi > u){
            sum -= data[ki+1].fi;
            ki++;
        }
    }

    if(l <= sum && sum <= u){
        ki++;
        while(ki<=ka){
            ans.PUB(data[ki].se);
            ki++;
        }
    }

    return ans;
}

Compilation message (stderr)

molecules.cpp: In function 'std::vector<int> find_subset(int, int, std::vector<int>)':
molecules.cpp:25:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<w.size();i++) data.PUB({w[i],i});
                 ~^~~~~~~~~

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