Cod sursa(job #473544)

Utilizator andrey932Andrei andrey932 Data 30 iulie 2010 12:12:30
Problema Loto Scor 5
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.25 kb
#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;

FILE *fin=fopen("loto.in","r"), *fout=fopen("loto.out","w");
int sum[8000008],S,sm,n,i,j,k,nr[202],tsm,lo,hi,mi,rez[6];
bool ok;

int main() {
	fscanf(fin,"%i %i",&n,&S);
	for(i=0;i<n;i++) {
		fscanf(fin,"%i",&nr[i]);
	}
	tsm=-1;
	for(i=0;i<n;i++) {
		for(j=i;j<n;j++) {
			for(k=j;k<n;k++) {
				tsm++;
				sum[tsm]=nr[i]+nr[j]+nr[k];
			}
		}
	}
	sort(&sum[0],&sum[tsm]);
	for(i=0;i<tsm;i++) cout<<sum[i]<<"\n";

	ok=false;
	for(i=0;i<=tsm;i++) {
		sm=S-sum[i];
		//cout<<"Caut "<<sm<<"\n";
		lo=0; hi=tsm;
		if (sm>0) {
		while(lo+1<hi) {
			mi=(lo+hi)>>1;
			//cout<<"-----"<<sum[mi]<<"\n";
			if (sm<sum[mi]) {
				hi=mi;
			}
			else if (sm>sum[mi]) {
				lo=mi;
			}
			else break;			
		}	
		if (sm==sum[lo]) mi=lo; if (sm==sum[hi]) mi=hi;
		if (sum[i]+sum[mi]==S) {
			ok=true;
			rez[0]=nr[i%n];
			rez[1]=nr[(i/n)%n];
			rez[2]=nr[(i/n/n)%n];
			rez[3]=nr[mi%n];
			rez[4]=nr[(mi/n)%n];
			rez[5]=nr[(mi/n/n)%n];
			sort(&rez[0],&rez[5]);
			fprintf(fout,"%i %i %i %i %i %i",rez[0],rez[1],rez[2],rez[3],rez[4],rez[5]);
			break;
		}
		}
	}
if (!ok) fprintf(fout,"%s","-1");
fprintf(fout,"%s","\n");
fclose(fout);
	return 0;
}