#include "molecules.h" #include "bits/stdc++.h" using namespace std; const int N = 2e5 + 5; int pr[N]; int get(int l, int r){ if(l == 0)return pr[r]; return pr[r] - pr[l - 1]; } vector < int > find_subset(int l, int u, vector < int > w){ int n = w.size(); map < int , vector < int > > id; pr[0] = w[0]; for(int i = 1; i < n; i++) pr[i] = pr[i - 1] + w[i]; for(int i = 0; i < w.size(); i++) id[w[i]].push_back(i); sort(w.begin(), w.end()); for(int i = 0; i < n; i++){ int lo = i, hi = n - 1, j = i; while(hi >= lo){ int mid = (lo + hi) >> 1; if(w[mid] - w[i] <= u - l){ lo = mid + 1; j = mid; } else hi = mid - 1; } int z = i; lo = i, hi = j; while(hi >= lo){ int mid = (lo + hi) >> 1; if(get(i, mid) >= l){ hi = mid - 1; z = mid; } else lo = mid + 1; } int ans = get(i, z); if(ans >= l && ans <= u){ vector < int > rs, y; for(int k = i; k <= z; k++) y.push_back(w[k]); for(auto i : y){ rs.push_back(id[i].back()); id[i].pop_back(); } reverse(rs.begin(), rs.end()); return rs; } } return {}; }

molecules.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~
molecules_c.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~