[51nod1479]小Y的数论题

Description

给出a,b,c，在区间(0,m)中间选择三个整数x,y,z，使得

$${x^a+y^b}\equiv z^c\pmod m$$
a,b,c<=10^9
数据组数<=10^5

Solution

一些神奇的算法。
既然我们只需要构造出一组解，那不妨往最特殊的方面去想。
众所周知$2^a+2^a=2^{a+1}$
然后，$2^{abk}+2^{abk}=2^{abk+1}$
那么我们设$x=2^{bk},y=2^{ak}$那么左边就能满足了。
右边呢？
我们设$z=2^l$，则$cl=abk+1$
然后$cl-abk=1$
c和ab都已知，直接exgcd解出来。
注意，l,k都必须是非负整数。
还有一种特殊情况，就是m是$2^t$这是我们求出来的x,y,z都可能为0.
那么就特判呗。
如果a>1,就是m/2,1,1.
如果b>1,就是1,m/2,1
如果c>1,就是m/2,m/2,m/2
如果全部=1，那么就1,1,2.
能不能不要那么水！！！

Code

#include<cmath>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
ll a,b,c,k,l,x,y,z;
int m,ty;
ll mi(ll x,ll y) {
    ll z=1;
    for(;y;y/=2,x=x*x%m) if (y&1) z=z*x%m;
    return z;
}
void exgcd(ll a,ll b,ll &x,ll &y) {
    if (!b) {x=1;y=0;return;}
    ll xx,yy;exgcd(b,a%b,xx,yy);
    x=yy;y=xx-a/b*yy;
}
int main() {
    for(scanf("%d",&ty);ty;ty--) {
        scanf("%d%lld%lld%lld",&m,&a,&b,&c);
        int t=log2(m);
        if (pow(2,t)==m) {
            x=1;y=1;z=1;
            if (a>1) x=m/2;
            else if (b>1) y=m/2;
            else if (c>1) x=y=z=m/2;
            else x=y=1,z=2;
            printf("%lld %lld %lld\n",x,y,z);
            continue;
        } 
        exgcd(c,a*b,l,k);k=-k;
        while (k<0||l<0) k+=c,l+=a*b;
        x=mi(2,b*k);y=mi(2,a*k);z=mi(2,l);
        printf("%lld %lld %lld\n",x,y,z);
    }
}