Cod sursa(job #1416067)

Utilizator alin.emanuelAlin Emanuel alin.emanuel Data 7 aprilie 2015 11:00:48
Problema Loto Scor 20
Compilator c Status done
Runda Arhiva de probleme Marime 2.94 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++) {
                t.v[0] = v[i];
                t.v[1] = v[j];
                t.v[2] = v[k];
                t.sum = v[i] + v[j] + v[k];
                threeSumsVect[m++] = t;
            }
        }
    }
}

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 = 0, j = n3 - 1;
    while (i < j) {
        three_sum left = threeSumsVect[i];
        three_sum right = threeSumsVect[j];
        if (left.sum + right.sum == desiredSum) {
            result[0] = left.v[0];
            result[1] = left.v[1];
            result[2] = left.v[2];
            result[3] = right.v[0];
            result[4] = right.v[1];
            result[5] = right.v[2];
            return 1;
        }
        if (left.sum + right.sum < desiredSum) {
            i++;
        }
        else {
            j--;
        }
    }
    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;

}