제출 #798734

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
798734		NeroZein	Detecting Molecules (IOI16_molecules)	C++17	19 / 100	1 ms	304 KiB

molecules

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

std::vector<int> find_subset(int l, int r, std::vector<int> w) {
  int n = (int) w.size(); 
  vector<long long> suf(n); 
  vector<long long> pref(n);
  vector<pair<int, int>> a(n);
  for (int i = 0; i < n; ++i) {
    a[i].first = w[i];
    a[i].second = i; 
    pref[i] = suf[i] = a[i].first;
  }
  for (int i = 1; i < n; ++i) {
    pref[i] += pref[i - 1];
  }
  for (int i = n - 2; i >= 0; --i) {
    suf[i] += suf[i + 1];
  }
  auto print = [&](int x, int y) {
    vector<int> ans;
    for (int i = x; i <= y; ++i) {
      ans.push_back(a[i].second); 
    }
    return ans; 
  };
  for (int sz = 1; sz <= n; ++sz) {
    long long ps = pref[sz - 1];
    long long sf = suf[n - sz];
    if (ps >= l && ps <= r) {
      return print(0, sz - 1);
    } else if (sf >= l && sf <= r) {
      return print(n - sz, n - 1);
    } else if (ps < l && sf > r) {
      for (int i = sz - 1; i < n; ++i) {
        int rng = pref[i] - (i >= sz ? pref[i - sz] : 0);
        if (rng >= l && rng <= r) {
          return print(i - sz + 1, i);
        }
      }
    }
  }
  return {};
}

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