Cod sursa(job #1480579)

Utilizator ciocan_catalinCiocan Catalin - Iulian ciocan_catalin Data 2 septembrie 2015 20:36:51
Problema Loto Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.69 kb
#include <fstream>
#include <algorithm>
#include <vector>
#define P 370003

using namespace std;
ifstream fin("loto.in");
ofstream fout("loto.out");

struct suma
{
    int val,i1,i2,i3;
};
vector<suma> h[370005];

int n,s,a[105],index[10];
bool ok = false;

void Solve()
{
    suma sum;
    int i,j,k,r,S,complement,t;
    for(i=1;i<=n;i++)
        for(j=1;j<=n;j++)
            for(k=1;k<=n;k++)
    {
        sum.val = 0;
        sum.val = a[i]+a[j]+a[k];
        sum.i1 = i;
        sum.i2 = j;
        sum.i3 = k;
        r = sum.val%P;
        if(h[r].size()==0 || h[r][0].val!=sum.val) h[r].push_back(sum);
    }
   for(i=1;i<=n && !ok;i++)
        for(j=1;j<=n && !ok;j++)
            for(k=1;k<=n && !ok;k++)
            {
                S  =a[i]+a[j]+a[k];
                complement = s-S;
                if(complement>0)
                {
                    r = complement%P;
                    for(t=0;t<h[r].size() && h[r][t].val!=complement;t++);
                    if(t < h[r].size() && h[r][t].val==complement)
                    {
                        index[1] = i;
                        index[2] = j;
                        index[3] = k;
                        index[4] = h[r][t].i1;
                        index[5] = h[r][t].i2;
                        index[6] = h[r][t].i3;
                        ok = true;
                    }
                }
            }
}

int main()
{
    fin>>n>>s;
    for(int i=1;i<=n;i++) fin>>a[i];
    Solve();
    if(ok)
    {
        sort(index+1,index+7);
        for(int i=1;i<=6;i++) fout<<index[i]<<" ";
        fout<<"\n";
    }
    else fout<<"-1\n";
    fout.close();
    return 0;
}