結果

問題 No.997 Jumping Kangaroo
ユーザー kotatsugamekotatsugame
提出日時 2020-02-21 22:55:41
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,353 ms / 2,000 ms
コード長 2,018 bytes
コンパイル時間 857 ms
コンパイル使用メモリ 76,536 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-04-17 08:02:14
合計ジャッジ時間 7,137 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 825 ms
5,376 KB
testcase_03 AC 5 ms
5,376 KB
testcase_04 AC 3 ms
5,376 KB
testcase_05 AC 3 ms
5,376 KB
testcase_06 AC 4 ms
5,376 KB
testcase_07 AC 4 ms
5,376 KB
testcase_08 AC 5 ms
5,376 KB
testcase_09 AC 5 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 1 ms
5,376 KB
testcase_13 AC 88 ms
5,376 KB
testcase_14 AC 435 ms
5,376 KB
testcase_15 AC 46 ms
5,376 KB
testcase_16 AC 996 ms
5,376 KB
testcase_17 AC 3 ms
5,376 KB
testcase_18 AC 399 ms
5,248 KB
testcase_19 AC 72 ms
5,376 KB
testcase_20 AC 595 ms
5,376 KB
testcase_21 AC 4 ms
5,376 KB
testcase_22 AC 597 ms
5,376 KB
testcase_23 AC 1,353 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 AC 1 ms
5,376 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:14:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   14 | main()
      | ^~~~

ソースコード

diff # 

#pragma GCC optimize ("O3")
#pragma GCC target ("avx")
#include<iostream>
using namespace std;
int W,N;
long K;
int A[99];
long mod=1e9+7;
int D[160][160];
long X[2][160];
int E[160][160];
long tmp[160][160];
long ans[160];
main()
{
	cin>>N>>W>>K;
	for(int i=0;i<N;i++)cin>>A[i];
	for(int k=0;k<2;k++)for(int i=0;i<2*W;i++)
	{
		for(int j=0;j<2;j++)for(int l=0;l<4*W;l++)X[j][l]=0;
		X[k][i]=1;
		for(int j=i;j<2*W;j++)for(int l=0;l<2;l++)
		{
			for(int I=0;I<N;I++)
			{
				if(j+A[I]<4*W)
				{
					int nl=l;
					nl+=(j+A[I])/W-j/W;
					if((j+A[I])%W==0)
					{
						if(nl<=2)nl=0;
						else continue;
					}
					if(nl<=1)(X[nl][j+A[I]]+=X[l][j])%=mod;
				}
			}
		}
		for(int j=0;j<2;j++)for(int l=0;l<2*W;l++)
		{
			D[j*2*W+l][k*2*W+i]=X[j][l+2*W];
		}
	}
	for(int i=0;i<4*W;i++)E[i][i]=1;
	long KK=K/2;
	while(KK>0)
	{
		if(KK&1)
		{
			for(int i=0;i<4*W;i++)for(int j=0;j<4*W;j++)tmp[i][j]=0;
			for(int i=0;i<4*W;i++)for(int j=0;j<4*W;j++)
			{
				for(int k=0;k<4*W;k+=4)
				{
					tmp[i][j]+=(long)E[i][k]*D[k][j];
					tmp[i][j]+=(long)E[i][k+1]*D[k+1][j];
					tmp[i][j]+=(long)E[i][k+2]*D[k+2][j];
					tmp[i][j]+=(long)E[i][k+3]*D[k+3][j];
					tmp[i][j]%=mod;
				}
			}
			for(int i=0;i<4*W;i++)for(int j=0;j<4*W;j++)E[i][j]=tmp[i][j];
		}
		KK>>=1;
		for(int i=0;i<4*W;i++)for(int j=0;j<4*W;j++)tmp[i][j]=0;
		for(int i=0;i<4*W;i++)for(int j=0;j<4*W;j++)
		{
			for(int k=0;k<4*W;k+=4)
			{
				tmp[i][j]+=(long)D[i][k]*D[k][j];
				tmp[i][j]+=(long)D[i][k+1]*D[k+1][j];
				tmp[i][j]+=(long)D[i][k+2]*D[k+2][j];
				tmp[i][j]+=(long)D[i][k+3]*D[k+3][j];
				tmp[i][j]%=mod;
			}
		}
		for(int i=0;i<4*W;i++)for(int j=0;j<4*W;j++)D[i][j]=tmp[i][j];
	}
	for(int i=0;i<4*W;i++)ans[i]=E[i][0];
	for(int j=0;j<2*W;j++)for(int i=0;i<2;i++)
	{
		for(int I=0;I<N;I++)
		{
			if(j+A[I]<2*W)
			{
				int nl=i;
				nl+=(j+A[I])/W-j/W;
				if((j+A[I])%W==0)
				{
					if(nl<=2)nl=0;
					else continue;
				}
				if(nl<2)(ans[nl*2*W+j+A[I]]+=ans[i*2*W+j])%=mod;
			}
		}
	}
	cout<<ans[K%2*W]<<endl;
}
0