I am Getting TLE in one test case

#1

Please Review my code

#include<bits/stdc++.h>
#define ll long long int
using namespace std;
ll pexp(ll a,ll i,ll p)
{
	if(i==0)
	{
		return 1;
	}
	if(i==1)
	{
		return a;
	}
	if(i%2==1)
	{
		return ((a*(pexp(a,i-1,p)%p))%p);
	}
	else
	{
	ll x=(pexp(a,(i/2),p))%p;
	return (((x)*(x))%p);
}
}
int main()
{
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	ll t;
	cin>>t;
	while(t--)
	{

		ll a,n,p;
		cin>>a>>n>>p;
		a=a%p;
		ll ans=a;
		// cout<<a<<"a"<<endl;
		for(ll i=2;i<=n;i++)
		{
			ll pe=pexp(ans,i,p);
			
			ans=(ans*pe)%p;
			// cout<<ans<<" "<<pe<<endl;
			
		}
		cout<<ans<<endl;
	}
}
#2

@ishanabc8

if(i%2==1)
	{
		return ((a*(pexp(a,i-1,p)%p))%p);                  This statement here is causing TLE, it is making the function near about linear.
	}

if(i%2==1)
	{    
                ll x=(pexp(a,(i/2),p))%p;
		return (a*((x*x)%p))%p;
	}

Correct statement
If this resolves your doubt mark it as resolved.

#3

I hope I’ve cleared your doubt. I ask you to please rate your experience here
Your feedback is very important. It helps us improve our platform and hence provide you
the learning experience you deserve.

On the off chance, you still have some questions or not find the answers satisfactory, you may reopen
the doubt.