Codeforces Round #628 (Div. 2) E. Ehab's REAL Number Theory Problem

思路:因爲不超過7個因數,所有至多有2個不同的質因子。
我們先把一些特殊的情況處理掉:
1.輸入包含完全平方數
2.存在兩個數相乘是完全平方數
那麼剩下的情況 答案必然是>=3
預處理出所有數的:冪次爲奇數的質因子,如果僅有一個的話,就給這個質因子打個標記,否則將兩個質因子連一個無向邊。
那麼如果存在解的話,必然分成兩種情況:一條路徑且兩端的點被打了標記;一個環。
由於輸入數的大小在1e6級別,所以求得答案的路徑上必然存在一個數<=1000。
所以枚舉環上的點跑bfs,判一判存不存在上述兩種情況。然後取一個最優解。

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e6 + 10;
#define fi first
#define se second
#define pb push_back
int n,a[N],p[N],b[N],cnt;
void P(){
  for(int i=2;i<N;i++){
    if(!p[i])b[++cnt]=i;
    for(int j=1;j<=cnt&&1ll*b[j]*i<N;j++){
      p[b[j]*i]=1;
      if(i%b[j]==0)break;
    }
  }
}
int co[N],vis[N];
vector<int>G[N];
int pri[N],so,dis[N],res[N];
int Q[N],sa[N];
int main() {
  ios::sync_with_stdio(false);
  cin>>n;P();
  for(int i=1;i<=n;i++){
    cin>>a[i];sa[a[i]]++;
    if(a[i]==1)return cout<<1,0;
    vector<pair<int,int>>v;
    int y=a[i];
    for(int j=1;1ll*b[j]*b[j]<=y&&j<=cnt;j++){
      if(y%b[j]==0){
        int num=0;
        while(y%b[j]==0){
          y/=b[j];
          num++;
        }
        if(num&1)v.pb({b[j],num});
      }
    }
    if(y>1)v.pb({y,1});
    if(!v.size())return cout<<1,0;//自己就是完全平方數
    if(v.size()==1){
      vis[v[0].fi]=1;
      pri[++so]=v[0].fi;
      co[v[0].fi]++;
    }else if(v.size()==2){
      G[v[0].fi].pb(v[1].fi);
      G[v[1].fi].pb(v[0].fi);
      pri[++so]=v[0].fi;
      pri[++so]=v[1].fi;
      co[v[0].fi*v[1].fi]++;
    }
  }
  for(int i=1;i<=1000000;i++)if(co[i]>=2 )return cout<<2,0;//存在兩個數相乘爲完全平方數
  int ans=n+1;
  for(int i=1;i<=1000;i++)if(G[i].size()){
      for(int j=1;j<=so;j++)res[pri[j]]=-1,dis[pri[j]]=n+1;
      dis[i]=0;
      res[i]=0;
      int idx=0;
      Q[++idx]=i;
      int le=n+1,ri=n+1;
      for(int j=1;j<=idx;j++){
        if(vis[Q[j]] && Q[j]!=i){//找到最近的打標記的點
          if(dis[Q[j]]<le){
            swap(le,ri);
            le=dis[Q[j]];
          }else if(dis[Q[j]]<ri){
            ri=dis[Q[j]];
          }
        }
        for(auto k:G[Q[j]]){
          if(res[k]!=-1 ){
            if(k!=Q[j] && dis[k]>=dis[Q[j]])ans=min(ans,dis[k]+dis[Q[j]]+1);//找到最小環
            continue;
          }
          dis[k]=dis[Q[j]]+1;
          res[k]=0;
          Q[++idx]=k;
        }
      }
      if(vis[i]){
        ans=min(ans,le+2);
      }else{
        ans=min(ans,le+ri+2);
      }
    }
  if(ans==n+1)cout<<-1<<'\n';
  else {
    cout<<ans<<'\n';
    assert(ans!=2);
  }
  return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章