#503183 (C++14) No.1099 Range Square Sum

• 問題: No.1099 Range Square Sum
• ユーザー: kotatsugame
• 提出日時: 2020-06-26 21:54:53
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 549 ms / 2,000 ms
• コード長: 2,542 bytes
• コンパイル時間: 895 ms
• コンパイル使用メモリ: 87,944 KB
• 実行使用メモリ: 22,912 KB
• 最終ジャッジ日時: 2024-07-04 20:57:05
• 合計ジャッジ時間: 7,160 ms
• ジャッジサーバーID （参考情報）: judge3 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	1 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	1 ms; 5,376 KB
testcase_06	AC	1 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	1 ms; 5,376 KB
testcase_11	AC	5 ms; 5,376 KB
testcase_12	AC	5 ms; 5,376 KB
testcase_13	AC	5 ms; 5,376 KB
testcase_14	AC	5 ms; 5,376 KB
testcase_15	AC	5 ms; 5,376 KB
testcase_16	AC	5 ms; 5,376 KB
testcase_17	AC	4 ms; 5,376 KB
testcase_18	AC	4 ms; 5,376 KB
testcase_19	AC	5 ms; 5,376 KB
testcase_20	AC	5 ms; 5,376 KB
testcase_21	AC	532 ms; 22,784 KB
testcase_22	AC	515 ms; 22,912 KB
testcase_23	AC	538 ms; 22,784 KB
testcase_24	AC	549 ms; 22,912 KB
testcase_25	AC	518 ms; 22,912 KB
testcase_26	AC	400 ms; 22,784 KB
testcase_27	AC	390 ms; 22,912 KB
testcase_28	AC	390 ms; 22,784 KB
testcase_29	AC	439 ms; 22,784 KB
testcase_30	AC	406 ms; 22,912 KB

コンパイルメッセージ

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

ソースコード

#include<iostream>
using namespace std;
#include<vector>
#include<functional>
template<typename T>
struct lazysegtree{
	using F=function<T(T,T)>;
	using G=function<T(T,T,int,int)>;
	const F calcfn,lazycalcfn;
	const G updatefn;
	int n;
	T defvalue,lazydefvalue;
	vector<T>dat,lazy;
	vector<bool>lazyflag;
	lazysegtree(int n_=0,T defvalue_=0,
		const F calcfn_=[](T a,T b){return a+b;},
		const F lazycalcfn_=[](T a,T b){return a+b;},
		const G updatefn_=[](T a,T b,int l,int r){return a+b*(r-l);},
		T lazydefvalue_=0
	):defvalue(defvalue_),lazydefvalue(lazydefvalue_),
		calcfn(calcfn_),lazycalcfn(lazycalcfn_),updatefn(updatefn_)
	{
		n=1;
		while(n<n_)n<<=1;
		dat.assign(2*n-1,defvalue);
		lazy.assign(2*n-1,lazydefvalue);
		lazyflag.assign(2*n-1,false);
	}
	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[2*i+1],dat[2*i+2]);
	}
	void eval(int i,int l,int r)
	{
		if(lazyflag[i])
		{
			dat[i]=updatefn(dat[i],lazy[i],l,r);
			if(r-l>1)
			{
				lazy[2*i+1]=lazycalcfn(lazy[2*i+1],lazy[i]);
				lazy[2*i+2]=lazycalcfn(lazy[2*i+2],lazy[i]);
				lazyflag[2*i+1]=lazyflag[2*i+2]=true;
			}
			lazy[i]=lazydefvalue;
			lazyflag[i]=false;
		}
	}
	void update(int a,int b,T x,int k=0,int l=0,int r=-1)//[a,b)
	{
		if(r<0)r=n;
		eval(k,l,r);
		if(b<=l||r<=a)return;
		else if(a<=l&&r<=b)
		{
			lazy[k]=lazycalcfn(lazy[k],x);
			lazyflag[k]=true;
			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);
			dat[k]=calcfn(dat[2*k+1],dat[2*k+2]);
		}
	}
	T query(int a,int b,int k=0,int l=0,int r=-1)//[a,b)
	{
		if(r<0)r=n;
		eval(k,l,r);
		if(b<=l||r<=a)return defvalue;
		else if(a<=l&&r<=b)return dat[k];
		else return calcfn(
			query(a,b,2*k+1,l,(l+r)/2),
			query(a,b,2*k+2,(l+r)/2,r)
		);
	}
};
int N;
main()
{
	cin>>N;
	vector<pair<long,long> >A(N);
	for(int i=0;i<N;i++)
	{
		cin>>A[i].first;
		A[i].second=A[i].first*A[i].first;
	}
	lazysegtree<pair<long,long> >P(N,make_pair(0L,0L),
	[](pair<long,long>a,pair<long,long>b){return make_pair(a.first+b.first,a.second+b.second);},
	[](pair<long,long>a,pair<long,long>b){return make_pair(a.first+b.first,0L);},
	[](pair<long,long>a,pair<long,long>b,int l,int r){
		return make_pair(a.first+b.first*(r-l),a.second+2*a.first*b.first+b.first*b.first*(r-l));
	},make_pair(0L,0L));
	P.copy(A);
	int Q;cin>>Q;
	for(;Q--;)
	{
		int q,l,r;cin>>q>>l>>r;
		if(q==1)
		{
			int x;cin>>x;
			P.update(l-1,r,make_pair(x,0));
		}
		else
		{
			cout<<P.query(l-1,r).second<<endl;
		}
	}
}