Compare the Triplets

Compare the Triplets from Hackerrank.com
https://www.hackerrank.com/challenges/compare-the-triplets/problem[restrict …]

#include<stdio.h>
#include <stdlib.h>
int* solve(int a_count, int* a, int b_count, int* b, int* result_count);
int main()
{
   int c,i;
   int *p = NULL;
   int a[] = {5,6,7},size_a;
   int b[] = {3,6,10},size_b;
   p = solve(3,a,3,b,&c);
   for(i=0;i<c;i++)
       printf("%d ",p[i]);
   printf("\n");
   return 0;
}
int* solve(int a_count, int* a, int b_count, int* b, int* result_count) {
   int ap,bp;
   ap = bp = 0;
   int i;
   int *arr = NULL;
   arr = (int*)malloc(sizeof(int)*2);
    for(i=0;i<a_count;i++)
    {
        if(a[i] > b[i])
        {
            ap++;
        }
        else if(a[i] < b[i])
        {
            bp++;
        }
    }
    arr[0] = ap;
    arr[1] = bp;
    *result_count = 2;
    return arr;
}

[/restrict]