結果

問題 No.613 Solitude by the window
ユーザー zimphazimpha
提出日時 2017-12-13 01:13:27
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
MLE  
実行時間 -
コード長 1,903 bytes
コンパイル時間 626 ms
コンパイル使用メモリ 53,632 KB
実行使用メモリ 813,568 KB
最終ジャッジ日時 2024-05-08 05:45:49
合計ジャッジ時間 4,269 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 562 ms
494,592 KB
testcase_06 MLE -
testcase_07 MLE -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:84:15: warning: 'extra' may be used uninitialized [-Wmaybe-uninitialized]
   84 |     n %= (end - extra);
      |          ~~~~~^~~~~~~~
main.cpp:60:22: note: 'extra' was declared here
   60 |     int64 x = 2 % m, extra, target = -1, end;
      |                      ^~~~~

ソースコード

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;
}

bool mark[1200000000];

int main() {
  int64 n, m;
  scanf("%lld%lld", &n, &m);
  if (m == 2) puts("0");
  else if (n <= 1000000) {
    int64 x = 2;
    for (int i = 1; i <= n; ++i) {
      x = (x * x + x * 4) % m;
    }
    printf("%lld\n", x);
  } else if (m == 3 || pow_mod(3, m / 2, m) == m - 1) {
    mark[2 % m] = 1;
    int64 x = 2 % m, extra, target = -1, end;
    for (int i = 1; i <= n; ++i) {
      x = (x * x + 4 * x) % m;
      if (mark[x]) {
        target = x;
        end = i;
        break;
      }
      mark[x] = 1;
    }
    if (target == -1) {
      printf("%lld\n", x);
      return 0;
    }
    x = 2 % m;
    for (int i = 0; i < n; ++i) {
      if (x == target) {
        extra = i;
        break;
      }
      x = (x * x + 4 * x) % m;
    }
    //printf("%lld %lld %lld\n", extra, end, target);
    n -= extra;
    n %= (end - extra);
    for (int i = 0; i < n; ++i) {
      x = (x * x + x * 4) % m;
    }
    printf("%lld\n", x);
  } else {
    int64 r = calc(n, m) * 2 % m + m - 2;
    printf("%lld\n", r % m);
  }
  return 0;
}
0