【PTA】查找数据(指针型函数代码训练)

编写函数
int *find(int a[],int n,int x)
功能为在有n个元素的数组a中查找首个整数x，若找到则返回该元素的地址(指针)，否则返回空指针。
主函数中输入N，再输出N个整数存入数组，再输入整数X，最后输出在数组中查找X的结果。

输入格式:

第一行是整数N。

第二行是否N个整数。

第三行是X。

输出格式:

找到了输出下标（从0开始），找不到则输出NOT FOUND.

输入样例: 

10
1 2 3 4 5 6 7 8 9 10
1
 

输出样例: 

0

#include <stdio.h>
#include <stdlib.h>
int *find(int a[],int n,int x)
{
    int i;
    for(i=0;i<n;i++)
    {
        if(a[i]==x)
        {
            return a+i;
        }

    }return NULL;
}
int main()
{
    int i;
    int N,x,*p,*q;
    scanf("%d",&N);
     p=(int*)malloc(N*sizeof(int));
    for(i=0;i<N;i++)
    {
        scanf("%d",&p[i]);
    }
    scanf("%d",&x);
    q=find(p,N,x);
    if(q!=NULL)
    {
        printf("%d",q-p);
    }else
    {
        printf("NOT FOUND");
    }
    return 0;
}