牛客21 C题,被卡一下午没过,求指点
实在是找不出两段代码的差别在哪儿,总是超时不过。下面是我写的代码:
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std ;
typedef long long ll ;
int const mod = 1000000007 ;
int T;
int n ;
ll ins[10005];
int s[10000];
vector<int > p;
priority_queue<int>que;
ll pow_(ll x , ll m){
ll ans =1 ;
while(m>0){
if(m&1){
ans*=x;
ans%=mod ;
}
m>>=1;
x*=x;
}
return ans ;
}
int main(){
for(int i =2 ;i<10000;i++ )
if(s[i]==0){
p.push_back(i);
for(int j = i* 2; j<10000;j+=i ) s[j]=1 ;
}
scanf("%d",&T);
while(T--){
scanf("%d",&n);
ll ans =1 ;
for(int i =0 ; i<n ; i++)scanf("%lld",&ins[i]);
for(int ind = 0 ; ind < p.size();ind++){
for(int j = 0 ; j< n ; j++) {
if(ins[j]==1||p[ind]>ins[j])continue;
int cnt = 0 ;
while(ins[j]%p[ind]==0){
ins[j]/=p[ind];
cnt++;
}
if(cnt>0)que.push(cnt);
}
while(que.size()>1){
int a= que.top();que.pop();
int b= que.top();que.pop();
--a;--b;
if(a>0)que.push(a);
if(b>0)que.push(b);
}
if(que.empty())que.push(0);
int tt= que.top();que.pop();
ans=(ans*pow_(p[ind],tt))%mod;
}
sort(ins,ins+n);
for(int i = 0 ; i< n ; i++)
if(ins[i]==ins[i+1])ins[i]=ins[i+1]=1;
for(int i = 0 ; i< n ;i++)ans=(ans* ins[i])%mod ;
printf("%lld\n",ans);
}
return 0;
}