結果

問題 No.613 Solitude by the window
コンテスト
ユーザー zimpha
提出日時 2017-12-13 00:45:35
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,140 bytes
コンパイル時間 475 ms
コンパイル使用メモリ 49,408 KB
最終ジャッジ日時 2025-01-05 05:10:59
ジャッジサーバーID
（参考情報） 		judge3 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 14 WA * 7
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:48:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   48 |   scanf("%lld%lld", &n, &m);
      |   ~~~~~^~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

#include <cstdio>
#include <vector>
#include <map>

using int64 = long long;

int64 phi(int64 n) {
  int64 r = n;
  for (int64 i = 2; i * i <= n; ++i) {
    if (n % i == 0) {
      r = r / i * (i  -1);
      while (n % i == 0) n /= i;;
    }
  }
  if (n > 1) r = r / n * (n - 1);
  return r;
}

int64 pow_mod(int64 a, int64 n, int64 mod) {
  int64 r = 1;
  for (; n; n >>= 1) {
    if (n & 1) r = r * a % mod;
    a = a * a % mod;
  }
  return r;
}

int64 calc(int64 n, int64 mod) {
  // (2 + sqrt{3}) ^ (2^n) % mod
  n = pow_mod(2, n, mod - 1);
  int64 a = 1, b = 0;
  int64 c = 2, d = 1;
  for (; n; n >>= 1) {
    if (n & 1) {
      int64 u = (a * c + 3 * b * d) % mod;
      int64 v = (a * d + b * c) % mod;
      a = u, b = v;
    }
    int64 u = (c * c + d * d * 3) % mod;
    int64 v = 2 * c * d % mod;
    c = u, d = v;
  }
  return a;
}

int main() {
  int64 n, m;
  scanf("%lld%lld", &n, &m);
  if (n <= 1000000) {
    int64 x = 2;
    for (int i = 1; i <= n; ++i) {
      x = (x * x + x * 4) % m;
    }
    printf("%lld\n", x);
  } else {
    int64 r = calc(n, m) * 2 % m;
    printf("%lld\n", (r + m - 2) % m);
  }
  return 0;
}
0