#include <stdio.h>
#include <malloc.h>
#include <omp.h>

void main(int argc, char **argv )
 {
  int i,j,threads;
  int x[10];
  int *sum_of_powers;
  int n=10;

  threads = atoi(argv[1]);
  //Set the number of threads 
  if (!omp_get_dynamic()) omp_set_num_threads(threads);
  else
   {
    printf("ERROR: set OMP_DYNAMIC=FALSE\n");
    exit(0);
   }
  printf("%d threads\n",omp_get_max_threads());

  sum_of_powers = (int *) malloc(n * sizeof(int)); 
  #pragma omp parallel for private(j) lastprivate(x) 
  for (i=0; i<n; i++)
   {
    x[0] = 1;
    for (j=1; j<4; j++) x[j] = x[j-1] * (i+1);
    sum_of_powers[i] = x[0] + x[1] + x[2] + x[3];
   }
  for (i=0; i<n; i++)
   printf("Sum of powers for %d = %d\n",i,sum_of_powers[i]);
  printf("n cubed = %d\n",x[3]);
 }