結果

問題 No.1036 Make One With GCD 2
コンテスト
ユーザー mugen_1337
提出日時 2020-04-25 08:39:52
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 333 ms / 2,000 ms
コード長 1,984 bytes
コンパイル時間 3,072 ms
コンパイル使用メモリ 202,252 KB
最終ジャッジ日時 2025-01-10 00:52:57
ジャッジサーバーID
（参考情報） 		judge1 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define ALL(x) x.begin(),x.end()
#define rep(i,n) for(int i=0;i<(n);i++)
#define debug(v) cout<<#v<<":";for(auto x:v){cout<<x<<' ';}cout<<endl;
#define INF 1000000000
#define mod 1000000007
using ll=long long;
const ll LINF=1001002003004005006ll;
int dx[]={1,0,-1,0};
int dy[]={0,1,0,-1};
// ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<class T>bool chmin(T &a,const T &b){if(b<a){a=b;return true;}return false;}

template<typename Monoid>
struct SWAG{
    using F=function<Monoid(Monoid,Monoid)>;
    stack<pair<Monoid,Monoid>> fr,ba;
    const F f;
    const Monoid gen;
    SWAG(const F f,const Monoid gen):f(f),gen(gen){}
    
    Monoid get(){
        Monoid a=ba.empty()?gen:ba.top().second;
        Monoid b=fr.empty()?gen:fr.top().second;
        return f(a,b);
    }

    void push(Monoid x){
        Monoid t=fr.empty()?gen:fr.top().second;
        fr.push(make_pair(x,f(t,x)));
    }

    void pop(){
        if(ba.empty()){
            Monoid t=gen;
            while(!fr.empty()){
                pair<Monoid,Monoid> p=fr.top();fr.pop();
                t=f(t,p.first);
                ba.push(make_pair(p.first,t));
            }
        }
        ba.pop();
    }

    bool empty(){
        return (ba.empty() and fr.empty());
    }

    size_t size(){
        return ba.size()+fr.size();
    }
};

signed main(){
    cin.tie(0);
    ios::sync_with_stdio(0);
    
    int n;cin>>n;
    vector<ll> a(n);
    rep(i,n){
        cin>>a[i];
    }

    function<ll(ll,ll)> g=[](ll a,ll b){return gcd(a,b);};
    SWAG<ll> sw(g,0);

    ll r=0,res=0;
    for(ll l=0;l<n;l++){
        while(r<n and g(sw.get(),a[r])!=1){
            sw.push(a[r]);
            r++;
        }

        res+=(ll)sw.size();
        if(r==l){
            r++;
        }else{
            sw.pop();
        }
    }
    cout<<(ll)n*(n+1)/2-res<<endl;
    return 0;
}
0