結果

問題 No.1548 [Cherry 2nd Tune B] 貴方と私とサイクルとモーメント
ユーザー kotatsugamekotatsugame
提出日時 2021-06-11 23:05:31
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 228 ms / 4,500 ms
コード長 1,534 bytes
コンパイル時間 1,154 ms
コンパイル使用メモリ 85,088 KB
実行使用メモリ 18,560 KB
最終ジャッジ日時 2024-12-15 02:05:18
合計ジャッジ時間 9,861 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 42
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:32:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   32 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
#include<algorithm>
#include<atcoder/modint>
#include<atcoder/lazysegtree>
using namespace std;
using mint=atcoder::modint998244353;
struct dat{
	mint S[5];
	dat(){}
	dat(mint x,mint len=1)
	{
		mint p=len;
		for(int i=0;i<5;i++)
		{
			S[i]=p;
			p*=x;
		}
	}
	dat operator+(const dat&rhs)const
	{
		dat ret;
		for(int i=0;i<5;i++)ret.S[i]=S[i]+rhs[i];
		return ret;
	}
	const mint&operator[](int k)const{return S[k];}
};
dat op(dat a,dat b){return a+b;}
dat e(){return dat();}
dat mp(int f,dat x){return f==-1?x:dat(mint(f),x[0]);}
int cmp(int f,int g){return f==-1?g:f;}
int id(){return-1;}
main()
{
	int N;
	cin>>N;
	vector<dat>A(N);
	for(int i=0;i<N;i++)
	{
		int a;cin>>a;
		A[i]=dat(mint(a));
	}
	atcoder::lazy_segtree<dat,op,e,int,mp,cmp,id>P(A);
	int Q;cin>>Q;
	for(;Q--;)
	{
		int t;
		cin>>t;
		int u,v,w;
		cin>>u>>v>>w;
		if(u>v)swap(u,v);
		u--,v--,w--;
		if(t==0)
		{
			int b;cin>>b;
			if(u<w&&w<v)P.apply(u,v+1,b);
			else
			{
				P.apply(0,u+1,b);
				P.apply(v,N,b);
			}
		}
		else
		{
			dat mid=P.prod(u,v+1);
			if(!(u<w&&w<v))
			{
				dat all=P.all_prod();
				if(u+1==v)mid=all;
				else
				{
					mid=P.prod(u+1,v);
					for(int i=0;i<5;i++)mid.S[i]=all[i]-mid[i];
				}
			}
			mint invlen=1/mid[0];
			for(int i=1;i<5;i++)mid.S[i]*=invlen;
			mint ans;
			if(t==1)ans=0;
			else if(t==2)ans=mid[2]-mid[1]*mid[1];
			else if(t==3)ans=mid[3]-3*mid[2]*mid[1]+2*mid[1].pow(3);
			else ans=mid[4]-4*mid[3]*mid[1]+6*mid[2]*mid[1]*mid[1]-3*mid[1].pow(4);
			cout<<ans.val()<<endl;
		}
	}
}
0