CERC2017 F: Faulty Factorial 简单数论题

#include  <iostream>
using namespace std;
#define ll long long 
const int N  = 10000006;
ll n,p,r;
ll poww(ll a,ll b){
    ll ans  =1ll;
    while(b){
        if(b&1) ans =ans*a%p;
        b>>=1;
        a=a*a%p;
    }
    return ans%p;
}
int  main()
{
    //每次都要注意-1 -1的情况
    scanf("%lld%lld%lld",&n,&p,&r);
    if(n>=2*p){
        if(r==0){
            printf("%lld 1
",p+1);
        }
         else{
             printf("-1 -1
");
         }
    }
    else if(n>=p){
        if(r==0){
            int flag =0;
            for(ll i=2;i<=n;i++){
                if(i!=p)//起初if的{}写错了
                {                    
                printf("%lld 1
",i);
                flag =1;
                break;
                }
            }
            if(!flag)printf("-1 -1
"); //n=p=2
        }
        else{
            ll ans=1ll;
            for(ll i=2;i<=n;i++) {
                if(i!=p) ans =ans*i%p;
            }
            int  flag =0;
            for(ll i=1;i<p;i++){
                if(ans*i%p==r) {
                    printf("%lld %lld
",p,i);
                    flag =1;
                    break;
                }
            }
            if(flag==0) printf("-1 -1
");
        }
    }
    else{
        ll ans =1ll;
        for(ll i=2;i<=n;i++) ans =ans*i%p;
        int  flag =0;
        ans = poww(ans,p-2);//放到里面会超时
        for(ll i=2;i<=n;i++){
            ll x= r*i%p*ans%p;
            // ans/i*x%p==r
            // x=r*i%p/ans%p;
            if(x>=1&&x<i){
                printf("%lld %lld
",i,x);
                flag =1;
                break;
            }
        }
        if(!flag) printf("-1 -1
");
    }
    return 0;
    
}