牛客练习赛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;
}

上一篇:P5325 【模板】Min_25筛


下一篇:最小生成树算法(Kruscal && Prim)