【第13届景驰-埃森哲杯广东工业大学ACM程序设计大赛】L题 用来作弊的药水

题目

链接：https://www.nowcoder.com/acm/contest/90/L
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 32768K，其他语言65536K
64bit IO Format: %lld
题目描述
在一个风雨交加的夜晚，来自异世界的不愿透露姓名的TMK同学获得了两种超强药水A、B。根据说明书，TMK知道了这两种药水的作用：
(1)药水A能使人的生命值提高，每饮用1个单位能使他生命值变成原来的x倍，即每饮用p个单位能使他的生命值变成原来的x^p(x的p次方)倍。
(2)药水B能使人的能量值提高，每饮用1个单位能使他能量值变成原来的y倍，即每饮用q个单位能使他的能量值变成原来的y^q(y的q次方)倍。
于是TMK迫不及待地喝下了所有的a个单位的药水A和b个单位的药水B，他立马体会到了自己能力的超强变化，然后他看了接下来的说明书：
药水A和药水B能互相抑制对方的负面效果，一旦生命值提升的倍数和能量值提升的倍数不相等，那么在五个小时后将会发生非常严重的后果。
于是TMK同学慌了，他想知道自己提升的生命值和能量值的倍数是否相等，由于他非常慌张，所以他把计算的重任交给你了。
作为埃森哲公司的一员，你觉得这个问题很简单，这得益于埃森哲公司分享知识的文化。

分享知识已成为埃森哲源远流长的文化。
埃森哲公司在帮助客户进行行之有效的知识管理的同时，它的管理层在其内部也进行了成功的知识管理的实践。如今，在埃森哲，分享知识已成为其源远流长的文化。在很大程度上，埃森哲公司的成功得益于其强大的知识管理系统。

输入描述:
第一行一个整数T，代表有T组数据。(1<=T<=5000)
每组数据仅一行，包含四个整数x,a,y,b，意义为题目描述。(1<=x,a,y,b<=10^9)
输出描述:
每组数据输出一行”Yes”或”No”(不包含双引号)，表示TMK提升的生命值和能量值的倍数是否相等，相等为”Yes”，不相等为”No”。
示例1
输入
4
2 20 4 10
20 20 20 20
20 21 21 20
32768 32768 1048576 24576
输出
Yes
Yes
No
Yes

分析

根据素数的唯一分解定律，可以将x^p进行素数分解，将y^q也进行素数分解，如果分解得到的答案不一致，则输出No；反之，则输出Yes。
另一种解法：
如果直接对x^p进行快速幂运算，结果64位也存不下。我们可以随机给一个M，进行快速幂取模，对于足够多的M都满足x^p(mod M)等于y^q（mod M）的话，那么我们就可以认为x^p与y^q确实相等。

代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#include <map>

using namespace std;
const int maxn=1e6+100;
typedef long long ll;
ll a[maxn],b[maxn],tot,ans[maxn],cnt;
bool valid[maxn];

void getPrime()
{
    tot=0;
    for(ll i=2;i<=maxn;i++) valid[i]=true;
    for(ll i=2;i<=maxn;i++) if(valid[i])
    {
        if(maxn/i<i) break;
        for(ll j=i*i;j<=maxn;j+=i) valid[j]=false;
    }
    for(ll i=2;i<=maxn;i++) if(valid[i])
        ans[++tot]=i;
}

void divid(ll x,ll p)
{
    cnt=0;
    ll now=x;
    memset(b,0,sizeof(b));
    for(ll i=1;i<=tot;i++)
    {
        if(now/ans[i]<ans[i]) break;
        if(x%ans[i]==0)
        {
            a[++cnt]=ans[i];
            while(x%ans[i]==0)
            {
                x/=ans[i];
                b[cnt]++;
            }
        }
        if(x==1) break;
    }
    if(x!=1)
    {
        a[++cnt]=x;
        b[cnt]=1;
    }
    for(ll i=1;i<=cnt;i++)
        b[i]*=p;
}

bool ok(ll y,ll q)
{
    ll sum=0,k,now=y;
    for(ll i=1;i<=tot;i++)
    {
        if(now/ans[i]<ans[i]) break;
        if(y%ans[i]==0)
        {
            ++sum;
            if(ans[i]!=a[sum]) return false;
            k=0;
            while(y%ans[i]==0)
            {
                y/=ans[i];
                ++k;
            }
            if(k*q!=b[sum]) return false;
        }
        if(y==1) break;
    }
    if(y!=1)
    {
        ++sum;
        if(y!=a[sum]) return false;
        k=1;
        if(k*q!=b[sum]) return false;
    }
    return true;
}

int main()
{
    ll T;
    scanf("%lld",&T);
    getPrime();
    while(T--)
    {
        ll x,p,y,q;
        scanf("%lld%lld%lld%lld",&x,&p,&y,&q);
        divid(x,p);
        if(ok(y,q))
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}