CS 300 Num Cube Sets 质因数分解

CS 300
题意:给出n个元素的集合A,m个元素的集合B.现在从A,B中分别选出其子集a,b 要求:任意一个x属于a,y属于b都有 x*y是完全立方数.
问|a|^2+|b|^2尽量大时(|a|:子集a的元素个数),a,b两个子集分别是多少? n+m<=5e5.a[i],b[i]<=1e6.


底相同两个数相乘(素数乘积形式下),底数的幂都为3k,才可能为立方数.


方便处理把所有素数的幂余3.
假如A中选的某个元素 x= a[1]^p[1] *a[2]^p[2]*..a[i]^p[i]
则B选的某个元素y是固定,只能和x互补 = b[1]^q[1] *b[2]^q[2]*..b[i]^q[i]. 其中b[i]=a[i],q[i]=3-p[i]


用val[i],match[i]表示第i个数其素数幂%3后的值,和它互补的值.
枚举在B中选中的是那个元素foo,然后算a中有多少个=match[val[foo]]即可.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+5;
int p[N];//p[i]:i's min prime factor.
ll val[N],match[N];//val[i] : the val of i after all prime factor's pow%3.
void init()
{
    for(int i=1;i<N;i++)
        p[i]=i;
    for(int i=2;i<N;i++)
        if(p[i]==i)
            for(int j=i+i;j<N;j+=i)
                if(p[j]==j)
                    p[j]=i;
    for(int i=1;i<N;i++)
    {
        int tmp=i;
        val[i]=match[i]=1;
        while(tmp>1)
        {
            int d=p[tmp];
            int cnt=0;
            while(tmp%d==0)
                tmp/=d,cnt++;
            cnt%=3;
            for(int j=0;j<cnt;j++)
                val[i]*=d;
            for(int j=0;j<(3-cnt)%3;j++)
                match[i]*=d;
        }
    }
}
ll sqr(ll x)
{
    return x*x;
}
int main()
{
    init();
    int T,cas=0,n,m;
    cin>>T;
    while(T--)
    {
        scanf("%d%d",&n,&m);
        map<int,int> np;
        for(int i=0;i<n;i++)
        {
            int foo;
            scanf("%d",&foo);
            np[val[foo]]++;
        }
        map<int,int> mp;
        ll ans=-1;
        for(int i=0;i<m;i++)
        {
            int foo;
            scanf("%d",&foo);
            mp[val[foo]]++;
            if(np.find(match[foo])!=np.end())
                ans=max(ans,sqr(np[match[foo]])+sqr(mp[val[foo]]));
        }
        cout<<ans<<endl;
    }
    return 0;
}