結果

問題 No.1514 Squared Matching
ユーザー umezo
提出日時 2021-05-22 00:45:53
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 776 bytes
コンパイル時間 4,520 ms
コンパイル使用メモリ 195,008 KB
最終ジャッジ日時 2025-01-21 17:02:41
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10 TLE * 2 MLE * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define ALL(v) v.begin(), v.end()
typedef long long ll;

#include <bits/stdc++.h>
using namespace std;

int B[50000010];
int C[50000010];

int main(){
  int n;
  cin>>n;
  
  int MAX=50000010;
  vector<int> prime(MAX,1);
  prime[0]=0,prime[1]=0;
  for(int i=2;i*i<=MAX;i++){
    if(!prime[i]) continue;
    for(int j=2*i;j<=MAX;j+=i) prime[j]=0;
  }
  vector<int> A;
  for(int i=1;i*i<=n;i++){
    if(prime[i]) A.push_back(i);
  }
  
  rep(i,n+1) B[i]=i;
  for(auto p:A){
    int q=p*p;
    int r=q;
    while(1){
      for(int i=q;i<=n;i+=q) B[i]/=r;
      if(q>n/r) break;
      q*=r;
    }
  }
  
  for(int i=1;i<=n;i++) C[B[i]]++;
  int ans=0;
  for(int i=1;i<=n;i++) ans+=C[i]*C[i];
  cout<<ans<<endl;
  
  return 0;
}
0