[Acwing] 4319. 合适数对 hash+算术基本定理

前言

传送门 :

思路

$a_i\ast a_j=x^k$

根据算术基本定理 :
$N=p_1^{a1}\ast p_2^{a2}...p_n^{an}$

可以将一个数拆为质因数的次幂的乘积

对于$x^k$
我们将其拆分成
$x=p_1^{a1}\ast p_2^{a2}...p_k^{ak}$

显然我们再把k次幂搞上去 , 对于每个$p$的指数都变成了 k的倍数

(啊 这里因为我对$x^k$进行质因数分解,看到$k$的倍数的时候感觉不合乎常理,总算懂了)

---

因此开始分析

$a_j=p_1^{a1}\ast p_2^{a2}...p_k^{ak}$
$a_i=p_1^{b1}\ast p_2^{b2}...p_k^{bk}$
根据上面推出来的
因此对于每一个$a_i+b_i\%k=0(a这里是指数$

数据范围$100000$,所有数质因子的数量$\frac{n}{l{n}^n}(1000-10000)$

但是对于每个数,出现过和其他数不同的质因子很少
$2\ast 3\ast 5\ast 7\ast 11=2310$
$2\ast 3\ast 5\ast 7\ast 11\ast 13=30030$
因此最多只能包含六个质因子
每个数最多只有$6$个质因子,需要补成$k$的倍数

因此这里可以$hash$一个匹配,但是这里可以使用他们的值直接当作$hash$值
$x=p_1^{t1}\ast p_2^{t2}\ast p_3^{t3}$
$y=q_1^{r1}\ast q_2^{r2}\ast q_3^{r3}$
如果这两个中但凡出现一个不同那么$xy$都不相同

code

const int N = 100010;

int n, m;
int cnt[N];

int power(int a, int b)
{
    LL res = 1;
    while (b -- )
    {
        res *= a;
        if (res >= N) res = 0;
    }
    return res;
}

int main()
{
    scanf("%d%d", &n, &m);

    LL res = 0;
    while (n -- )
    {
        int x;
        scanf("%d", &x);
        LL y = 1, z = 1;
        
        for (int i = 2; i * i <= x; i ++ )
            if (x % i == 0)
            {
                int s = 0;
                while (x % i == 0) x /= i, s ++ ;
                s %= m;
                if (s) 
                {
                    y *= power(i, s);
                    z *= power(i, m - s);
                }
            }
            
        /**
        这一整个过程都是hash
        **/

        if (x > 1) y *= x, z *= power(x, m - 1);
        if (z >= N) z = 0;

        res += cnt[z];
        cnt[y] ++ ;
    }

    printf("%lld\n", res);
    return 0;
}

Stl暴力 4483ms

const int N  = 1e5+10;
int a[N];
map<vector<pii>,int> cnt;

void solve()
{
	int n,m;cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>a[i];
	
	ll ans = 0 ;
	for(int i=1;i<=n;i++){//枚举每一个 a[i]
		int x = a[i];
		vector<pii> b,c;
		
		for(int j=2;j<=x/j;j++){//筛选质因数 O sqrt(n)
			int t = 0 ;
			
			while(x%j == 0) x/=j,++t;
			//j^t
		
			
			if(t%m) b.pb({j,t%m});
		}
		
		if(x>1){//分解质因子
			b.pb({x,1});
		}
		
		c = b;
		for(auto &[x,y] : c) y = m-y; //
		
		ans+= 1ll*cnt[c];
		cnt[b]++;
		
	}
	
	cout<<ans<<endl;
	
	
}