結果

問題 No.685 Logical Operations
コンテスト
ユーザー yuppe19 😺
提出日時 2018-11-15 14:57:09
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 2,162 bytes
コンパイル時間 1,693 ms
コンパイル使用メモリ 97,328 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-24 13:55:38
合計ジャッジ時間 2,482 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bitset>
#include <cmath>
#include <iostream>
#include <vector>
using namespace std;
using i64 = int64_t;

constexpr int mod = static_cast<int>(powl(10, 9)) + 7;

string to_bin(const i64 a) {
  if(a == 0) { return "0"; }
  string s = bitset<64>(a).to_string();
  int p = static_cast<int>(s.find('1'));
  return s.substr(p);
}

int cmp(int x, int y) {
  if(x <  y) { return 0; }
  if(x == y) { return 1; }
  return 2;
}

int mycmp(int x, int y, int val) {
  int res = cmp(x, y);
  if(res == 1) { res = val; }
  return res;
}

i64 f(i64 x) {
  string s = to_bin(x);
  int n = static_cast<int>(s.size());
  // dp[yがNより未満/丁度/超過][xがyより未満/丁度/超過][ANDがXORより未満/丁度/超過][XORがORより未満/丁度/超過] := パターン数
  vector<vector<vector<vector<i64>>>> dp(3, vector<vector<vector<i64>>>(3, vector<vector<i64>>(3, vector<i64>(3, 0))));
  dp[1][1][1][1] = 1;
  for(int i=n-1; i>=0; --i) {
    vector<vector<vector<vector<i64>>>> ndp(3, vector<vector<vector<i64>>>(3, vector<vector<i64>>(3, vector<i64>(3, 0))));
    for(int flag0=0; flag0<3; ++flag0) {
      for(int flag1=0; flag1<3; ++flag1) {
        for(int and_xor=0; and_xor<3; ++and_xor) {
          for(int xor_or=0; xor_or<3; ++xor_or) {
            if(!dp[flag0][flag1][and_xor][xor_or]) { continue; }
            for(int yi=0; yi<2; ++yi) {
              int nflag0 = mycmp(yi, s[i]-'0', flag0);
              for(int xi=0; xi<2; ++xi) {
                int nflag1 = mycmp(xi, yi, flag1),
                    n_and_xor = mycmp(xi & yi, xi ^ yi, and_xor),
                    n_xor_or  = mycmp(xi ^ yi, xi | yi, xor_or);
                ndp[nflag0][nflag1][n_and_xor][n_xor_or] += dp[flag0][flag1][and_xor][xor_or];
                ndp[nflag0][nflag1][n_and_xor][n_xor_or] %= mod;
              }
            }
          }
        }
      }
    }
    dp = ndp;
  }
  i64 res = 0;
  for(int flag0=0; flag0<2; ++flag0) {
    for(int flag1=0; flag1<2; ++flag1) {
      res += dp[flag0][flag1][0][0];
      res %= mod;
    }
  }
  return res;
}

int main(void) {
  i64 x; scanf("%ld", &x);
  i64 res = f(x);
  printf("%ld\n", res);
  return 0;
}
0