[SDOI2011]计算器

题目

展开
题目描述
你被要求设计一个计算器完成以下三项任务：

给定 y,z,py,z,p，计算 y^z \bmod py
z
modp 的值；
给定 y,z,py,z,p，计算满足 xy \equiv z \pmod pxy≡z(modp) 的最小非负整数 xx；
给定 y,z,py,z,p，计算满足 y^x \equiv z \pmod py
x
≡z(modp) 的最小非负整数 xx。
为了拿到奖品，全力以赴吧！

输入格式
输入文件包含多组数据。

第一行包含两个正整数 T,KT,K，分别表示数据组数和询问类型（对于一个测试点内的所有数据，询问类型相同）。

以下 TT 行每行包含三个正整数 y,z,py,z,p，描述一个询问。

输出格式
输出文件包括 TT 行。

对于每个询问，输出一行答案。

对于询问类型 2 和 3，如果不存在满足条件的，则输出 Orz, I cannot find x!。

输入输出样例
输入 #1复制
3 1
2 1 3
2 2 3
2 3 3
输出 #1复制
2
1
2
输入 #2复制
3 2
2 1 3
2 2 3
2 3 3
输出 #2复制
2
1
0
输入 #3复制
4 3
2 1 3
2 2 3
2 3 3
2 4 3
输出 #3复制
0
1
Orz, I cannot find x!
0
说明/提示
测试点共分为三类，各类测试点占总测试点的比例如下：

K=K= 测试点占比
11 20%20%
22 35%35%
33 45%45%
所有数据均满足：1 \leq y,z,p \leq 10^91≤y,z,p≤10
9
，pp 是质数，1 \leq T \leq 101≤T≤10。

思路

对于询问1，快速幂解决

对于询问2，扩展欧几里得解决

对于询问3，采用BSGS算法

如果p不是素数，可以用扩展BSGS

求yx=z(modp)yx=z(modp)

设x=km+ix=km+i

ykm∗yi≡zykm∗yi≡z

yi≡z∗ine(ykm)yi≡z∗ine(ykm)(逆元)

用费马小定理显然可得ine(ym)≡yp−1−mine(ym)≡yp−1−m

设其为ine(ykm)≡ine(y(k−1)m)∗Tine(ykm)≡ine(y(k−1)m)∗T

把yi(0<=i<=m)yi(0<=i<=m) 放入hash或者map

然后枚举k，查询z∗ine(ykm)z∗ine(ykm)

代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
int t,l,y,z,p,x;
map<int,int>ma;
int power(int a,int b){
    if(b==0)return 1;
    int x=power(a,b>>1);
    if(b&1)return (x*x%p)*a%p;
    else return x*x%p;
}
void bsgs(){
    if(y==0&&z==0){printf("1\n");return;}
    if(y==0&&z!=0){printf("Orz, I cannot find x!\n");return;}
    int m=sqrt(p);
    ma.clear();
    int t=z%p;
    ma[t]=0;
    for(int j=1;j<=m;j++){
        t=t*y%p;
        ma[t]=j;
    }
    t=1;
    int mi=power(y,m);
    for(int i=1;i*i<=p+1;i++){
        t=t*mi%p;
        if(ma.count(t)){
            x=i*m-ma[t];
            x=(x%p+p)%p;
            printf("%d\n",x);
            return ;
        }
    }
    printf("Orz, I cannot find x!\n");
}
signed main()
{
    scanf("%d%d",&t,&l);
    if(l==1){
        for(int i=1;i<=t;i++){
        	scanf("%lld%lld%lld",&y,&z,&p);
            y%=p;
            x=power(y,z);
            printf("%d\n",x);
        }
    }
    else if(l==2){
        for(int i=1;i<=t;i++){
        	scanf("%lld%lld%lld",&y,&z,&p);
            y%=p;
            z%=p;
            if(y==0&&z!=0){
            	printf("Orz, I cannot find x!\n");
            	continue;
			}
            x=power(y,p-2)*z%p;
            printf("%d\n",x);
        }
    }
    else if(l==3){
        for(int i=1;i<=t;i++){
            scanf("%d%d%d",&y,&z,&p);y%=p;
            bsgs();
        }
    }
    return 0;
}