結果

問題 No.1549 [Cherry 2nd Tune] BANning Tuple
コンテスト
ユーザー kotatsugame
提出日時 2021-06-11 23:59:47
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2,578 ms / 4,000 ms
コード長 1,493 bytes
コンパイル時間 1,496 ms
コンパイル使用メモリ 96,916 KB
実行使用メモリ 6,992 KB
最終ジャッジ日時 2024-12-15 04:04:45
合計ジャッジ時間 32,149 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 20
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:11:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   11 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
#include<map>
#include<set>
#include<vector>
#include<atcoder/modint>
using namespace std;
using mint=atcoder::modint998244353;
long N;
int Q;
mint C[101][3001];
main()
{
	cin>>N>>Q;
	for(int i=0;i<=100;i++)
	{
		mint x=1,y=1;
		for(int j=0;j<=3000;j++)
		{
			C[i][j]=x/y;
			x*=N-i+j+1;
			y*=j+1;
		}
	}
	map<long,set<pair<int,int> > >mp;
	for(int q=0;q<Q;q++)
	{
		long K;int A,B,S,T;
		cin>>K>>A>>B>>S>>T;
		B++;
		{
			set<pair<int,int> >now;
			for(pair<int,int>p:mp[K])
			{
				if(p.second<A)now.insert(p);
				else if(B<p.first)now.insert(p);
				else
				{
					if(A>p.first)A=p.first;
					if(B<p.second)B=p.second;
				}
			}
			now.insert(make_pair(A,B));
			mp[K].swap(now);
		}
		vector<vector<mint> >dp(mp.size()+1,vector<mint>(T+1));
		dp[0][0]=1;
		int cnt=0;
		for(auto&&S:mp)
		{
			vector<vector<mint> >ep(cnt+2,vector<mint>(T+1));
			for(int i=cnt;i>=0;i--)
			{
				for(pair<int,int>p:S.second)
				{
					int a=p.first,b=p.second;
					for(int j=0;j+a<=T;j++)
					{
						ep[i+1][j+a]+=dp[i][j];
						int r=j+b;
						if(r<=T)ep[i+1][r]-=dp[i][j];
					}
				}
			}
			cnt++;
			for(int i=0;i<=cnt;i++)
			{
				for(int j=1;j<=T;j++)ep[i][j]+=ep[i][j-1];
				for(int j=0;j<=T;j++)dp[i][j]+=ep[i][j];
			}
		}
		mint ans=0;
		for(int i=0;i<=cnt;i++)
		{
			for(int j=0;j<=T;j++)
			{
				mint f=dp[i][j];
				if(i&1)f=-f;
				mint val=C[i][T-j];
				if(S>j)
				{
					val-=C[i][S-j-1];
				}
				ans+=f*val;
			}
		}
		cout<<ans.val()<<endl;
	}
}
0