#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>

int is_prime(int n) {
    if (n <= 1)
        return 0;
    for (int d = 2; d * d <= n; d++)
        if (n % d == 0)
            return 0;
    return 1;
}

int primes_count_in_interval(int start, int finish) {
    int ret = 0;
    for (int i = start; i < finish; i++)
        if (is_prime(i) != 0)
            ret++;
    return ret;
}

// The structure that will be passed to the threads, corresponding to an
// interval to count the number of primes in.
typedef struct prime_counter_request {
    int start, finish;
} prime_counter_request;

void *prime_counter(void *arg) {
    /*
      TODO
      Complete this function. It takes a primes_counter_request as a parameter
      and returns the number of primes in the specified interval. Be careful how
      you return the return value.
    */
    prime_counter_request *req = arg;
    int *res = malloc(sizeof(int));
    *res = primes_count_in_interval(req->start, req->finish);
    return res;
}

int main(int argc, char *argv[]) {
    int n = atoi(argv[1]), n_threads = atoi(argv[2]);
    int segment_size = n / n_threads;

    pthread_t *threads = malloc(n_threads * sizeof(pthread_t));

    prime_counter_request *requests =
        malloc(n_threads * sizeof(prime_counter_request));

    void **results = malloc(n_threads * sizeof(void *));

    for (int i = 0; i < n_threads; i++) {
        /*
          TODO
          Complete this loop. You shall spawn <n_threads> threads, each of which
          computes the number of primes in a segment of length <segment_size>.
          The union of those intervals shall correspond to the interval [0, n).
          First, initialize the value of requests[i] with a simple mathematical
          computation. Be careful not to overrun the last interval. Second,
          spawn a thread that takes requests[i] as an argument. Be careful how
          you pass the arguments.
        */
        requests[i].start = i * segment_size;
        requests[i].finish = requests[i].start + segment_size;
        if (requests[i].finish > n)
            requests[i].finish = n;

        pthread_create(&threads[i], NULL, &prime_counter, (void*) &requests[i]);
    }

    for (int i = 0; i < n_threads; i++) {
        /*
          TODO
          Join the results of threads into the results array.
        */
        pthread_join(threads[i], (void**) &results[i]);
    }

    int total_result = 0;
    for (int i = 0; i < n_threads; i++)
        total_result += *(int *)results[i];

    /*
      TODO
      Free the allocated memory.
    */
    for (int i = 0; i < n_threads; i++)
        free(results[i]);

    printf("%d\n", total_result);

    exit(EXIT_SUCCESS);
}