Ehab's REAL Number Theory Problem
前置知识
质数
分解质因数
无向无权图最小环<讲>
给 (n) 个数 (a_i)((a_i) 的因数个数不超过 (7)),求最少选出多少个数,使得乘积为完全平方。无解输出 (-1)。
数据范围:(1le nle 10^5),(1le a_ile 10^6)。
没想到一场普通的 ( exttt{CF}) 比赛能出出这么毒瘤好的题!
很明显,把每个 (a_i) 的平方因子除尽后对答案没有影响,所以可以把每个 (a_i) 的平方因子除尽。
如果某个 (a_i) 的质因子除尽后为 (1),直接选它便解决了问题。
然后剩下的质因子的幂次肯定为 (1)。
而且最多只有两个质因子,因为如果 (a_i) 有三个质因子,按照约数个数定理,(d(a_i)=(1+1)^3=8>7),矛盾。
最后问题简化为,选最少的数,使乘积包含的质因子幂次都为 2
(可以自己想为什么不需要选幂次为 (4))。
过程:
[18=2 imes 3^2 o 2 ]
有一个极其巧妙的方法是建一个图,节点是质数,然后把每个数转换为它的两个质因子之间的一条边,求最小环。
如果某个数 (a_i) 质因数个数为 (1)(不存在为 (0) 的,因为已经满足方案),把 (1) 也看做质数节点,连 (1) 和 (a_i)。
根据环的性质,每个点的度为 (2),所以边对应的数的乘积每个质因子幂次都为 (2)。
边没有长度,是无向边,所以问题又简化为了求无向无权图最小环
。
过程:
(a_i):2 3 6 15
。
最小环为 ((1,2,3))。
无向无权图最小环使不得 ( exttt{Floyd})!这里的点数最大约是 (78500),(Theta(n^3)) 能跑到射手座去了。
可以枚举起点,然后 ( exttt{Bfs}),因为问题特殊,所以可以有很大优化。
因为 (1le a_ile 10^6),所以每个 (a_i) 对应的边不可能连接两个 (>1000) 的质数。
所以如果有环,那么环必然有一个起点对应的质数 (in[1,1000])。
所以可以枚举这个起点 (s),然后 ( exttt{Bfs})。
设 (dep_x) 表示节点 (x) 的深度,所以 (dep_s=0)。每次 (Bfs) 前清空。
然后沿着队列顶的点 (x) 连的边走如果走到一个 (dep) 未赋值的节点 (to),就令 (dep_{to}=dep_x+1)。
如果走到一个已经遍历过的点,那么说明这里有一个环,令 (ans=min{ans,dep_{to}+dep_x+1})。
( exttt{Bfs}) 过程中可以走重复的点,不能走重复的边。
这里有一个问题:如何知道这个环是否以 (s) 为其中一个起点呢?
答案是不需要知道,无论 (s) 在不在环上都直接 (ans=min{ans,dep_{to}+dep_x+1})。
因为如果 (s) 不在环上,(dep_{to}+dep_x+1) 肯定比 (s) 在环上大(别忘了每个 (s) 都要枚举过去的啊!)。
时间复杂度 (Theta(n sqrt n))。
过程:
只展示 (s=1) 的 ( exttt{Bfs}) 过程:
代码实现的时候,可以把质数离散化一下。
( exttt{code})
#include <bits/stdc++.h>
using namespace std;
//&Start
#define inf 0x3f3f3f3f
#define re register
#define il inline
typedef long long lng;
typedef vector<int> veci;
//&Data
#define N 100000
#define MX 1000000
#define P 78500--->1000000内质数数量
int n,a[N+10];
//&Prime--->筛质数
bitset<MX+10> np;
int p[P+10],ip[MX+10],pcnt,S;
il void Prime(){
np[1]=true,ip[1]=p[++pcnt]=S=1;
for(re int i=2;i<=MX;i++){
if(!np[i]) p[++pcnt]=i,ip[i]=pcnt,S+=(i<=999);
for(re int j=1;j<=pcnt&&i*p[j]<=MX;j++)
np[i*p[j]]=1;
}
}
//&Graph
veci e[P+10];
int E=1,to[(N<<1)+10];//---->同网络流思想,使互为反边的两条边通过^1可得
il void add(re int x,re int y){ //加双向边
e[x].push_back(++E),to[E]=y;
e[y].push_back(++E),to[E]=x;
}
il void Add(re int x){ // 把数转换为边
re int dcnt=0,div[4];
for(re int j=2;j<=pcnt&&p[j]*p[j]<=x;j++)
if(x%p[j]==0){
while(x%(p[j]*p[j])==0) x/=(p[j]*p[j]);
if(x%p[j]==0) div[++dcnt]=j,x/=p[j];
}
if(x>1) div[++dcnt]=ip[x],x=1;
if(dcnt==0) puts("1"),exit(0);
else if(dcnt==1) add(1,div[1]);
else add(div[1],div[2]);
}
int sz=inf,q[P+10][2],dep[P+10];
il void Bfs(re int s){ //以s为起点Bfs
fill(dep+1,dep+P+1,inf);
re int qcnt=0;
q[++qcnt][1]=s,dep[s]=0;
for(re int ft=1;ft<=qcnt;ft++){
re int x=q[ft][1],f=q[ft][0];
for(re int i:e[x])if(i!=(f^1)){
if(dep[to[i]]==inf){
dep[to[i]]=dep[x]+1;
q[++qcnt][1]=to[i];
q[qcnt][0]=i;
} else sz=min(sz,dep[x]+dep[to[i]]+1); //找到环
}
}
}
//&Main
int main(){
Prime();
scanf("%d",&n);
for(re int i=1;i<=n;i++)
scanf("%d",a+i),Add(a[i]);
for(re int i=1;i<=S;i++) Bfs(i); //枚举起点
if(sz==inf) puts("-1");
else printf("%d
",sz);
return 0;
}
祝大家学习愉快!