Cod sursa(job #1416081)

Utilizator alin.emanuelAlin Emanuel alin.emanuel Data 7 aprilie 2015 11:55:09
Problema Loto Scor 10
Compilator c Status done
Runda Arhiva de probleme Marime 3.08 kb
#include <stdio.h>
#include <stdlib.h>

typedef struct header_data {
    int n;
    int s;
} header_data;

typedef struct three_sum {
    int v[3];
    int sum;
} three_sum;

header_data readHeader (FILE* in) {
    header_data h;
    fscanf(in, "%d %d", &(h.n), &(h.s));
    return h;
}

void readNumbers (int* v, int n, FILE* in) {
    int i;
    for (i = 0; i < n; i++) {
        fscanf(in, "%d ", &v[i]);
    }
}

void setThreeSums (int* v, int n, three_sum* threeSumsVect) {
    int i, j, k, m = 0;
    three_sum t;
    for (i = 0; i < n; i++) {
        for (j = 0; j < n; j++) {
            for (k = 0; k < n; k++) {
                threeSumsVect[m].v[0] = v[i];
                threeSumsVect[m].v[1] = v[j];
                threeSumsVect[m].v[2] = v[k];
                threeSumsVect[m].sum = v[i] + v[j] + v[k];
                m++;
            }
        }
    }
}

int findDesiredSum (three_sum* threeSumsVect, int first, int last, int desiredSum) {
    int mid = (last - first) / 2;
    if (first == last) {
        return -1;
    }
    if (threeSumsVect[mid].sum == desiredSum) {
        return mid;
    }
    return findDesiredSum(threeSumsVect, first, mid, desiredSum);
    return findDesiredSum(threeSumsVect, mid, last, desiredSum);
}

int setTheSixNumbers (three_sum* threeSumsVect, int n, int desiredSum, int* result) {

    int n3 = n * n * n;
    int i;
    for (i = n3 - 1; i >= 0; i--) {

        if (threeSumsVect[i].sum * 2 < desiredSum || threeSumsVect[i].sum > desiredSum) {
            continue;
        }

        int pos = findDesiredSum(threeSumsVect, 0, n3, desiredSum - threeSumsVect[i].sum);

        if (threeSumsVect[i].sum + threeSumsVect[pos].sum == desiredSum) {
            result[0] = threeSumsVect[i].v[0];
            result[1] = threeSumsVect[i].v[1];
            result[2] = threeSumsVect[i].v[2];
            result[3] = threeSumsVect[pos].v[0];
            result[4] = threeSumsVect[pos].v[1];
            result[5] = threeSumsVect[pos].v[2];
            return 1;
        }
    }
    return 0;
}

int printResult (int* result, int exists, FILE* out) {
    if (exists) {
        int i;
        for (i = 0; i < 6; i++) {
            fprintf(out, "%d ", result[i]);
        }
    }
    else {
        fprintf(out, "%d", -1);
    }
}

int cmpfunc (const void * a, const void * b) {
   return ( (*(three_sum*)a).sum - (*(three_sum*)b).sum );
}

int cmpint (const void *a, const void *b) {
    return ( *(int*)a - *(int*)b );
}

int main () {

    FILE *out = fopen ("loto.out","w");
    FILE *in = fopen ("loto.in","r");

    header_data h = readHeader(in);
    int n = h.n;
    int s = h.s;

    int v[n];
    readNumbers(v, n, in);


    three_sum* threeSumVect = (three_sum*) malloc(n * n * n * sizeof(three_sum));
    setThreeSums(v, n, threeSumVect);

    qsort(threeSumVect, n * n * n, sizeof(three_sum), cmpfunc);

    int result[6];
    int exists = setTheSixNumbers(threeSumVect, n, s, result);

    printResult(result, exists, out);

    free(threeSumVect);

    return 0;

}