結果

問題 No.2807 Have Another Go (Easy)
ユーザー 沙耶花
提出日時 2024-07-12 21:51:27
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 23 ms / 3,000 ms
コード長 801 bytes
コンパイル時間 4,340 ms
コンパイル使用メモリ 251,824 KB
最終ジャッジ日時 2025-02-22 03:51:22
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <stdio.h>
#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
using namespace std;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf32 1000000001
#define Inf64 1000000000000000001

int main(){

	int N,M,K;
	cin>>N>>M>>K;
	
	vector<mint> dp(N*M+1);
	dp[0] = 1;
	rep(i,N*M+1){
		for(int j=1;j<=6;j++){
			if(i+j>N*M) continue;
			dp[i+j] += dp[i];
		}
	}
	rep(_,K){
		int t;
		cin>>t;
		mint ans = 0;
		for(int i=N*M-6;i<N*M;i++){
			for(int j=1;j<=6;j++){
				if(i+j>=N*M){
					{
						int X = t;
						if(X<=i){
							ans += dp[X] * dp[i-X];
						}
						X = N+t;
						if(X<=i){
							ans += dp[X] * dp[i-X];
							ans -= dp[t] * dp[N] * dp[i-X];
						}
					}
					
				}
			}
		}
		cout<<ans.val()<<endl;
	}
	
	return 0;
}
0