Submission #654741

#	Time	Username	Problem	Language	Result	Execution time	Memory
654741		atigun	Detecting Molecules (IOI16_molecules)	C++14	100 / 100	47 ms	8620 KiB

molecules

#include<bits/stdc++.h>
#include"molecules.h"

using namespace std;
typedef long long ll;

vector<int> find_subset(int l, int u, vector<int> w){
  int n = w.size();
  vector<pair<int, int>> P(n+5);
  vector<int> ret;
  for(int i = 1; i <= n; i++)
    P[i] = make_pair(w[i-1], i-1);
  sort(P.begin()+1, P.begin()+n+1);
  vector<ll> prefix(n+5), suffix(n+5);
  for(int i = 1; i <= n; i++)
    prefix[i] = prefix[i-1]+P[i].first;
  for(int i = n; i >= 1; i--)
    suffix[i] = suffix[i+1]+P[i].first;
  for(int k = 1; k <= n; k++){
    if(prefix[k] <= u && suffix[n-k+1] >= l){
      int L = 1, R = k;
      ll sum = prefix[k];
      while(R <= n){
        if(sum <= u && sum >= l){
          for(int i = L; i <= R; i++)
            ret.push_back(P[i].second);
          return ret;
        }
        sum+= P[R+1].first;
        sum-= P[L].first;
        L++, R++;
      }
    }
  }
  return ret;
}

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