Cod sursa(job #473539)

Utilizator andrey932Andrei andrey932 Data 30 iulie 2010 12:01:04
Problema Loto Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.96 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;
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]);

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