D - Square Pair

原題連結

題解

多想幾種暴力

1.遍歷所有數對： \(O(n^2)\)

2.求有多少數對其乘積為平方數 \(\to\) 求有多少平方數能被數對乘積： \(O(n^2)\)

3.如果兩個數的乘積為平方數，代表他們的質因數，要麼都是奇數，要麼都是偶數 : \(O(?)\)

4.如果 \(a \times b\) 是完全平方數，代表 \(a\times b\) 的質因子都是偶數，所以如果 \(a,b\) 的質因子為偶數，可以去掉，對答案沒有影響，因為去掉之後質因子仍然是偶數

這樣下來，去掉了所有偶數質因子的 \(a'\) 和 \(b'\)，如果他們的乘積還是平方數，當且僅當 \(a',b'\) 具有的質因子一樣，由於我們只需要每個數具有的奇數質因子，所以我們將所有的奇數質因子個數變為1，偶數變為0

所以，我們可以對所有數去除質因子，然後統計相同數數對

code

#include<bits/stdc++.h>
using namespace std;
/*
#define double long double
#define lowbit(x) ((x)&(-x))
const int inf=1e18;

const int N=4e5;

int inv(int x)
{
    return qpow(x,mod-2);
}
int fa[2000005];
int finds(int now){return now==fa[now]?now:finds(fa[now]);}

vector<int> G[200005];

int dfn[200005],low[200005];
int cnt=0,num=0;
int in_st[200005]={0};
stack<int> st;
int belong[200005]={0};

void scc(int now,int fa)
{
    dfn[now]=++cnt;
    low[now]=dfn[now];
    in_st[now]=1;
    st.push(now);

    for(auto next:G[now])
    {
        if(next==fa) continue;

        if(!dfn[next])
        {
            scc(next,now);
            low[now]=min(low[now],low[next]);
        }
        else if(in_st[next])
        {
            low[now]=min(low[now],dfn[next]);
        }
    }

    if(low[now]==dfn[now])
    {
        int x;
        num++;
        do
        {
            x=st.top();
            st.pop();
            in_st[x]=0;
            belong[x]=num;
        }while(x!=now);
    }
}

*/

#define int long long
vector<int> prime;
bool mark[200005]={0};
int fac[200005]={0};
void shai()
{
    for(int i=2;i<=200000;i++)
    {
        if(!mark[i])
        {
            fac[i]=i;
            prime.push_back(i);
        }

        for(auto it:prime)
        {
            if(it*i>200000) break;
            mark[it*i]=1;
            fac[it*i]=it;
            if(it%i==0) break;
        }
    }
}
int qpow(int a,int n)
{
    int res=1;
    while(n)
    {
        if(n&1) res=res*a;
        a=a*a;
        n>>=1;
    }
    return res;
}
int a[200005],vis[200005]={0};
void solve()
{
    int n;
    cin>>n;

    int ans=0;
    for(int i=1;i<=n;i++) cin>>a[i];
    sort(a+1,a+1+n);
    for(int i=1;i<=n;i++)
    {
        if(a[i]==0)
        {
            ans+=n-i;
            continue;
        }
        int tem=a[i];

        while(tem!=1)
        {
            int f=fac[tem];
            int cnt=0;
            while(tem!=1&&fac[tem]==f)
            {
                cnt++;
                tem/=fac[tem];
            }
            if(cnt%2==0)
            {
                a[i]/=qpow(f,cnt);
            }
            else a[i]/=qpow(f,cnt-1);
        }
        //printf("i:%d  a:%d\n",i,a[i]);
        ans+=vis[a[i]];
        vis[a[i]]++;
    }

    cout<<ans;
}
signed main()
{
    shai();
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int TT=1;
    //cin>>TT;
    while(TT--) solve();
    return 0;
}