結果

問題 No.362 門松ナンバー
ユーザー yuppe19 😺yuppe19 😺
提出日時 2016-04-18 14:09:46
言語 C++11
(gcc 11.4.0)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 2,054 bytes
コンパイル時間 521 ms
コンパイル使用メモリ 52,144 KB
最終ジャッジ日時 2024-11-14 19:41:57
合計ジャッジ時間 1,069 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

コンパイルメッセージ
main.cpp: In function ‘i64 g(i64)’:
main.cpp:18:3: error: ‘vector’ was not declared in this scope
   18 |   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]
      |   ^~~~~~
main.cpp:3:1: note: ‘std::vector’ is defined in header ‘<vector>’; did you forget to ‘#include <vector>’?
    2 | #include <algorithm>
  +++ |+#include <vector>
    3 | using namespace std;
main.cpp:18:41: error: expected primary-expression before ‘>>’ token
   18 |   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]
      |                                         ^~
main.cpp:18:43: error: expected primary-expression before ‘>>’ token
   18 |   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]
      |                                           ^~
main.cpp:18:45: error: expected primary-expression before ‘>’ token
   18 |   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]
      |                                             ^
main.cpp:18:86: error: expected primary-expression before ‘>>’ token
   18 |   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]
      |                                                                                      ^~
main.cpp:18:88: error: expected primary-expression before ‘>>’ token
   18 |   vector<vector<vector<v

ソースコード

diff # 

#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