結果

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

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#include <list>
#include <atcoder/all>
#define popcount __builtin_popcount
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef pair<int, int> P;
using mint=modint998244353;
const int MAX=1000010;
int pf[MAX];
void sieve(){
    for(int i=2; i<MAX; i++) pf[i]=i;
    for(int i=2; i<MAX; i++){
        if(pf[i]==i){
            for(int j=(i<<1); j<MAX; j+=i){
                pf[j]=i;
            }
        }
    }
}
using ull=unsigned long long;
mt19937_64 mt(334);
int n;
int a[200020];
ull x[MAX];
ull v[200020], vs[200020];
int main()
{
    cin>>n;
    for(int i=0; i<n; i++) cin>>a[i];
    sieve();
    for(int i=2; i<MAX; i++){
        if(pf[i]==i) x[i]=mt();
    }
    for(int i=0; i<n; i++){
        if(a[i]==1) continue;
        while(pf[a[i]]!=a[i]){
            v[i]^=x[pf[a[i]]];
            a[i]/=pf[a[i]];
        }
        v[i]^=x[a[i]];
    }
    for(int i=0; i<n; i++){
        vs[i+1]=(vs[i]^v[i]);
    }
    map<ull, ll> mp;
    for(int i=0; i<=n; i++) mp[vs[i]]++;
    ll ans=0;
    for(auto p:mp){
        ans+=p.second*(p.second-1)/2;
    }
    cout<<ans<<endl;
    return 0;
}
0