#87408 (C++11) No.362 門松ナンバー

• 問題: No.362 門松ナンバー
• ユーザー: anta
• 提出日時: 2016-04-17 23:01:40
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 3 ms / 3,000 ms
• コード長: 2,254 bytes
• コンパイル時間: 680 ms
• コンパイル使用メモリ: 83,464 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-10-04 10:46:57
• 合計ジャッジ時間: 1,654 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,248 KB
testcase_02	AC	2 ms; 5,248 KB
testcase_03	AC	2 ms; 5,248 KB
testcase_04	AC	2 ms; 5,248 KB
testcase_05	AC	2 ms; 5,248 KB
testcase_06	AC	2 ms; 5,248 KB
testcase_07	AC	2 ms; 5,248 KB
testcase_08	AC	2 ms; 5,248 KB
testcase_09	AC	2 ms; 5,248 KB
testcase_10	AC	3 ms; 5,248 KB
testcase_11	AC	2 ms; 5,248 KB
testcase_12	AC	2 ms; 5,248 KB
testcase_13	AC	2 ms; 5,248 KB
testcase_14	AC	2 ms; 5,248 KB
testcase_15	AC	2 ms; 5,248 KB
testcase_16	AC	2 ms; 5,248 KB
testcase_17	AC	2 ms; 5,248 KB
testcase_18	AC	2 ms; 5,248 KB
testcase_19	AC	2 ms; 5,248 KB
testcase_20	AC	2 ms; 5,248 KB
testcase_21	AC	2 ms; 5,248 KB

コンパイルメッセージ

main.cpp: In function ‘int main()’:
main.cpp:77:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   77 |         scanf("%d", &T);
      |         ~~~~~^~~~~~~~~~
main.cpp:80:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   80 |                 scanf("%lld", &K);
      |                 ~~~~~^~~~~~~~~~~~

ソースコード

#include <string>
#include <vector>
#include <algorithm>
#include <numeric>
#include <set>
#include <map>
#include <queue>
#include <iostream>
#include <sstream>
#include <cstdio>
#include <cmath>
#include <ctime>
#include <cstring>
#include <cctype>
#include <cassert>
#include <limits>
#include <functional>
#define rep(i,n) for(int (i)=0;(i)<(int)(n);++(i))
#define rer(i,l,u) for(int (i)=(int)(l);(i)<=(int)(u);++(i))
#define reu(i,l,u) for(int (i)=(int)(l);(i)<(int)(u);++(i))
#if defined(_MSC_VER) || __cplusplus > 199711L
#define aut(r,v) auto r = (v)
#else
#define aut(r,v) __typeof(v) r = (v)
#endif
#define each(it,o) for(aut(it, (o).begin()); it != (o).end(); ++ it)
#define all(o) (o).begin(), (o).end()
#define pb(x) push_back(x)
#define mp(x,y) make_pair((x),(y))
#define mset(m,v) memset(m,v,sizeof(m))
#define INF 0x3f3f3f3f
#define INFL 0x3f3f3f3f3f3f3f3fLL
using namespace std;
typedef vector<int> vi; typedef pair<int, int> pii; typedef vector<pair<int, int> > vpii; typedef long long ll;
template<typename T, typename U> inline void amin(T &x, U y) { if(y < x) x = y; }
template<typename T, typename U> inline void amax(T &x, U y) { if(x < y) x = y; }

bool isok(int a, int b, int c) {
	return a != b && a != c && b != c && (min({ a, b, c }) == b || max({ a, b, c }) == b);
}

ll memo[30][11][11][2];
ll rec(int i, int pd, int pp, bool ppp) {
	ll &r = memo[i + 1][pd + 1][pp + 1][ppp];
	if(r != -1) return r;
	if(i == -1) return r = ppp ? 1 : 0;
	r = 0;
	rep(d, 10) {
		if(pp == -1 || isok(pp, pd, d)) {
			r += rec(i - 1, pd == -1 && d == 0 ? -1 : d, pd, pp != -1);
			if(r > INFL) r = INFL;
		}
	}
	return r;
}

string ans;
void solve(ll k, int i, int pd, int pp, bool ppp) {
	if(i == -1)
		return;
	rep(d, 10) {
		if(pp == -1 || isok(pp, pd, d)) {
			int nd = pd == -1 && d == 0 ? -1 : d;
			ll t = rec(i - 1, nd, pd, pp != -1);
			if(k < t) {
				ans += char('0' + d);
				return solve(k, i - 1, nd, pd, pp != -1);
			}
			k -= t;
		}
	}
}

int main() {
	mset(memo, -1);
	int T;
	scanf("%d", &T);
	for(int ii = 0; ii < T; ++ ii) {
		long long K;
		scanf("%lld", &K);
		ans.clear();
		solve(K - 1, 28, -1, -1, false);
		{
			int i = 0;
			while(ans[i] == '0') ++ i;
			ans = ans.substr(i);
		}
		puts(ans.c_str());
	}
	return 0;
}