牛客練習賽48 B 小w的a=b問題

題意： 給你兩個序列，長度分別是n，m，問這兩個序列中的每個數階乘的累積是否相等。
思路： 典型的質因子分解，在一個序列中，對每個數求階乘的連乘積，可以統計從2開始每個數的乘積次數，然後把每個數質因子分解，轉化成求這個質數乘積的次數即可。最後判兩個序列中質數乘積次數是否相等。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
map<int,int>mp;
bool IsPrime[100010];
int Prim[700000];
int a[110000],b[110000];
int num1[110000],num2[110000];
/**********************************************Head-----Template****************************************/
bool Finish_read;
template<class T>inline void read(T &x){Finish_read=0;x=0;int f=1;char ch=getchar();while(!isdigit(ch)){if(ch=='-')f=-1;if(ch==EOF)return;ch=getchar();}while(isdigit(ch))x=x*10+ch-'0',ch=getchar();x*=f;Finish_read=1;}
template<class T>inline void print(T x){if(x/10!=0)print(x/10);putchar(x%10+'0');}
template<class T>inline void writeln(T x){if(x<0)putchar('-');x=abs(x);print(x);putchar('\n');}
template<class T>inline void write(T x){if(x<0)putchar('-');x=abs(x);print(x);}
ll gcd(ll a,ll b){return b==0?a:gcd(b,a%b);}
ll lcm(ll a,ll b){ll gg=gcd(a,b);a/=gg;if(a<=LLONG_MAX/b) return a*b;return LLONG_MAX;}
/********************************Head----Temlate**********************************************/

void init()
{
    for(int i=0;i<100000;i++)   a[i]=b[i]=num2[i]=num1[i]=0;
}

int  euler_prime(int n)
{
    int num = 0, j;
    for(int i = 2; i <= n; i ++)
    {
        if(!IsPrime[i]){
            Prim[num ++] = i;
        }
        for(j  = 0; j < num; j ++)
        {
            if(i * Prim[j] > n)
                break;
            IsPrime[i * Prim[j]] = true;
            if(i % Prim[j] == 0)
                break;
        }
    }
    return num;
}

int main()
{
    int t,top,n;
    ll sum=0;
    top=euler_prime(10000); //素數篩
    int T;
    read(T);
    while(T--){
        init();
        int n,m;
        scanf("%d%d",&n,&m);
        int maxa=-1,maxb=-1;
        for(int i=0;i<n;i++){
            int x;
            read(x);
            a[x]++;
            maxa=max(x,maxa);
        }
        for(int i=0;i<m;i++){
            int x;
            read(x);
            b[x]++;
            maxb=max(x,maxb);
        }
        for(int i=max(maxa,maxb);i>=1;i--){ //每個數要乘積次數
            a[i-1]+=a[i];
            b[i-1]+=b[i];
        }
        for(int i=2;i<=max(maxa,maxb);i++){//把每個數質因子分解
            int t=i;
            for(int j = 0; Prim[j]*Prim[j] <= t ; ++j)  //我之前把每素數判斷一遍T了，後來改成這樣子
            {
                while(t > 1 && t % Prim[j] == 0)
                {
                    t /= Prim[j];
                    num1[Prim[j]]+=a[i];
                    num2[Prim[j]]+=b[i];
                }
            }
            if(t > 1) num1[t]+=a[i],num2[t]+=b[i];
        }
        int flag=0;
        for(int i=2;i<=100000;i++){
            if(num1[i]!=num2[i]){
                flag=1;
                break;
            }
        }
        if(flag)    printf("unequal\n");
        else printf("equal\n");
    }
    return 0;
}