#525164 (C++17) No.877 Range ReLU Query

• 問題: No.877 Range ReLU Query
• ユーザー: fura
• 提出日時: 2020-08-06 03:50:36
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 384 ms / 2,000 ms
• コード長: 2,283 bytes
• コンパイル時間: 2,621 ms
• コンパイル使用メモリ: 215,004 KB
• 実行使用メモリ: 55,784 KB
• 最終ジャッジ日時: 2024-11-08 10:30:39
• 合計ジャッジ時間: 7,246 ms
• ジャッジサーバーID （参考情報）: judge4 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	4 ms; 5,248 KB
testcase_02	AC	3 ms; 5,248 KB
testcase_03	AC	3 ms; 5,248 KB
testcase_04	AC	3 ms; 5,248 KB
testcase_05	AC	2 ms; 5,248 KB
testcase_06	AC	3 ms; 5,248 KB
testcase_07	AC	3 ms; 5,248 KB
testcase_08	AC	4 ms; 5,248 KB
testcase_09	AC	3 ms; 5,248 KB
testcase_10	AC	2 ms; 5,248 KB
testcase_11	AC	350 ms; 48,208 KB
testcase_12	AC	304 ms; 47,708 KB
testcase_13	AC	228 ms; 32,544 KB
testcase_14	AC	240 ms; 30,592 KB
testcase_15	AC	375 ms; 54,892 KB
testcase_16	AC	374 ms; 54,172 KB
testcase_17	AC	384 ms; 54,932 KB
testcase_18	AC	382 ms; 55,224 KB
testcase_19	AC	195 ms; 55,652 KB
testcase_20	AC	329 ms; 55,784 KB

ソースコード

#include <bits/stdc++.h>

#define rep(i,n) for(int i=0;i<(n);i++)

using namespace std;
using lint=long long;

template<class M>
class segment_tree{
	int sz;
	vector<M> seg;
public:
	segment_tree(){}
	segment_tree(int n){ build(n); }
	template<class T>
	segment_tree(const vector<T>& a){ build(a); }
	void build(int n){
		for(sz=1;sz<n;sz<<=1);
		seg.assign(2*sz,M());
	}
	template<class T>
	void build(const vector<T>& a){
		int n=a.size();
		build(n);
		rep(i,n) seg[sz+i]=a[i];
		for(int i=sz-1;i>0;i--){
			seg[i]=seg[2*i]*seg[2*i+1];
		}
	}
	const M& operator[](int i)const{
		return seg[sz+i];
	}
	void update(int i,const M& x){
		i+=sz;
		seg[i]=x;
		for(i>>=1;i>0;i>>=1) seg[i]=seg[2*i]*seg[2*i+1];
	}
	M get(int l,int r)const{
		M lcum,rcum;
		for(int a=l+sz,b=r+sz;a<b;a>>=1,b>>=1){
			if(a&1) lcum=lcum*seg[a++];
			if(b&1) rcum=seg[--b]*rcum;
		}
		return lcum*rcum;
	}
	template<class N,class F>
	N query(int l,int r,const F& f){
		N lcum,rcum;
		for(int a=l+sz,b=r+sz;a<b;a>>=1,b>>=1){
			if(a&1) lcum=lcum*f(seg[a++]);
			if(b&1) rcum=f(seg[--b])*rcum;
		}
		return lcum*rcum;
	}
};

template<class T>
class sequence_sum_monoid{
	vector<T> a,cum;
public:
	sequence_sum_monoid():cum(1){}
	sequence_sum_monoid(const T& val):a(1,val),cum{0,val}{}
	sequence_sum_monoid(const vector<T>& x):a(x),cum(x.size()+1){
		sort(a.begin(),a.end());
		rep(i,x.size()) cum[i+1]=cum[i]+a[i];
	}
	sequence_sum_monoid operator*(const sequence_sum_monoid& x)const{
		vector<T> res;
		merge(a.begin(),a.end(),x.a.begin(),x.a.end(),back_inserter(res));
		return res;
	}
	lint f(lint t){
		int idx=upper_bound(a.begin(),a.end(),t)-a.begin();
		return cum.back()-cum[idx]-(a.size()-idx)*t;
	}
};

template<class T>
class add_monoid{
	T a;
public:
	add_monoid():a(){}
	add_monoid(const T& val):a(val){}
	add_monoid operator*(const add_monoid& x)const{
		return a+x.a;
	}
	T& get(){ return a; }
	const T& get()const{ return a; }
};

int main(){
	int n,q; scanf("%d%d",&n,&q);
	vector<int> a(n);
	rep(i,n) scanf("%d",&a[i]);

	segment_tree<sequence_sum_monoid<lint>> S(a);
	rep(_,q){
		int l,r,t; scanf("%*d%d%d%d",&l,&r,&t); l--;
		auto ans=S.query<add_monoid<lint>>(l,r,[&](sequence_sum_monoid<lint>& x)->add_monoid<lint>{
			return x.f(t);
		});
		printf("%lld\n",ans.get());
	}

	return 0;
}