結果

問題 No.1514 Squared Matching
コンテスト
ユーザー sgsw
提出日時 2021-05-23 01:13:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3,797 ms / 4,000 ms
コード長 1,498 bytes
コンパイル時間 2,120 ms
コンパイル使用メモリ 196,556 KB
最終ジャッジ日時 2025-01-21 17:21:10
ジャッジサーバーID
（参考情報） 		judge2 / judge2
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>

using namespace std;

#define rep(i,n) for (int i = 0; i < n; i++)
#define rep1(i,n) for(int i = 1;i<=(n);i++)
#define rrep(i,n) for(int i = (n)-1;i>=0;i--)
#define rrep1(i, n) for (int i = (n); i > 0; i--)
#define REP(i, a, b) for (int i = a; i < b; i++)
#define ALL(a) a.begin(),a.end()
#define elif else if
#define foreach(elem,a) for (auto elem : a)
#define list vector

template<class T> ostream& operator << (ostream &s, vector<T> &P)
{ for (int i = 0; i < P.size(); ++i) { if (i > 0) { s << " "; } s << P[i]; } return s; }

template <class T>bool chmax(T &a, T b){if (a < b){a = b;return true;}return false;}
template <class T>bool chmin(T &a, T b){if (a > b){a = b;return true;}return false;}
template <class T = int>T gcd(T a, T b){return (b == 0) ? a : gcd(b, a % b);}
template <class T = int>T lcm(T a, T b){return a / gcd(a, b) * b;}
using P  = pair<int,int>;

const int MAXN = 50000010;

list<int> A(MAXN);
list<bool> isprime(MAXN,true);
list<int> CNT(MAXN,0);

void setup_init(){
    rep(i,MAXN){A[i]=i;}
    isprime[0] = isprime[1] = false;
    for (int i = 2; i < MAXN;i++){
        if (!isprime[i])continue;
        int sq = i*i;
        for (int j = 2*i; j < MAXN;j+=i){
            isprime[j]=false;
            while(A[j]%sq == 0){A[j]/=sq;}
        }
    }
}

signed main(){
    setup_init();
    int n;cin>>n;
    rep1(i,n){CNT[A[i]]++;}
    
    int ans = 0;
    rep1(i,n){
        ans += CNT[i]*CNT[i];
    }
    cout << ans << endl;
    return 0;
}
0