結果

問題 No.1533 Don't be Negative!
ユーザー leukocyte
提出日時 2022-07-02 12:26:49
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 5,647 ms / 8,000 ms
コード長 640 bytes
コンパイル時間 1,892 ms
コンパイル使用メモリ 167,292 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-18 13:24:10
合計ジャッジ時間 111,384 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 53
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
const int N=4e4,S=8e4;
const int mod=998244353;
int n,m,k;
int f[S+5],g[S+5];
int c[N+5];
int ans;
int main(){
	scanf("%d %d %d",&n,&m,&k);
	f[0]=1;
	int ms=2*m-(k>0),tmp=1;
	for(int i=1;i<=n;i++){
		int r=min(i,n-i)*m;
		ll s=0;
		for(int j=-m;j<m;j++) s+=f[abs(j)];
		for(int j=0;j<=r;j++){
			s+=f[j+m];
			if(k) g[j]=(s-f[j+k]-f[abs(j-k)])%mod;
			else g[j]=(s-f[j])%mod;
			s-=f[abs(j-m)];
		}
		memcpy(f,g,(r+1)*sizeof(int));
		tmp=1ll*tmp*ms%mod;
		int c=499122177ll*(tmp-f[0])%mod;
		ans=(1ll*ans*ms+1ll*c*(n-i+1))%mod;
	}
	printf("%d\n",(ans+mod)%mod);
	return 0;
}
0