Time complexity : primality

Question
Code :
#include<iostream>
using namespace std;
int prime(int b);
int main(){
    int t;
    cin>>t;
    while(t--){
        int a;
        cin>>a;
        if(prime(a)){
            cout<<"Prime"<<endl;
        }
        else
            cout<<"Not prime"<<endl;
    }
    return 0;
}
int prime(int a){
    int i;
    if(a==1){
        return 0;
    }
    if(a==2){
        return 1;
    }
    else{
        for(i=2;i*i<=a;i++){
            if(a%i==0){
                return 0;
            }
        }
        return 1;
    }
    return 0;
}