結果

問題 No.1548 [Cherry 2nd Tune B] 貴方と私とサイクルとモーメント
ユーザー kotatsugamekotatsugame
提出日時 2021-06-11 23:05:31
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 219 ms / 4,500 ms
コード長 1,534 bytes
コンパイル時間 917 ms
コンパイル使用メモリ 85,016 KB
実行使用メモリ 18,580 KB
最終ジャッジ日時 2024-05-08 19:19:13
合計ジャッジ時間 9,406 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 156 ms
10,716 KB
testcase_03 AC 52 ms
7,168 KB
testcase_04 AC 150 ms
17,648 KB
testcase_05 AC 16 ms
5,376 KB
testcase_06 AC 110 ms
10,368 KB
testcase_07 AC 147 ms
18,304 KB
testcase_08 AC 162 ms
10,808 KB
testcase_09 AC 114 ms
7,424 KB
testcase_10 AC 112 ms
17,536 KB
testcase_11 AC 129 ms
17,688 KB
testcase_12 AC 78 ms
5,376 KB
testcase_13 AC 87 ms
17,588 KB
testcase_14 AC 118 ms
6,784 KB
testcase_15 AC 99 ms
18,304 KB
testcase_16 AC 131 ms
11,152 KB
testcase_17 AC 161 ms
17,280 KB
testcase_18 AC 143 ms
18,048 KB
testcase_19 AC 63 ms
6,912 KB
testcase_20 AC 107 ms
17,664 KB
testcase_21 AC 104 ms
10,536 KB
testcase_22 AC 216 ms
18,396 KB
testcase_23 AC 217 ms
18,432 KB
testcase_24 AC 219 ms
18,368 KB
testcase_25 AC 210 ms
18,432 KB
testcase_26 AC 218 ms
18,432 KB
testcase_27 AC 213 ms
18,580 KB
testcase_28 AC 215 ms
18,388 KB
testcase_29 AC 214 ms
18,332 KB
testcase_30 AC 214 ms
18,432 KB
testcase_31 AC 218 ms
18,420 KB
testcase_32 AC 197 ms
18,436 KB
testcase_33 AC 197 ms
18,420 KB
testcase_34 AC 196 ms
18,376 KB
testcase_35 AC 199 ms
18,340 KB
testcase_36 AC 190 ms
18,336 KB
testcase_37 AC 189 ms
18,436 KB
testcase_38 AC 186 ms
18,432 KB
testcase_39 AC 190 ms
18,432 KB
testcase_40 AC 190 ms
18,332 KB
testcase_41 AC 188 ms
18,432 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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