# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
223779 | bharat2002 | Detecting Molecules (IOI16_molecules) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
/*input
10 20
4
15 17 16 18
*/
#include "molecules.h"
#include<bits/stdc++.h>
using namespace std;
const int N=2e5 + 100;
const int mod=1e9 + 7;
#define pii pair<long long, int>
#define int long long
#define mp make_pair
#define f first
#define s second
int n;vector< pii > vals;
bool sf(pii a, pii b);long long pref[N];
vector<int> ans;int L, U;
void check(int num)
{
if(pref[num-1]<=U)
{
for(int i=0;i<num;i++) ans.push_back(vals[i].s);return;
}
for(int i=num;i<n;i++)
{
if(pref[i]-pref[i-num]>=L&&pref[i]-pref[i-num]<=U)
{
for(int j=i-num+1;j<=i;j++) ans.push_back(vals[j].s);return;
}
}
}
#undef int long long
vector<int> find_subset(long long l, long long u, vector< long long > w)
{
#define int long long
vals.clear();L=l;U=u;
n=w.size();
for(int i=0;i<w.size();i++)
{
vals.push_back(mp(w[i], i));
}
sort(vals.begin(), vals.end(), sf);
pref[0]=0;;
for(int i=0;i<vals.size();i++)
{
pref[i]=vals[i].f;
if(i>0) pref[i]+=pref[i-1];
}
for(int i=1;i<=n;i++)
{
if(i==n)
{
if(pref[n-1]>=l&&pref[n-1]<=u)
{
for(int i=0;i<n;i++) ans.push_back(vals[i].s);
}
break;
}
int maxsum=pref[i-1], minsum=pref[n-1] - pref[n-i-1];
// cout<<maxsum<<" "<<minsum<<endl;
if(minsum>u) break;
if(maxsum<l) continue;check(i);if(!ans.empty()) break;
}
#undef int long long
vector<int> ret;
for(int i=0;i<ans.size();i++) ret.push_back((int)ans[i]);
#define int long long
return ret;
}
bool sf(pii a, pii b)
{
return a.f>b.f;
}
/*signed main()
{
int l, u;cin>>l>>u;
int n;cin>>n;vector<int> vs;
for(int i=1;i<=n;i++)
{
int v;cin>>v;vs.push_back(v);
}
find_subset(l, u, vs);
}*/