結果

問題 No.1548 [Cherry 2nd Tune B] 貴方と私とサイクルとモーメント
ユーザー kotatsugamekotatsugame
提出日時 2021-06-11 23:05:31
言語 C++14
(gcc 12.3.0 + boost 1.83.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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 161 ms
10,752 KB
testcase_03 AC 52 ms
6,912 KB
testcase_04 AC 157 ms
17,664 KB
testcase_05 AC 15 ms
5,248 KB
testcase_06 AC 112 ms
10,240 KB
testcase_07 AC 146 ms
18,304 KB
testcase_08 AC 163 ms
10,880 KB
testcase_09 AC 116 ms
7,296 KB
testcase_10 AC 118 ms
17,408 KB
testcase_11 AC 137 ms
17,664 KB
testcase_12 AC 82 ms
5,248 KB
testcase_13 AC 91 ms
17,536 KB
testcase_14 AC 119 ms
6,912 KB
testcase_15 AC 104 ms
18,304 KB
testcase_16 AC 136 ms
11,136 KB
testcase_17 AC 172 ms
17,280 KB
testcase_18 AC 153 ms
17,920 KB
testcase_19 AC 65 ms
6,912 KB
testcase_20 AC 114 ms
17,536 KB
testcase_21 AC 108 ms
10,496 KB
testcase_22 AC 225 ms
18,432 KB
testcase_23 AC 228 ms
18,432 KB
testcase_24 AC 223 ms
18,432 KB
testcase_25 AC 220 ms
18,304 KB
testcase_26 AC 218 ms
18,432 KB
testcase_27 AC 223 ms
18,432 KB
testcase_28 AC 224 ms
18,560 KB
testcase_29 AC 223 ms
18,432 KB
testcase_30 AC 220 ms
18,304 KB
testcase_31 AC 220 ms
18,432 KB
testcase_32 AC 201 ms
18,560 KB
testcase_33 AC 198 ms
18,432 KB
testcase_34 AC 206 ms
18,432 KB
testcase_35 AC 202 ms
18,432 KB
testcase_36 AC 192 ms
18,560 KB
testcase_37 AC 197 ms
18,304 KB
testcase_38 AC 188 ms
18,432 KB
testcase_39 AC 192 ms
18,432 KB
testcase_40 AC 195 ms
18,432 KB
testcase_41 AC 197 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