#377026 (C++14) No.877 Range ReLU Query

• 問題: No.877 Range ReLU Query
• ユーザー: kotatsugame
• 提出日時: 2019-09-07 03:41:55
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 387 ms / 2,000 ms
• コード長: 1,898 bytes
• コンパイル時間: 1,463 ms
• コンパイル使用メモリ: 96,124 KB
• 実行使用メモリ: 10,596 KB
• 最終ジャッジ日時: 2024-11-08 10:13:01
• 合計ジャッジ時間: 6,994 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

入力	結果	実行時間; 実行使用メモリ
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	5 ms; 5,248 KB
testcase_04	AC	2 ms; 5,248 KB
testcase_05	AC	3 ms; 5,248 KB
testcase_06	AC	4 ms; 5,248 KB
testcase_07	AC	3 ms; 5,248 KB
testcase_08	AC	4 ms; 5,248 KB
testcase_09	AC	2 ms; 5,248 KB
testcase_10	AC	4 ms; 5,248 KB
testcase_11	AC	383 ms; 10,420 KB
testcase_12	AC	332 ms; 10,268 KB
testcase_13	AC	262 ms; 8,128 KB
testcase_14	AC	282 ms; 7,952 KB
testcase_15	AC	387 ms; 10,448 KB
testcase_16	AC	375 ms; 10,296 KB
testcase_17	AC	378 ms; 10,440 KB
testcase_18	AC	368 ms; 10,324 KB
testcase_19	AC	339 ms; 10,592 KB
testcase_20	AC	374 ms; 10,596 KB

コンパイルメッセージ

main.cpp:55:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   55 | main()
      | ^~~~

ソースコード

#include<iostream>
#include<algorithm>
using namespace std;
#include<vector>
#include<functional>
#include<limits>
template<typename T>
struct segtree{
	function<T(T,T)>calcfn,updatefn;
	int n;
	T defvalue;
	vector<T>dat;
	segtree(int n_=0,T defvalue_=numeric_limits<T>::max(),
		function<T(T,T)>calcfn_=[](T a,T b){return a<b?a:b;},
		function<T(T,T)>updatefn_=[](T a,T b){return b;}
	):defvalue(defvalue_),calcfn(calcfn_),updatefn(updatefn_)
	{
		n=1;
		while(n<n_)n<<=1;
		dat.assign(2*n-1,defvalue);
	}
	void copy(const vector<T>&v)
	{
		for(int i=0;i<v.size();i++)dat[i+n-1]=v[i];
		for(int i=n-2;i>=0;i--)dat[i]=calcfn(dat[i*2+1],dat[i*2+2]);
	}
	void update(int i,T a)
	{
		i+=n-1;
		dat[i]=updatefn(dat[i],a);
		while(i>0)
		{
			i=(i-1)/2;
			dat[i]=calcfn(dat[2*i+1],dat[2*i+2]);
		}
	}
	T query(int a,int b,int i=0,int l=0,int r=-1)//[a,b)
	{
		if(r<0)r=n;
		if(r<=a||b<=l)return defvalue;
		else if(a<=l&&r<=b)return dat[i];
		else
		{
			return calcfn(
				query(a,b,i*2+1,l,(l+r)/2),
				query(a,b,i*2+2,(l+r)/2,r)
			);
		}
	}
};
int N,Q;
vector<pair<pair<int,int>,pair<int,int> > >E;
vector<pair<int,int> >A;
long ans[1<<17];
main()
{
	cin>>N>>Q;
	segtree<pair<long,int> >P(N,make_pair(0,0),
	[](pair<long,int>a,pair<long,int>b){return make_pair(a.first+b.first,a.second+b.second);}
	);
	for(int i=0;i<N;i++)
	{
		int a;cin>>a;
		A.push_back(make_pair(a,i));
		P.update(i,make_pair(a,1));
	}
	for(int i=0;i<Q;i++)
	{
		int c,l,r,x;cin>>c>>l>>r>>x;
		E.push_back(make_pair(make_pair(x,i),make_pair(l,r)));
	}
	sort(A.begin(),A.end());
	sort(E.begin(),E.end());
	int id=0;
	for(pair<pair<int,int>,pair<int,int> >p:E)
	{
		while(id<N&&A[id].first<p.first.first)
		{
			P.update(A[id].second,make_pair(0,0));
			id++;
		}
		pair<long,int>a=P.query(p.second.first-1,p.second.second);
		ans[p.first.second]=a.first-1L*a.second*p.first.first;
	}
	for(int i=0;i<Q;i++)cout<<ans[i]<<endl;
}