#438057 (C++14) No.362 門松ナンバー

• 問題: No.362 門松ナンバー
• ユーザー: kotatsugame
• 提出日時: 2020-03-05 16:38:48
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 147 ms / 3,000 ms
• コード長: 1,078 bytes
• コンパイル時間: 712 ms
• コンパイル使用メモリ: 75,264 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-10-14 01:19:34
• 合計ジャッジ時間: 3,713 ms
• ジャッジサーバーID （参考情報）: judge3 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	37 ms; 5,248 KB
testcase_01	AC	31 ms; 5,248 KB
testcase_02	AC	30 ms; 5,248 KB
testcase_03	AC	16 ms; 5,248 KB
testcase_04	AC	10 ms; 5,248 KB
testcase_05	AC	49 ms; 5,248 KB
testcase_06	AC	90 ms; 5,248 KB
testcase_07	AC	52 ms; 5,248 KB
testcase_08	AC	70 ms; 5,248 KB
testcase_09	AC	114 ms; 5,248 KB
testcase_10	AC	145 ms; 5,248 KB
testcase_11	AC	141 ms; 5,248 KB
testcase_12	AC	142 ms; 5,248 KB
testcase_13	AC	144 ms; 5,248 KB
testcase_14	AC	143 ms; 5,248 KB
testcase_15	AC	143 ms; 5,248 KB
testcase_16	AC	139 ms; 5,248 KB
testcase_17	AC	128 ms; 5,248 KB
testcase_18	AC	146 ms; 5,248 KB
testcase_19	AC	73 ms; 5,248 KB
testcase_20	AC	147 ms; 5,248 KB
testcase_21	AC	17 ms; 5,248 KB

コンパイルメッセージ

main.cpp:46:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   46 | main()
      | ^~~~

ソースコード

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
long count(long L)
{
	vector<int>S;
	while(L)
	{
		S.push_back(L%10);
		L/=10;
	}
	reverse(S.begin(),S.end());
	long dp[20][2][11][11][11]={};
	dp[0][0][10][10][10]=1;
	for(int i=0;i<S.size();i++)for(int j=0;j<2;j++)
	{
		int lim=j?9:S[i];
		for(int a=0;a<=10;a++)for(int b=0;b<=10;b++)for(int c=0;c<=10;c++)
		{
			if(dp[i][j][a][b][c]==0)continue;
			for(int d=c==10;d<=lim;d++)
			{
				if(b<10)
				{
					if(b!=c&&c!=d&&d!=b&&(b<c&&d<c||b>c&&d>c))
					{
						dp[i+1][j||d<lim][b][c][d]+=dp[i][j][a][b][c];
					}
				}
				else
				{
					dp[i+1][j||d<lim][b][c][d]+=dp[i][j][a][b][c];
				}
			}
			if(c==10)dp[i+1][1][b][c][10]+=dp[i][j][a][b][c];
		}
	}
	long ret=0;
	for(int a=0;a<10;a++)for(int b=0;b<10;b++)for(int c=0;c<10;c++)
	{
		ret+=dp[S.size()][0][a][b][c]+dp[S.size()][1][a][b][c];
	}
	return ret;
}
main()
{
	int T;cin>>T;
	for(;T--;)
	{
		long K;
		cin>>K;
		long L=100,R=1e15;
		while(R-L>1)
		{
			long M=(L+R)/2;
			if(count(M)>=K)R=M;
			else L=M;
		}
		cout<<R<<endl;
	}
}