結果

問題 No.737 PopCount
ユーザー 0x19f0x19f
提出日時 2018-09-28 23:02:30
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 3 ms / 1,000 ms
コード長 1,823 bytes
コンパイル時間 2,650 ms
コンパイル使用メモリ 181,908 KB
実行使用メモリ 4,380 KB
最終ジャッジ日時 2023-08-02 12:01:33
合計ジャッジ時間 2,870 ms
ジャッジサーバーID
(参考情報)
judge11 / judge13
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,376 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 1 ms
4,376 KB
testcase_04 AC 2 ms
4,376 KB
testcase_05 AC 2 ms
4,376 KB
testcase_06 AC 3 ms
4,380 KB
testcase_07 AC 2 ms
4,380 KB
testcase_08 AC 3 ms
4,376 KB
testcase_09 AC 3 ms
4,376 KB
testcase_10 AC 2 ms
4,376 KB
testcase_11 AC 1 ms
4,376 KB
testcase_12 AC 2 ms
4,376 KB
testcase_13 AC 3 ms
4,380 KB
testcase_14 AC 2 ms
4,380 KB
testcase_15 AC 1 ms
4,376 KB
testcase_16 AC 1 ms
4,376 KB
testcase_17 AC 2 ms
4,380 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#define REP(i, a, n) for (ll i = ((ll) a); i < ((ll) n); i++)
using namespace std;
typedef long long ll;

const ll MOD = 1000000007;

int main(void) {
  ll N;
  cin >> N;
  vector<ll> S;
  for (ll n = N; n > 0; n /= 2) S.push_back(n % 2);
  reverse(S.begin(), S.end());

  {
    ll x = 0;
    REP(i, 0, S.size()) x = x * 2 + S[i];
    assert(x == N);
  }

#define update(x, y, c, d) (((x) = (((x) + ((y) * 2 % MOD)) % MOD + ((c) * (d)) % MOD) % MOD)) % MOD

  vector<vector<vector<ll>>> dp(S.size() + 1, vector<vector<ll>>(S.size() + 1, vector<ll>(2, 0)));
  vector<vector<vector<ll>>> cnt(S.size() + 1, vector<vector<ll>>(S.size() + 1, vector<ll>(2, 0)));
  vector<vector<vector<bool>>> ok(S.size() + 1, vector<vector<bool>>(S.size() + 1, vector<bool>(2, false)));
  cnt[0][0][1] = 1;
  ok[0][0][1] = true;
  REP(i, 0, S.size()) REP(j, 0, S.size() + 1) {
    if (ok[i][j][0]) {
      update(dp[i + 1][j][0], dp[i][j][0], cnt[i][j][0], 0);
      cnt[i + 1][j][0] += cnt[i][j][0];
      ok[i + 1][j][0] = true;
    }
    if (ok[i][j][0] && j + 1 <= S.size()) {
      update(dp[i + 1][j + 1][0], dp[i][j][0], cnt[i][j][0], 1);
      cnt[i + 1][j + 1][0] += cnt[i][j][0];
      ok[i + 1][j + 1][0] = true;
    }
    if (ok[i][j][1]) {
      update(dp[i + 1][j][S[i] == 0], dp[i][j][1], cnt[i][j][1], 0);
      cnt[i + 1][j][S[i] == 0] += cnt[i][j][1];
      ok[i + 1][j][S[i] == 0] = true;
    }
    if (ok[i][j][1] && j + 1 <= S.size() && S[i] == 1) {
      update(dp[i + 1][j + 1][1], dp[i][j][1], cnt[i][j][1], 1);
      cnt[i + 1][j + 1][1] += cnt[i][j][1];
      ok[i + 1][j + 1][1] = true;
    }
  }

  ll ans = 0;
  REP(i, 0, S.size() + 1) ans = (ans + (dp[S.size()][i][0] * i) % MOD) % MOD;
  REP(i, 0, S.size() + 1) ans = (ans + (dp[S.size()][i][1] * i) % MOD) % MOD;
  cout << ans << endl;
}
0