Cod sursa(job #608495)

Utilizator a_h1926Heidelbacher Andrei a_h1926 Data 16 august 2011 22:50:20
Problema Loto Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.22 kb
#include <iostream>
#include <cstdio>
#include <vector>

#define U 666013
#define NMax 105

using namespace std;

vector <int> Hash[U+5];
int N, Sum, Sum2, V[NMax], S[7];

bool Search (int X)
{
    int Key=X%U;
    for (unsigned i=0; i<Hash[Key].size (); ++i)
    {
        if (Hash[Key][i]==X)
        {
            return true;
        }
    }
    return false;
}

void Insert (int X)
{
    int Key=X%U;
    if (!Search (X))
    {
        Hash[Key].push_back (X);
    }
}

int main()
{
    freopen ("loto.in", "r", stdin);
    freopen ("loto.out", "w", stdout);
    scanf ("%d %d", &N, &Sum);
    for (int i=1; i<=N; ++i)
    {
        scanf ("%d", &V[i]);
    }
    for (int i=1; i<=N; ++i)
    {
        for (int j=1; j<=N; ++j)
        {
            for (int k=1; k<=N; ++k)
            {
                if (V[i]+V[j]+V[k]<=Sum)
                {
                    Insert (V[i]+V[j]+V[k]);
                }
            }
        }
    }
    S[0]=-1;
    for (int i=1; i<=N; ++i)
    {
        for (int j=1; j<=N; ++j)
        {
            for (int k=1; k<=N; ++k)
            {
                if (Search (Sum-V[i]-V[j]-V[k]))
                {
                    S[0]=i;
                    S[1]=j;
                    S[2]=k;
                    Sum2=Sum-V[i]-V[j]-V[k];
                    break;
                }
            }
            if (S[0]==i)
            {
                break;
            }
        }
        if (S[0]==i)
        {
            break;
        }
    }
    if (S[0]==-1)
    {
        printf ("-1\n");
        return 0;
    }
    for (int i=1; i<=N; ++i)
    {
        for (int j=1; j<=N; ++j)
        {
            for (int k=1; k<=N; ++k)
            {
                if (V[i]+V[j]+V[k]==Sum2)
                {
                    S[3]=i;
                    S[4]=j;
                    S[5]=k;
                    break;
                }
            }
            if (S[3]==i)
            {
                break;
            }
        }
        if (S[3]==i)
        {
            break;
        }
    }
    for (int i=0; i<6; ++i)
    {
        printf ("%d ", S[i]);
    }
    printf ("\n");
    return 0;
}