Cod sursa(job #2325809)

Utilizator shantih1Alex S Hill shantih1 Data 22 ianuarie 2019 22:32:22
Problema Loto Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.42 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>

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

int n,s,i,j,l,nr,st,dr,m,nd,md=666013;
int v[105];
struct per
{	int s,t1,t2,t3;	
	bool operator<(const per &alt)const
	{	return s<alt.s;		}
} z,d[1000005];
vector<per> H[666113];

void adaug(int s,int t1,int t2,int t3)
{
	int id=s%md,ok=1;
	for(auto i:H[id])
		if(i.s==s)	ok=0;
	if(ok)	H[id].push_back({s,t1,t2,t3});
}
per vezi(int s)
{
	int id=s%md;
	for(auto i:H[id])
		if(i.s==s)
			return i;
	return {0,0,0,0};
}

int main() {
	
	fin>>n>>s;
	for(i=1;i<=n;i++)
		fin>>v[i];
	
	for(i=1;i<=n;i++)
		for(j=1;j<=n;j++)
			for(l=1;l<=n;l++)
				adaug(v[i]+v[j]+v[l], v[i], v[j], v[l]);
	
	for(i=1;i<md;i++)
		for(auto j:H[i])
			d[++nd]=j;
	sort(d+1,d+nd+1);
	
	for(i=1;i<=n;i++)
		for(j=1;j<=n;j++)
			for(l=1;l<=n;l++)
			{
				nr=s-(v[i]+v[j]+v[l]);
				
				st=1;	dr=nd;
				while(st<=dr)
				{
					m=(st+dr)/2;
					if(d[m].s==nr)
					{
						fout<<v[i]<<" "<<v[j]<<" "<<v[l]<<" "<<d[m].t1<<" "<<d[m].t2<<" "<<d[m].t3<<"\n";
						return 0;
					}
					if(d[m].s<nr)	st=m+1;
					else	dr=m-1;
				}
				if(d[m-1].s==nr)
				{
					fout<<v[i]<<" "<<v[j]<<" "<<v[l]<<" "<<d[m].t1<<" "<<d[m].t2<<" "<<d[m].t3<<"\n";
					return 0;
				}
				if(d[m+1].s==nr)
				{
					fout<<v[i]<<" "<<v[j]<<" "<<v[l]<<" "<<d[m].t1<<" "<<d[m].t2<<" "<<d[m].t3<<"\n";
					return 0;
				}
			}
	fout<<-1<<"\n";
}