Cod sursa(job #2382197)

Utilizator butnaru_vlad2003Butnaru Vlad butnaru_vlad2003 Data 17 martie 2019 20:44:45
Problema Loto Scor 0
Compilator cpp-64 Status done
Runda cautare-binara Marime 1.93 kb
#include <bits/stdc++.h>
using namespace std;
ifstream in ("loto.in");
ofstream out ("loto.out");
struct suma{int x1,x2,x3,s;} sume[100001];
bool verfmutare(suma a,suma b)
{return a.s<b.s;}
int main()
{
    int n,s,cnt=1;
    int v[100001];
    in>>n>>s;
    for (int i=1;i<=n;++i)
        in>>v[i];
    for (int i=1;i<=n;++i)
        for (int k=i;k<=n;++k)
            for (int j=k;j<=n;++j)
                {
                    sume[cnt].s=v[i]+v[j]+v[k];
                    sume[cnt].x1=v[i];
                    sume[cnt].x2=v[j];
                    sume[cnt++].x3=v[k];
                }
    sort(sume+1 , sume+cnt,verfmutare);
    for (int i=1;i<=n;++i)
        for (int k=i;k<=n;++k)
            for (int j=k;j<=n;++j)
            {
                int s1=s-(v[i]+v[j]+v[k]);
                int l=1,r=(n+1)*(n+2)/2,ok=0,x,ok1=0;
                while (l<=r)
                {
                    int mid=(l+r)/2;
                    if (sume[mid].s==s1)
                    {
                        x=sume[mid].s;
                        ok=mid;//loc unde am gasit suma
                        ok1=1;
                    }
                    if (v[mid]<s1)
                        l=mid+1;
                    else
                        r=mid-1;
                }
                if (ok1==1 || i==n)
                {
                    int x1,x2,x3;
                    x1=sume[ok].x1;
                    x2=sume[ok].x2;
                    x3=sume[ok].x3;
                    int ver=v[x1]+v[x2]+v[x3]+v[i]+v[j]+v[k];
                    if (ver==s)
                    {
                        out<<v[i]<<' '<<v[j]<<' '<<v[k]<<' '<<v[x1]<<' '<<v[x2]<<' '<<v[x3];
                        return 0;
                    }
                    else
                    {
                        out<<-1;
                        return 0;
                    }
                }
            }
}