#377489 (C++11) No.880 Yet Another Segment Tree Problem

提出ソース

結果

問題	No.880 Yet Another Segment Tree Problem
ユーザー	chocorusk
提出日時	2019-09-08 14:58:22
言語	C++11 (gcc 11.4.0)
結果	AC
実行時間	699 ms / 5,000 ms
コード長	2,827 bytes
コンパイル時間	1,253 ms
コンパイル使用メモリ	106,284 KB
実行使用メモリ	12,160 KB
最終ジャッジ日時	2024-09-22 19:36:59
合計ジャッジ時間	14,321 ms
ジャッジサーバーID （参考情報）	judge1 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 5,248 KB
testcase_01	AC	3 ms 5,248 KB
testcase_02	AC	4 ms 5,376 KB
testcase_03	AC	4 ms 5,376 KB
testcase_04	AC	3 ms 5,376 KB
testcase_05	AC	3 ms 5,376 KB
testcase_06	AC	2 ms 5,376 KB
testcase_07	AC	4 ms 5,376 KB
testcase_08	AC	4 ms 5,376 KB
testcase_09	AC	5 ms 5,376 KB
testcase_10	AC	4 ms 5,376 KB
testcase_11	AC	604 ms 11,904 KB
testcase_12	AC	585 ms 12,032 KB
testcase_13	AC	418 ms 11,904 KB
testcase_14	AC	576 ms 12,160 KB
testcase_15	AC	615 ms 12,160 KB
testcase_16	AC	643 ms 12,032 KB
testcase_17	AC	699 ms 12,032 KB
testcase_18	AC	686 ms 12,032 KB
testcase_19	AC	305 ms 12,160 KB
testcase_20	AC	314 ms 12,160 KB
testcase_21	AC	314 ms 12,160 KB
testcase_22	AC	313 ms 12,160 KB
testcase_23	AC	320 ms 12,160 KB
testcase_24	AC	266 ms 12,160 KB
testcase_25	AC	277 ms 12,160 KB
testcase_26	AC	282 ms 12,032 KB
testcase_27	AC	278 ms 12,160 KB
testcase_28	AC	282 ms 12,160 KB
testcase_29	AC	561 ms 12,032 KB
testcase_30	AC	601 ms 12,032 KB
testcase_31	AC	642 ms 12,160 KB
testcase_32	AC	205 ms 12,160 KB
testcase_33	AC	249 ms 12,160 KB
testcase_34	AC	265 ms 12,160 KB
testcase_35	AC	250 ms 12,160 KB
testcase_36	AC	257 ms 12,160 KB
testcase_37	AC	256 ms 12,032 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#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>
#define popcount __builtin_popcount
using namespace std;
typedef long long int ll;
typedef pair<int, int> P;
const ll INF=2e18;
ll gcd(ll a, ll b){
	if(b==0) return a;
	return gcd(b, a%b);
}
ll lcm(ll a, ll b){
	if(a==INF || b==INF) return INF;
	ll g=gcd(a, b);
	if(a/g>=(INF+b-1)/b) return INF;
	else return a/g*b;
}

int n, sz;
vector<ll> mx, sum, vlcm, lazy;
ll a[1<<17];

void init(){
	sz=1;
	while(sz<n) sz<<=1;
	mx.resize(2*sz-1);
	sum.resize(2*sz-1);
	vlcm.resize(2*sz-1, 1);
	lazy.resize(2*sz-1);
	for(int i=0; i<n; i++){
		mx[i+sz-1]=sum[i+sz-1]=vlcm[i+sz-1]=a[i];
	}
	for(int i=sz-2; i>=0; i--){
		mx[i]=max(mx[2*i+1], mx[2*i+2]);
		sum[i]=sum[2*i+1]+sum[2*i+2];
		vlcm[i]=lcm(vlcm[2*i+1], vlcm[2*i+2]);
	}
}

void eval(int k, int l, int r){
	if(lazy[k]!=0){
		mx[k]=vlcm[k]=lazy[k];
		sum[k]=(r-l)*lazy[k];
		if(k<sz-1){
			lazy[2*k+1]=lazy[k];
			lazy[2*k+2]=lazy[k];
		}
	}
	lazy[k]=0;
}

void update(int a, int b, ll x, int k, int l, int r){
	eval(k, l, r);
	if(r<=a || b<=l) return;
	if(a<=l && r<=b){
		lazy[k]=x;
		eval(k, l, r);
	}else{
		update(a, b, x, 2*k+1, l, (l+r)/2);
		update(a, b, x, 2*k+2, (l+r)/2, r);
		mx[k]=max(mx[2*k+1], mx[2*k+2]);
		sum[k]=sum[2*k+1]+sum[2*k+2];
		vlcm[k]=lcm(vlcm[2*k+1], vlcm[2*k+2]);
	}
}

void update_gcd(int a, int b, ll x, int k, int l, int r){
	eval(k, l, r);
	if(r<=a || b<=l || x%vlcm[k]==0) return;
	if(a<=l && r<=b && mx[k]*(r-l)==sum[k]){
		lazy[k]=gcd(x, mx[k]);
		eval(k, l, r);
	}else{
		update_gcd(a, b, x, 2*k+1, l, (l+r)/2);
		update_gcd(a, b, x, 2*k+2, (l+r)/2, r);
		mx[k]=max(mx[2*k+1], mx[2*k+2]);
		sum[k]=sum[2*k+1]+sum[2*k+2];
		vlcm[k]=lcm(vlcm[2*k+1], vlcm[2*k+2]);
	}
}

ll max_query(int a, int b, int k, int l, int r){
	eval(k, l, r);
	if(r<=a || b<=l) return 0;
	if(a<=l && r<=b) return mx[k];
	else return max(max_query(a, b, 2*k+1, l, (l+r)/2), max_query(a, b, 2*k+2, (l+r)/2, r));
}

ll sum_query(int a, int b, int k, int l, int r){
	eval(k, l, r);
	if(r<=a || b<=l) return 0;
	if(a<=l && r<=b) return sum[k];
	else return sum_query(a, b, 2*k+1, l, (l+r)/2)+sum_query(a, b, 2*k+2, (l+r)/2, r);
}

int main()
{
	int q; cin>>n>>q;
	for(int i=0; i<n; i++) cin>>a[i];
	init();
	for(int i=0; i<q; i++){
		int t, l, r; cin>>t>>l>>r;
		ll x;
		if(t<=2) cin>>x;
		if(t==1) update(l-1, r, x, 0, 0, sz);
		else if(t==2) update_gcd(l-1, r, x, 0, 0, sz);
		else if(t==3) cout<<max_query(l-1, r, 0, 0, sz)<<endl;
		else cout<<sum_query(l-1, r, 0, 0, sz)<<endl;
	}
	return 0;
}

yukicoder

結果

テストケース

ソースコード