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;
}
全部评论

相关推荐

02-12 20:22
重庆大学 Java
字节暑期刚入职四天,因为是年前,所以很多正职都放假走了,也就没有给我分配mt,然后有一个老哥在我来的时候给我发了一个landing手册,然后还有关于部门业务的白皮书,还有一些业务代码。然后本人是java面的,进来第一次接触go语言&nbsp;前面几天熟悉了一下go的语法和go的框架,可以读但是还不太会写,然后业务白皮书也看的很头疼,包括landing手册里要了解的很多东西说实话我看文档真的快看死了,一个嵌套一个,问题是我还完全不知道咋用这个我了解的东西,还有就是那个项目代码,那个老哥喊我去写写单测,熟悉一下go的语法,但也进行的很困难(这是我第一段实习,之前都是springboot那一套,真不太熟悉这个)想问问大家的建议,就是我从现在开始到在开年回来之前应该做些什么,我目前就一个想法&nbsp;就是复现一个landing手册上的go框架小项目&nbsp;就是相当于帮自己锻炼锻炼怎么写go&nbsp;或者各位大佬有没有更好的锻炼go语法的建议还有就是大家都在说vibe&nbsp;coding,那我应该怎么锻炼自己使用ai的能力,感觉我除了给一些需求然后它给我生成代码,好像就没别的用法了,那些什么工作流、拆解、skill啥的都不知道从哪一个地方开始,包括我现在正在实习,不知道精力该怎么分配,去网上想找找关于agent开发的一些学习流程,说实话,众说纷纭,有的是从python开始打基础然后系统学那些rag&nbsp;prompt&nbsp;langchain&nbsp;mcp等等,有的是说直接找一个github上的ai项目然后反复问ai,我确实有点迷茫,恳求各位大佬能留下你们宝贵的建议,我一定认真反复深刻学习有一说一&nbsp;我觉得字节饭挺好吃的!
Jasonnnnnn...:直接把项目代码喂给AI然后让它帮你分析,如果组里已经有一些流程图总结的话最好,没有的话自己画一个 Go的话其实只要把基础语法搞明白就行了,项目里很多都是直接让ai帮你写好然后自己稍微改下,不用学的特别深 ai的话,可以自己写一些md文件来搞点小东西,但除非你打算转算法,否则不用把rag langchain学的特别深,了解下就行了
字节跳动公司福利 1371人发布
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务