Cod sursa(job #1009076)

Utilizator miu_mik93FMI - Paduraru Miruna miu_mik93 Data 12 octombrie 2013 13:48:18
Problema Loto Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.28 kb
#include <stdio.h>
#include <math.h>
#include <iostream>
#include <string>
#include <stdlib.h>
#define NMax 101
#define NSum 1000000+1
using namespace std;
struct SUM
{
	int x1, x2, x3;
};
void merge_sort(int li, int ls, int *v, SUM *b)
{
	if(li < ls)
	{
		int m = (li + ls) / 2;
		merge_sort(li, m, v, b);
		merge_sort(m+1, ls, v, b);
		int i = li, j = m+1, k = 0;
		int *rez = new int[NSum];
		SUM *rez1 = new SUM[NSum];
		while(i <= m && j <= ls)
		{
			if(v[i] < v[j])
			{
				rez[k] = v[i++];
				rez1[k] = b[i++];
				k++;
			}
			else
			{
				rez[k] = v[j++];
				rez1[k] = b[j++];
				k++;
			}
		}
		while(i <= m)
			rez[k++] = v[i++];
		while(j <= ls)
			rez[k++] = v[j++];
		k = 0;
		for(i=li; i<=ls; i++)
			v[i] = rez[k++];
		for (int i = li; i<=ls; i++)
			b[i] = rez1[k++];
	}
}
int caut(int li, int ls, int *v, int suma)
{
	int m = (li + ls) / 2;
	if (v[m] == suma)
	{
		return m;
	}
	else
	{
		if (li < ls)
		{
			if (v[m] < suma)
				return caut(m+1, ls, v, suma);			
			else
				return caut(li, m-1, v, suma);
		}
		else
			return -1;
	}
}
int compare (const void * a, const void * b)
{
  return ( *(int*)a - *(int*)b );
}
int main()
{
	FILE *f = fopen("loto.in", "r"), *g = fopen("loto.out", "w");
	int n; int s; int v[NMax];
	fscanf(f, "%d %d", &n, &s); 
	for (int i=0; i<n; i++)
	{
		fscanf(f, "%d", &v[i]);
	}
	int *sume = new int[NSum];
	SUM *retin = new SUM[NSum];
	int k = 0;
	for (int i = 0; i < n; i++)
		for (int j = 0; j < n; j++)
			for (int x = 0; x < n; x++)
			{
				int l = v[i] + v[j] + v[x];
				if ( l < s )
				{
					sume[k] = l;
					retin[k].x1 = v[i], retin[k].x2 = v[j], retin[k].x3 = v[x];
					k++;
				}
			}
	//qsort(sume, k-1, sizeof(int), compare);
	merge_sort(0, n-1, sume, retin);
	int gasit = 0;
	for (int i=0; i<n; i++)
		for (int j=0; j<n; j++)
			for (int x = 0; x<n; x++)
			{
				int l = v[i] + v[j] + v[x];
				int result = caut(0, k-1, sume, s - l);
				if ( result >= 0)
				{
					fprintf(g, "%d %d %d %d %d %d", retin[result].x1, retin[result].x2, retin[result].x3, v[i], v[j], v[x]);
					fclose(f); fclose(g);
					delete sume; delete retin;
					return 0; 
				}
			}
	fprintf(g, "-1");
	delete sume; delete retin;
	fclose(f); fclose(g);
	return 0;
}