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

提出ソース

結果

問題	No.880 Yet Another Segment Tree Problem
ユーザー	HIR180
提出日時	2019-09-10 21:47:17
言語	C++11 (gcc 11.4.0)
結果	WA
実行時間	-
コード長	3,078 bytes
コンパイル時間	5,762 ms
コンパイル使用メモリ	174,024 KB
実行使用メモリ	11,388 KB
最終ジャッジ日時	2023-09-15 13:15:38
合計ジャッジ時間	11,952 ms
ジャッジサーバーID （参考情報）	judge11 / judge14

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	3 ms 9,716 KB
testcase_01	AC	4 ms 9,732 KB
testcase_02	AC	4 ms 9,968 KB
testcase_03	WA	-
testcase_04	AC	4 ms 9,668 KB
testcase_05	WA	-
testcase_06	AC	4 ms 9,704 KB
testcase_07	WA	-
testcase_08	AC	5 ms 9,784 KB
testcase_09	AC	4 ms 9,808 KB
testcase_10	AC	4 ms 9,680 KB
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	AC	731 ms 11,348 KB
testcase_18	AC	714 ms 11,212 KB
testcase_19	AC	150 ms 10,372 KB
testcase_20	AC	150 ms 10,476 KB
testcase_21	AC	150 ms 10,456 KB
testcase_22	AC	154 ms 10,512 KB
testcase_23	AC	153 ms 10,364 KB
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-
testcase_30	WA	-
testcase_31	WA	-
testcase_32	AC	245 ms 9,752 KB
testcase_33	AC	113 ms 9,692 KB
testcase_34	AC	118 ms 9,660 KB
testcase_35	AC	113 ms 9,628 KB
testcase_36	AC	113 ms 9,692 KB
testcase_37	AC	113 ms 9,800 KB

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

ソースコード

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 1500000000
#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){
   return __gcd(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 L[1<<18],mx[1<<18],sum[1<<18];
ll upd[1<<18];
void push(int k,ll val){
    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;
	L[x] = sum[x] = mx[x] = v;
	while(x){
		x = (x-1)/2;
		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){
		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);
	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 && L[k]*(r-l+1) == sum[k]){
		L[k] = gcdd(x,L[k]);
		sum[k] = L[k] * (r-l+1);
		upd[k] = L[k];
		mx[k] = L[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) L[k] = gcdd(L[k],x);
	else 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

結果

テストケース

ソースコード