标签:布尔 复杂度 main sqrt 个数 ast rip class ret
先把公比为1,即前项 中项 末项相同的统计出来。对每一类数C(n,3)即可。
然后我们发现,因为a1*a3=(a2)^2,所以a1和a3进行质因子分解之后,每一个质因子的指数的奇偶性必然相同,否则无法满足乘积为完全平方数。
然后sqrt(100000)以内的素数只有65个,我们对于每一个数,用unsigned long long存一个01串,代表前64个素因子的奇偶性,再单独用一个布尔存第65个。
然后该数还有可能有一个大素因子(>sqrt(x)),单独存一下,这样用一个三元组唯一标示每一个数。
a1和a3的三元组必然完全相同。
于是对于这样一个三元组,开个map套vector记录下符合其的数的个数(不超过log个),然后就很容易统计了。
复杂度其实很低,就是常数挺大。
#include<cstdio> #include<bitset> #include<map> #include<iostream> #include<cmath> #include<vector> #include<algorithm> using namespace std; typedef long long ll; typedef unsigned long long ull; struct Point{ ull S; int x; int y; Point (const ull &A,const int &X,const int &Y){ S=A; x=X; y=Y; } Point(){ } }; bool operator < (const Point &a,const Point &b){ if(a.S!=b.S){ return a.S<b.S; } if(a.x!=b.x){ return a.x<b.x; } return a.y<b.y; } map<Point,vector<int> >ma; ll ans; bool vis[100000]; int pri[1000],pr; void shai(){ vis[1]=1; for(int i=1;i*i<=100000;++i){ if(!vis[i]){ pri[pr++]=i; for(int j=i*i;j*j<=100000;j+=i){ vis[j]=1; } } } } int n,m,a[1000010]; ll num[100010]; int main(){ // freopen("e.in","r",stdin); shai(); scanf("%d",&n); for(int i=1;i<=n;++i){ scanf("%d",&a[i]); ++num[a[i]]; } for(int i=1;i<=100000;++i){ ans+=(num[i]*(num[i]-1ll)*(num[i]-2ll))/6ll; } sort(a+1,a+n+1); int last=1; for(int i=1;i<=n;++i){ if(a[i]!=a[i-1]){ int x=a[i]; ull S=0; for(int j=0;j<pr-1;++j){ int cnt=0; while(x%pri[j]==0){ ++cnt; x/=pri[j]; } if(cnt%2!=0){ S|=((ull)1<<j); } } int y=0,z; while(x%pri[pr-1]==0){ ++y; x/=pri[pr-1]; } y%=2; z=x; vector<int> v=ma[Point(S,y,z)]; for(int j=0;j<v.size();++j){ ans+=num[a[i]]*num[v[j]]*num[(int)(sqrt((ll)a[i]*(ll)v[j])+0.5)]; } ma[Point(S,y,z)].push_back(a[i]); last=i; } } cout<<ans<<endl; return 0; }
【map】【分解质因数】CDOJ1572 Espec1al Triple
标签:布尔 复杂度 main sqrt 个数 ast rip class ret
原文地址:http://www.cnblogs.com/autsky-jadek/p/6718367.html