牛客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; 
}
全部评论
这段代码时答案给的参考代码: #include<bits/stdc++.h> using namespace std; const int mod=1000000007; int f[10005]; vector<int> p; int a[10005]; priority_queue<int> que; int read() { char c=getchar(); int x=0; while (c<'0' || c>'9') c=getchar(); while (c>='0' && c<='9') { x=x*10+c-'0'; c=getchar(); } return x; } int main() { memset(f,0,sizeof(f)); for (int i=2; i<=10000; i++) { if (f[i]==0) { p.push_back(i); for (int j=i*2; j<=10000; j+=i) f[j]=1; } } int T=read(); while (T--) { int n=read(); for (int i=1; i<=n; i++) a[i]=read(); long long ans=1; for (int i=0; i<p.size(); i++) { for (int j=1; j<=n; j++) { int t=0; while (a[j]%p[i]==0) a[j]/=p[i],++t; if (t>0) que.push(t); } 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 cnt=que.top(); que.pop(); while (cnt--) ans=(ans*p[i])%mod; } sort(a+1,a+n+1); for (int i=1; i<=n-1; i++) { if (a[i]==a[i+1]) { a[i]=a[i+1]=1; } } for (int i=1; i<=n; i++) ans=(ans*a[i])%mod; printf("%lld\n", ans); } return 0; }
点赞 回复
分享
发布于 2018-08-25 16:32

相关推荐

点赞 评论 收藏
转发
点赞 收藏 评论
分享
牛客网
牛客企业服务