結果

問題 No.737 PopCount
コンテスト
ユーザー yuppe19 😺
提出日時 2018-11-07 16:20:47
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 1,000 ms
コード長 1,707 bytes
コンパイル時間 1,002 ms
コンパイル使用メモリ 88,328 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-20 20:45:57
合計ジャッジ時間 1,628 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

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;

vector<int> p2;

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

i64 f(i64 x) {
  string s = to_bin(x);
  int n = static_cast<int>(s.size());
  vector<vector<i64>> dp(3, vector<i64>(n+1, 0));
  vector<vector<i64>> dq(3, vector<i64>(n+1, 0));
  dp[1][0] = 1;
  for(int i=n-1; i>=0; --i) {
    vector<vector<i64>> ndp(3, vector<i64>(n+1, 0));
    vector<vector<i64>> ndq(3, vector<i64>(n+1, 0));
    for(int flag=0; flag<3; ++flag) {
      for(int cnt1=0; cnt1<n; ++cnt1) {
        if(!dp[flag][cnt1]) { continue; }
        for(int d=0; d<2; ++d) {
          int dflag = d == s[i] - '0';
          if(d > s[i] - '0') { dflag = 2; }
          int nflag = flag * dflag;
          if(nflag >= 2) { nflag = 2; }
          if(flag == 0 && dflag == 2) { nflag = 2; }
          int ncnt1 = cnt1 + (d == 1);
          ndp[nflag][ncnt1] += dp[flag][cnt1];
          ndp[nflag][ncnt1] %= mod;
          ndq[nflag][ncnt1] += dq[flag][cnt1] + dp[flag][cnt1] * d * p2[n-1-i];
          ndq[nflag][ncnt1] %= mod;
        }
      }
    }
    dp = ndp;
    dq = ndq;
  }
  i64 res = 0;
  for(int flag=0; flag<2; ++flag) {
    for(int cnt1=1; cnt1<=n; ++cnt1) {
      res += dq[flag][cnt1] * cnt1;
      res %= mod;
    }
  }
  return res;
}

int main(void) {
  p2.resize(110);
  p2[0] = 1;
  for(int i=0; i<105; ++i) {
    p2[i+1] = p2[i] * 2 % mod;
  }
  i64 n; scanf("%ld", &n);
  i64 res = f(n);
  printf("%ld\n", res);
  return 0;
}
0