結果

問題 No.1746 Sqrt Integer Segments
ユーザー ぷら
提出日時 2021-11-18 00:48:11
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 230 ms / 2,000 ms
コード長 1,497 bytes
コンパイル時間 2,336 ms
コンパイル使用メモリ 209,460 KB
最終ジャッジ日時 2025-01-25 19:15:33
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

mt19937_64 rng;

struct primenumber {
    vector<int> spf;
    vector<long long>pura;
    primenumber(int N) {
        init(N);
    }
    void init(int N) {
        spf.assign(N+1,0);
        pura.assign(N+1,0);
        for(int i = 0; i <= N; i++) {
            spf[i] = i;
            pura[i] = rng()%1001001001001001001;
        }
        for(int i = 2; i*i <= N; i++) {
            if(spf[i] == i) {
                for(int j = i*i; j <= N; j += i) {
                    if(spf[j] == j) {
                        spf[j] = i;
                    }
                }
            }
        }
    }
    bool is_prime(long long n) {
        bool flag = true;
        if(n == 1) {
            flag = false;
        }
        for(long long i = 2; i*i <= n; i++) {
            if(n%i == 0) {
                flag = false;
            }
        }
        return flag;
    }
    long long get(int n) {
        long long m = 0;
        while(n != 1) {
            m ^= pura[spf[n]];
            n /= spf[n];
        }
        return m;
    }
};

int main() {
    int n;
    cin >> n;
    vector<int>a(n);
    vector<long long>b(n+1);
    primenumber zz(1000001);
    for(int i = 0; i < n; i++) {
        cin >> a[i];
        b[i+1] = b[i]^zz.get(a[i]);
    }
    sort(b.begin(),b.end());
    long long ans = 0;
    map<long long,int>tmp;
    for(int i = 0; i < b.size(); i++) {
        ans += tmp[b[i]];
        tmp[b[i]]++;
    }
    cout << ans << endl;
}
0