Cod sursa(job #2407954)

Utilizator DovlecelBostan Andrei Dovlecel Data 17 aprilie 2019 13:28:49
Problema Loto Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.09 kb
#include <fstream>
#include <algorithm>

using namespace std;
ifstream in("loto.in");
ofstream out("loto.out");
int n,s,cnt,sol[10],v[105];
struct trei
{
    int sum,a,b,c;
}a[1000005];
bool cmp(trei x,trei y)
{
    return(x.sum<y.sum);
}
int binsc(int poz,int x)
{
    int p=poz;
    for(int bit=28;bit>=0;bit--)
    {
        if(p+(1<<bit)<=cnt && a[p+(1<<bit)].sum<=x)
            p=p+(1<<bit);
    }
    return p;
}
int main()
{
    in>>n>>s;
    for(int i=1;i<=n;i++)
        in>>v[i];
    for(int i=1;i<=n;i++)
        for(int j=i;j<=n;j++)
            for(int k=j;k<=n;k++)
            {
                a[++cnt].sum=v[i]+v[j]+v[k];
                a[cnt].a=v[i];
                a[cnt].b=v[j];
                a[cnt].c=v[k];
            }
    sort(a+1,a+cnt+1,cmp);
    int poz;
    for(int i=1;i<=cnt;i++)
    {
        poz=binsc(i,s-a[i].sum);
        if(a[poz].sum==s-a[i].sum)
        {
            out<<a[i].a<<' '<<a[i].b<<' '<<a[i].c<<' '<<a[poz].a<<' '<<a[poz].b<<' '<<a[poz].c;
            return 0;
        }
    }
    out<<-1;
    return 0;
}