結果

問題 No.1513 simple 門松列 problem
コンテスト
ユーザー kotatsugame
提出日時 2021-05-21 21:38:22
言語 C++14
(gcc 13.3.0 + boost 1.87.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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます
コンパイルメッセージ
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