結果

問題 No.2310 [Cherry 5th Tune A] Against Regret
ユーザー kotatsugamekotatsugame
提出日時 2023-05-19 22:23:35
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,315 ms / 6,000 ms
コード長 1,615 bytes
コンパイル時間 972 ms
コンパイル使用メモリ 81,440 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-18 03:12:01
合計ジャッジ時間 22,109 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<algorithm>
#include<atcoder/modint>
using namespace std;
using mint=atcoder::modint998244353;
int N;
mint dp[401][401];
mint X[401][401];
mint ep[401];
mint fp[62][2];
mint M[62][62],M2[62][62];
int K;
int A[30],B[30];
mint C[30];
int main()
{
	cin>>N;
	for(int i=0;i<=N;i++)for(int j=0;j<=N;j++)
	{
		int x;cin>>x;
		if(x==998244353)X[i][j]=mint::raw(0);
		else X[i][j]=mint::raw(x);
	}
	for(int s=0;s<N;s++)
	{
		for(int i=s;i<=N;i++)ep[i]=mint::raw(0);
		ep[s]=mint::raw(1);
		for(int i=s;i<N;i++)for(int j=i+1;j<=N;j++)ep[j]+=ep[i]*X[i][j];
		for(int t=s+1;t<=N;t++)dp[s][t]=ep[t];
	}
	int Q;cin>>Q;
	vector<int>vs;
	vs.reserve(62);
	for(;Q--;)
	{
		cin>>K;
		vs.clear();
		vs.push_back(0);
		vs.push_back(N);
		for(int i=0;i<K;i++)
		{
			cin>>A[i]>>B[i];
			vs.push_back(A[i]);
			vs.push_back(B[i]);
			int c;cin>>c;
			if(c==998244353)C[i]=mint::raw(0);
			else C[i]=mint::raw(c);
		}
		sort(vs.begin(),vs.end());
		vs.erase(unique(vs.begin(),vs.end()),vs.end());
		for(int i=0;i<vs.size();i++)for(int j=0;j<vs.size();j++)
		{
			M[i][j]=dp[vs[i]][vs[j]];
			M2[i][j]=mint::raw(0);
		}
		for(int i=0;i<K;i++)
		{
			int a=lower_bound(vs.begin(),vs.end(),A[i])-vs.begin();
			int b=lower_bound(vs.begin(),vs.end(),B[i])-vs.begin();
			M2[a][b]+=C[i];
		}
		for(int i=0;i<vs.size();i++)
		{
			fp[i][0]=fp[i][1]=mint::raw(0);
		}
		fp[0][1]=mint::raw(1);
		for(int i=0;i<vs.size();i++)for(int j=i+1;j<vs.size();j++)
		{
			fp[j][0]+=fp[i][1]*M[i][j];
			fp[j][1]+=(fp[i][0]+fp[i][1])*M2[i][j];
		}
		cout<<(fp[vs.size()-1][0]+fp[vs.size()-1][1]).val()<<"\n";
	}
}
0