結果

問題 No.2489 X and Xor 2
ユーザー SSRSSSRS
提出日時 2023-09-29 21:46:50
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 325 ms / 2,000 ms
コード長 2,576 bytes
コンパイル時間 2,062 ms
コンパイル使用メモリ 208,756 KB
実行使用メモリ 4,380 KB
最終ジャッジ日時 2023-09-29 22:08:52
合計ジャッジ時間 12,390 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 48 ms
4,376 KB
testcase_01 AC 297 ms
4,380 KB
testcase_02 AC 48 ms
4,376 KB
testcase_03 AC 52 ms
4,376 KB
testcase_04 AC 80 ms
4,376 KB
testcase_05 AC 48 ms
4,380 KB
testcase_06 AC 301 ms
4,380 KB
testcase_07 AC 325 ms
4,380 KB
testcase_08 AC 277 ms
4,380 KB
testcase_09 AC 293 ms
4,376 KB
testcase_10 AC 274 ms
4,380 KB
testcase_11 AC 245 ms
4,380 KB
testcase_12 AC 268 ms
4,376 KB
testcase_13 AC 270 ms
4,380 KB
testcase_14 AC 286 ms
4,380 KB
testcase_15 AC 286 ms
4,380 KB
testcase_16 AC 249 ms
4,376 KB
testcase_17 AC 313 ms
4,376 KB
testcase_18 AC 273 ms
4,376 KB
testcase_19 AC 305 ms
4,380 KB
testcase_20 AC 280 ms
4,380 KB
testcase_21 AC 241 ms
4,380 KB
testcase_22 AC 264 ms
4,380 KB
testcase_23 AC 252 ms
4,376 KB
testcase_24 AC 275 ms
4,380 KB
testcase_25 AC 279 ms
4,380 KB
testcase_26 AC 245 ms
4,376 KB
testcase_27 AC 325 ms
4,376 KB
testcase_28 AC 274 ms
4,376 KB
testcase_29 AC 285 ms
4,380 KB
testcase_30 AC 268 ms
4,376 KB
testcase_31 AC 244 ms
4,380 KB
testcase_32 AC 276 ms
4,376 KB
testcase_33 AC 247 ms
4,376 KB
testcase_34 AC 270 ms
4,376 KB
testcase_35 AC 278 ms
4,376 KB
testcase_36 AC 247 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;  
const long long MOD = 998244353;
const int LOG = 60;
vector<vector<long long>> matmul(vector<vector<long long>> A, vector<vector<long long>> B){
  int N = A.size();
  vector<vector<long long>> C(N, vector<long long>(N, 0));
  for (int i = 0; i < N; i++){
    for (int j = 0; j < N; j++){
      for (int k = 0; k < N; k++){
        C[i][k] += A[i][j] * B[j][k];
        C[i][k] %= MOD;
      }
    }
  }
  return C;
}
vector<vector<long long>> matpow(vector<vector<long long>> A, long long K){
  int N = A.size();
  vector<vector<long long>> ans(N, vector<long long>(N, 0));
  for (int i = 0; i < N; i++){
    ans[i][i] = 1;
  }
  while (K > 0){
    if (K % 2 == 1){
      ans = matmul(ans, A);
    }
    A = matmul(A, A);
    K /= 2;
  }
  return ans;
}
int main(){
  long long N, M;
  cin >> N >> M;
  vector<long long> POW(LOG);
  for (int i = 0; i < LOG; i++){
    POW[i] = (long long) 1 << i;
  }
  vector<vector<long long>> S(LOG, vector<long long>(2, 0));
  for (int i = 0; i < LOG; i++){
    long long b = POW[i];
    S[i][0] = M / (b * 2) * b + min(M % (b * 2), b);
    S[i][1] =  M / (b * 2) * b + max(M % (b * 2) - b, (long long) 0);
    S[i][0] %= MOD;
    S[i][1] %= MOD;
  }
  vector<vector<long long>> mat(LOG * 2, vector<long long>(LOG * 2, 0));
  for (int i = 0; i < LOG; i++){
    for (int j = 0; j < 2; j++){
      for (int k = 0; k < LOG; k++){
        for (int l = 0; l < 2; l++){
          vector<vector<long long>> dp(LOG + 1, vector<long long>(2, 0));
          dp[LOG][0] = 1;
          for (int x = LOG - 1; x >= 0; x--){
            for (int y = 0; y < 2; y++){
              for (int d = 0; d < 2; d++){
                if (!(x == i && d == j) && !(x == k && d != l)){
                  if (!((M >> x & 1) == 0 && y == 0 && d == 1)){
                    int y2 = y;
                    if ((M >> x & 1) == 1 && d == 0){
                      y2 = 1;
                    }
                    dp[x][y2] += dp[x + 1][y];
                  }
                }
              }
            }
          }
          long long cnt = dp[0][1] % MOD;
          mat[i * 2 + j][k * 2 + l] = cnt * (POW[k] % MOD) % MOD;
        }
      }
    }
  }
  mat = matpow(mat, N - 2);
  long long ans = 0;
  for (int i = 0; i < LOG; i++){
    for (int j = 0; j < 2; j++){
      for (int k = 0; k < LOG; k++){
        for (int l = 0; l < 2; l++){
          ans += S[i][j] * (POW[i] % MOD) % MOD * mat[i * 2 + j][k * 2 + l] % MOD * S[k][l ^ 1] % MOD;
          ans %= MOD;
        }
      }
    }
  }
  cout << ans << endl;
}
0