結果

問題 No.362 門松ナンバー
コンテスト
ユーザー yuppe19 😺
提出日時 2016-04-18 14:09:46
言語 C++11(old_compat)
(gcc 12.4.0 + boost 1.89.0)
コンパイル:
g++-12 -O2 -lm -std=gnu++11 -Wuninitialized -DONLINE_JUDGE -include bits/stdc++.h -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 37 ms / 3,000 ms
コード長 2,054 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,722 ms
コンパイル使用メモリ 181,116 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2026-03-08 16:06:15
合計ジャッジ時間 2,732 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:65:15: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   65 |   int t; scanf("%d", &t);
      |          ~~~~~^~~~~~~~~~
main.cpp:67:17: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   67 |     i64 k; scanf("%lld", &k);
      |            ~~~~~^~~~~~~~~~~~

ソースコード

diff #
raw source code 

#include <iostream>
#include <algorithm>
using namespace std;
using i64 = long long;

template<typename T>
class Range{private:struct I{T x;T operator*(){return x;}bool operator!=(I& lhs){return x<lhs.x;}void operator++(){++x;}};I i,n;
public:Range(T n):i({0}),n({n}){}Range(T i,T n):i({i}),n({n}){}I& begin(){return i;}I& end(){return n;}};
using range = Range<i64>;

bool is_kado(int a, int b, int c) {
  return (a!=b&&b!=c&&c!=a) && ((a>b&&b<c) || (a<b&&b>c));
}

i64 g(i64 x) {
  string s = to_string(x);
  int n = s.size();
  vector<vector<vector<vector<vector<i64>>>>> dp(n+1, vector<vector<vector<vector<i64>>>>(2, vector<vector<vector<i64>>>(2, vector<vector<i64>>(11, vector<i64>(11, 0))))); // dp[n+1][2][2][11][11]
  // [][][題意を満たすか][一個前の桁][二個前の桁]
  dp[0][0][1][10][10] = 1;
  for(auto&& i : range(n)) {
    for(auto&& j : range(2)) {
      for(auto&& ok : range(2)) {
        for(auto&& b1 : range(11)) {
          for(auto&& b2 : range(11)) {
            if(!dp[i][j][ok][b1][b2]) { continue; }
            for(auto&& d : range(j?10:s[i]-'0'+1)) {
              int flag = j || d < s[i] - '0',
                  nb1 = (b1==10 && d==0) /* not nuke0 yet */ ? 10 : d,
                  nb2 = b1,
                  nok = ok && (is_kado(b2, b1, d) || b2 == 10);
              if(!nok) { continue; }
              dp[i+1][flag][nok][nb1][nb2] += dp[i][j][ok][b1][b2];
            }
          }
        }
      }
    }
  }
  i64 res = 0;
  for(auto&& j : range(2)) {
    for(auto&& b1 : range(11)) {
      for(auto&& b2 : range(11)) {
        res += dp[n][j][1][b1][b2];
      }
    }
  }
  return res;
}

i64 f(i64 k) {
  k += g(101);
  i64 lo = 0, hi = i64(pow(2, 60));
  while(hi - lo > 10) {
    i64 md = (lo + hi) / 2;
    (g(md) < k ? lo : hi) = md;
  }
  for(auto&& x : range(lo, hi+1)) {
    if(g(x) == k) { return x; }
  }
  throw;
}

int main(void) {
  int t; scanf("%d", &t);
  for(auto&& loop : range(t)) {
    i64 k; scanf("%lld", &k);
    i64 res = f(k);
    printf("%lld\n", res);
  }
  return 0;
}
0