結果

問題 No.1036 Make One With GCD 2
ユーザー chocorusk
提出日時 2020-03-14 18:51:04
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 2,414 bytes
コンパイル時間 1,417 ms
コンパイル使用メモリ 118,960 KB
実行使用メモリ 33,084 KB
最終ジャッジ日時 2024-11-07 02:01:06
合計ジャッジ時間 9,586 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 4
other TLE * 1 -- * 40
権限があれば一括ダウンロードができます

ソースコード

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>
#define popcount __builtin_popcount
using namespace std;
typedef long long int ll;
typedef pair<int, int> P;
template<typename Monoid>
struct SegmentTree{
	using F=function<Monoid(Monoid, Monoid)>;
	int sz;
	vector<Monoid> seg;
	const F f;
	const Monoid e;

	SegmentTree(int n, const F f, const Monoid &e): f(f), e(e){
		sz=1;
		while(sz<n) sz<<=1;
		seg.resize(2*sz, e);
	}

	SegmentTree(int n, const F f, const Monoid &e, vector<Monoid> v): f(f), e(e){
		sz=1;
		while(sz<n) sz<<=1;
		seg.resize(2*sz, e);
		for(int i=0; i<n; i++) seg[i+sz]=v[i];
		for(int i=sz-1; i>=1; i--){
			seg[i]=f(seg[2*i], seg[2*i+1]);
		}
	}

	void update(int k, const Monoid &x){
		k+=sz;
		seg[k]=x;
		while(k>1){
			k>>=1;
			seg[k]=f(seg[2*k], seg[2*k+1]);
		}
	}

	Monoid query(int a, int b){
		a+=sz, b+=sz;
		Monoid ret=e;
		for(;a<b; a>>=1, b>>=1){
			if(b&1) ret=f(ret, seg[--b]);
			if(a&1) ret=f(ret, seg[a++]);
		}
		return ret;
	}

	template<typename C>
	int binarysearch(int st, C &check, Monoid &s, int k, int l, int r){
		if(k>=sz){
			s=f(s, seg[k]);
			if(check(s)) return k-sz;
			else return -1;
		}
		int m=(l+r)>>1;
		if(m<=st) return binarysearch(st, check, s, 2*k+1, m, r);
		if(st<=l && !check(f(s, seg[k]))){
			s=f(s, seg[k]);
			return -1;
		}
		int vl=binarysearch(st, check, s, 2*k, l, m);
		if(vl!=-1) return vl;
		return binarysearch(st, check, s, 2*k+1, m, r);
	}
	
	template<typename C>
	int binarysearch(int st, C &check){
		Monoid s=e;
		return binarysearch(st, check, s, 1, 0, sz);
	}

	Monoid operator[](const int &k) const{
		return seg[k+sz];
	}
};
ll gcd(ll a, ll b){
	if(b==0) return a;
	return gcd(b, a%b);
}
int main()
{
    int n;
	scanf("%d", &n);
	vector<ll> a(n);
	for(int i=0; i<n; i++){
		scanf("%lld", &a[i]);
	}
	SegmentTree<ll> seg(n, [&](ll x, ll y){ return gcd(x, y);}, 0, a);
	ll ans=0;
	for(int i=0; i<n; i++){
		auto check=[](ll x){ return x==1;};
		int r=seg.binarysearch(i, check);
		if(r!=-1) ans+=n-r;
	}
	printf("%lld\n", ans);
	return 0;
}
0