竞赛讨论区 > 牛客21 C题,被卡一下午没过,求指点
头像
bonjour201801211612594
编辑于 2018-08-25 16:34
+ 关注

牛客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; 
}

全部评论

(1) 回帖
加载中...
话题 回帖

等你来战

查看全部

热门推荐