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

提出ソース

結果

問題	No.880 Yet Another Segment Tree Problem
ユーザー	HIR180
提出日時	2019-09-10 21:32:49
言語	C++11 (gcc 11.4.0)
結果	TLE
実行時間	-
コード長	3,362 bytes
コンパイル時間	1,388 ms
コンパイル使用メモリ	176,012 KB
実行使用メモリ	16,884 KB
最終ジャッジ日時	2023-09-15 13:13:53
合計ジャッジ時間	14,160 ms
ジャッジサーバーID （参考情報）	judge14 / judge15

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	3 ms 11,768 KB
testcase_01	AC	5 ms 11,828 KB
testcase_02	AC	6 ms 11,780 KB
testcase_03	AC	5 ms 11,704 KB
testcase_04	AC	4 ms 11,776 KB
testcase_05	AC	4 ms 11,800 KB
testcase_06	AC	4 ms 11,716 KB
testcase_07	AC	5 ms 11,816 KB
testcase_08	AC	5 ms 11,712 KB
testcase_09	AC	5 ms 11,708 KB
testcase_10	AC	5 ms 11,716 KB
testcase_11	AC	567 ms 12,960 KB
testcase_12	AC	558 ms 13,112 KB
testcase_13	AC	421 ms 12,956 KB
testcase_14	AC	559 ms 13,180 KB
testcase_15	AC	593 ms 13,084 KB
testcase_16	AC	625 ms 13,448 KB
testcase_17	AC	807 ms 13,332 KB
testcase_18	AC	838 ms 13,204 KB
testcase_19	TLE	-
testcase_20	--	-
testcase_21	--	-
testcase_22	--	-
testcase_23	--	-
testcase_24	--	-
testcase_25	--	-
testcase_26	--	-
testcase_27	--	-
testcase_28	--	-
testcase_29	--	-
testcase_30	--	-
testcase_31	--	-
testcase_32	--	-
testcase_33	--	-
testcase_34	--	-
testcase_35	--	-
testcase_36	--	-
testcase_37	--	-

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

ソースコード

raw source code

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
typedef pair<int,P> P1;
typedef pair<P,P> P2;
#define pu push
#define pb push_back
#define mp make_pair
#define eps 1e-7
#define INF 2000000000
#define mod 998244353
#define fi first
#define sc second
#define rep(i,x) for(int i=0;i<x;i++)
#define repn(i,x) for(int i=1;i<=x;i++)
#define SORT(x) sort(x.begin(),x.end())
#define ERASE(x) x.erase(unique(x.begin(),x.end()),x.end())
#define POSL(x,v) (lower_bound(x.begin(),x.end(),v)-x.begin())
#define POSU(x,v) (upper_bound(x.begin(),x.end(),v)-x.begin())
ll gcdd(ll a,ll b){
    if(a < b) swap(a,b);
    if(b == 0) return a;
    return gcdd(b,a%b);
}
ll makeL(ll a,ll b){
    if(a == 0) return b;
    if(b == 0) return a;
    if(a >= INF || b >= INF) return INF;
    return min((ll)(INF),a*b/gcdd(a,b));
}
ll G[1<<18],L[1<<18],mx[1<<18],sum[1<<18];
ll upd[1<<18];
void push(int k,ll val){
    G[k*2+1] = G[k*2+2] = L[k*2+1] = L[k*2+2] = mx[k*2+1] = mx[k*2+2] = upd[k*2+1] = upd[k*2+2] = upd[k];
    sum[k*2+1] = sum[k*2+2] = val * upd[k];
    upd[k] = 0;
}
void update(int x,ll v){
	x += (1<<17)-1;
	G[x] = L[x] = sum[x] = mx[x] = v;
	while(x){
		x = (x-1)/2;
		G[x] = gcdd(G[x*2+1],G[x*2+2]);
		assert(gcdd(L[2*x+1],L[2*x+2]) > 0);
		L[x] = makeL(L[x*2+1],L[x*2+2]);
		sum[x] = sum[x*2+1]+sum[x*2+2];
		mx[x] = max(mx[x*2+1],mx[x*2+2]);
	}
}
void set_val(int a,int b,int k,int l,int r,ll x){
	if(r<a || b<l) return;
	if(a<=l && r<=b){
		G[k] = L[k] = x;
		sum[k] = x * (r-l+1);
		mx[k] = x;
		upd[k] = x;
		return;
	}
	if(upd[k]) push(k,(r-l+1)/2);
	set_val(a,b,k*2+1,l,(l+r)/2,x);
	set_val(a,b,k*2+2,(l+r)/2+1,r,x);
	G[k] = gcdd(G[k*2+1],G[k*2+2]);
	L[k] = makeL(L[k*2+1],L[k*2+2]);
	sum[k] = sum[k*2+1]+sum[k*2+2];
	mx[k] = max(mx[k*2+1],mx[k*2+2]);
}
void gcd_val(int a,int b,int k,int l,int r,ll x){
	if(r<a || b<l || (L[k]<=x && x%L[k]==0)) return;
	if(a<=l && r<=b && G[k] == L[k]){
		G[k] = L[k] = gcdd(x,G[k]);
		sum[k] = G[k] * (r-l+1);
		upd[k] = G[k];
		mx[k] = G[k];
		return;
	}
	if(upd[k]) push(k,(r-l+1)/2);
	gcd_val(a,b,k*2+1,l,(l+r)/2,x);
	gcd_val(a,b,k*2+2,(l+r)/2+1,r,x);
	//assume time complexity is log^2
	if(a<=l && r<=b){
		G[k] = gcdd(G[k],x);
		if(L[k] < 2e9) L[k] = gcdd(L[k],x);
	}
	else{
		G[k] = gcdd(G[k*2+1],G[k*2+2]);
		L[k] = makeL(L[k*2+1],L[k*2+2]);
	}
	sum[k] = sum[k*2+1]+sum[k*2+2];
	mx[k] = max(mx[k*2+1],mx[k*2+2]);
}
ll calc_max(int a,int b,int k,int l,int r){
	if(r<a || b<l) return 0;
	if(a<=l && r<=b){
		return mx[k];
	}
	if(upd[k]) push(k,(r-l+1)/2);
	return max(calc_max(a,b,k*2+1,l,(l+r)/2),calc_max(a,b,k*2+2,(l+r)/2+1,r));
}
ll calc_sum(int a,int b,int k,int l,int r){
	if(r<a || b<l) return 0;
	if(a<=l && r<=b){
		return sum[k];
	}
	if(upd[k]) push(k,(r-l+1)/2);
	return (calc_sum(a,b,k*2+1,l,(l+r)/2)+calc_sum(a,b,k*2+2,(l+r)/2+1,r));
}
int main(){
	int n,q;
	scanf("%d%d",&n,&q);
	repn(i,n){
		ll a; scanf("%lld",&a);
		update(i,a);
	}
	
	rep(i,q){
		int ty,L,R; ll x; scanf("%d%d%d",&ty,&L,&R);
		if(ty <= 2) scanf("%lld",&x);
		if(ty == 1){
			set_val(L,R,0,0,(1<<17)-1,x);
		}
		else if(ty == 2){
			gcd_val(L,R,0,0,(1<<17)-1,x);
		}
		else if(ty == 3){
			printf("%lld\n",calc_max(L,R,0,0,(1<<17)-1));
		}
		else{
			printf("%lld\n",calc_sum(L,R,0,0,(1<<17)-1));
		}
	}
	return 0;
}

yukicoder

結果

テストケース

ソースコード