結果

問題 No.1513 simple 門松列 problem
コンテスト
ユーザー tails
提出日時 2021-05-23 01:07:04
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 13 ms / 3,000 ms
コード長 1,037 bytes
コンパイル時間 571 ms
コンパイル使用メモリ 36,292 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-11 01:23:39
合計ジャッジ時間 1,523 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c:16:1: warning: return type defaults to 'int' [-Wimplicit-int]
   16 | main(){
      | ^~~~
main.c: In function 'main':
main.c:64:9: warning: implicit declaration of function 'write' [-Wimplicit-function-declaration]
   64 |         write(1,wp,wbuf+sizeof wbuf-wp);
      |         ^~~~~
main.c:65:9: warning: implicit declaration of function '_exit'; did you mean '_Exit'? [-Wimplicit-function-declaration]
   65 |         _exit(0);
      |         ^~~~~
      |         _Exit

ソースコード

diff # 

#pragma GCC optimize("Ofast")
#pragma GCC target("avx2")

char*mmap();
#define rd(v) int v=0;{int _c;while(_c=*rp++-48,_c>=0)v=v*10+_c;}

#define wt_rev(v) {long _z=v;do*--wp=_z%10+48;while(_z/=10);}

#define rep(v,e) for(int v=0;v<e;++v)
#define rep3(v,s,e) for(int v=s;v<e;++v)

long dp1[2][200][200];

#define MD 998244353

main(){
	char*rp=mmap(0l,1l<<12,1,2,0,0ll);
	rd(n);
	rd(k);

	rep(b,k)rep(a,b){
		dp1[0][a][b]=1;
	}
	rep(i,n-2){
		if(!(i&1)){
			rep(b,k){
				long v=0;
				rep(a,b){
					v+=dp1[0][a][b];
				}
				v%=MD;
				rep(c,b){
					dp1[1][b][c]=v-dp1[0][c][b];
				}
			}
		}else{
			rep(b,k){
				long v=0;
				rep3(a,b+1,k){
					v+=dp1[1][a][b];
				}
				v%=MD;
				rep3(c,b+1,k){
					dp1[0][b][c]=v-dp1[1][c][b];
				}
			}
		}
	}

	long ans1=0;
	rep(i,k)rep(j,k){
		ans1+=dp1[n&1][i][j];
	}
	ans1%=MD;
	ans1+=ans1<0?MD:0;
	long ans2=ans1*(k-1)*n%MD;
	ans1*=2;
	ans1-=ans1<MD?0:MD;

	char wbuf[64],*wp=wbuf+sizeof wbuf;
	wt_rev(ans2);
	*--wp=' ';
	wt_rev(ans1);
	write(1,wp,wbuf+sizeof wbuf-wp);
	_exit(0);
}
0