結果

問題 No.1513 simple 門松列 problem
ユーザー kotatsugamekotatsugame
提出日時 2021-05-21 21:38:22
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 106 ms / 3,000 ms
コード長 1,434 bytes
コンパイル時間 694 ms
コンパイル使用メモリ 71,348 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-10 08:10:07
合計ジャッジ時間 1,971 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
6,816 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 106 ms
6,816 KB
testcase_03 AC 2 ms
6,816 KB
testcase_04 AC 2 ms
6,816 KB
testcase_05 AC 2 ms
6,816 KB
testcase_06 AC 2 ms
6,820 KB
testcase_07 AC 2 ms
6,816 KB
testcase_08 AC 3 ms
6,816 KB
testcase_09 AC 2 ms
6,820 KB
testcase_10 AC 2 ms
6,816 KB
testcase_11 AC 2 ms
6,816 KB
testcase_12 AC 2 ms
6,820 KB
testcase_13 AC 2 ms
6,820 KB
testcase_14 AC 105 ms
6,820 KB
testcase_15 AC 104 ms
6,816 KB
testcase_16 AC 105 ms
6,816 KB
testcase_17 AC 67 ms
6,816 KB
testcase_18 AC 23 ms
6,820 KB
testcase_19 AC 3 ms
6,816 KB
testcase_20 AC 15 ms
6,816 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:8:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
    8 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
#include<atcoder/modint>
using namespace std;
using mint=atcoder::modint998244353;
int N,K;
mint dp[2][200][200];
mint val[2][200][200];
main()
{
	cin>>N>>K;
	int now=0;
	for(int i=0;i<K;i++)for(int j=0;j<K;j++)if(i!=j)
	{
		dp[now][i][j]=1;
		val[now][i][j]=i+j;
	}
	for(int i=2;i<N;i++)
	{
		int nxt=1-now;
		for(int i=0;i<K;i++)for(int j=0;j<K;j++)
		{
			dp[nxt][i][j]=0;
			val[nxt][i][j]=0;
		}
		for(int i=0;i<K;i++)for(int j=0;j<K;j++)if(i!=j)
		{
			if(i<j)
			{
				dp[nxt][j][0]+=dp[now][i][j];
				val[nxt][j][0]+=val[now][i][j];
				dp[nxt][j][i]-=dp[now][i][j];
				val[nxt][j][i]-=val[now][i][j];
				dp[nxt][j][i+1]+=dp[now][i][j];
				val[nxt][j][i+1]+=val[now][i][j];
				dp[nxt][j][j]-=dp[now][i][j];
				val[nxt][j][j]-=val[now][i][j];
			}
			else
			{
				dp[nxt][j][j+1]+=dp[now][i][j];
				val[nxt][j][j+1]+=val[now][i][j];
				dp[nxt][j][i]-=dp[now][i][j];
				val[nxt][j][i]-=val[now][i][j];
				if(i<K-1)
				{
					dp[nxt][j][i+1]+=dp[now][i][j];
					val[nxt][j][i+1]+=val[now][i][j];
				}
			}
		}
		for(int i=0;i<K;i++)
		{
			for(int j=0;j<K-1;j++)
			{
				dp[nxt][i][j+1]+=dp[nxt][i][j];
				val[nxt][i][j+1]+=val[nxt][i][j];
			}
			for(int j=0;j<K;j++)
			{
				val[nxt][i][j]+=dp[nxt][i][j]*j;
			}
		}
		now=nxt;
	}
	mint ans1=0,ans2=0;
	for(int i=0;i<K;i++)for(int j=0;j<K;j++)
	{
		ans1+=dp[now][i][j];
		ans2+=val[now][i][j];
	}
	cout<<ans1.val()<<" "<<ans2.val()<<endl;
}
0