結果

問題 No.1548 [Cherry 2nd Tune B] 貴方と私とサイクルとモーメント
ユーザー kotatsugamekotatsugame
提出日時 2021-06-11 23:05:31
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 201 ms / 4,500 ms
コード長 1,534 bytes
コンパイル時間 883 ms
コンパイル使用メモリ 82,220 KB
実行使用メモリ 18,412 KB
最終ジャッジ日時 2023-08-21 13:55:53
合計ジャッジ時間 9,188 ms
ジャッジサーバーID
(参考情報)
judge13 / judge15
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 145 ms
10,824 KB
testcase_03 AC 46 ms
6,988 KB
testcase_04 AC 138 ms
17,436 KB
testcase_05 AC 14 ms
4,380 KB
testcase_06 AC 106 ms
10,048 KB
testcase_07 AC 130 ms
17,944 KB
testcase_08 AC 148 ms
10,508 KB
testcase_09 AC 107 ms
7,000 KB
testcase_10 AC 105 ms
17,392 KB
testcase_11 AC 120 ms
17,196 KB
testcase_12 AC 76 ms
5,008 KB
testcase_13 AC 77 ms
17,284 KB
testcase_14 AC 111 ms
6,788 KB
testcase_15 AC 89 ms
18,396 KB
testcase_16 AC 122 ms
10,896 KB
testcase_17 AC 151 ms
17,016 KB
testcase_18 AC 127 ms
18,052 KB
testcase_19 AC 57 ms
6,668 KB
testcase_20 AC 97 ms
17,556 KB
testcase_21 AC 98 ms
9,968 KB
testcase_22 AC 199 ms
18,412 KB
testcase_23 AC 201 ms
18,212 KB
testcase_24 AC 201 ms
18,372 KB
testcase_25 AC 200 ms
18,380 KB
testcase_26 AC 201 ms
18,280 KB
testcase_27 AC 197 ms
18,232 KB
testcase_28 AC 198 ms
18,384 KB
testcase_29 AC 196 ms
18,356 KB
testcase_30 AC 199 ms
18,384 KB
testcase_31 AC 200 ms
18,348 KB
testcase_32 AC 183 ms
18,224 KB
testcase_33 AC 183 ms
18,228 KB
testcase_34 AC 181 ms
18,212 KB
testcase_35 AC 184 ms
18,412 KB
testcase_36 AC 180 ms
18,084 KB
testcase_37 AC 178 ms
18,380 KB
testcase_38 AC 173 ms
18,344 KB
testcase_39 AC 179 ms
18,372 KB
testcase_40 AC 178 ms
18,368 KB
testcase_41 AC 179 ms
18,280 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:32:1: 警告: ISO C++ では型の無い ‘main’ の宣言を禁止しています [-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