結果

問題 No.737 PopCount
コンテスト
ユーザー pazzle1230
提出日時 2018-09-29 00:53:21
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 1,000 ms
コード長 1,496 bytes
コンパイル時間 1,521 ms
コンパイル使用メモリ 170,596 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-12 07:49:39
合計ジャッジ時間 2,356 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#define INF_LL (int64)1e18
#define INF (int32)1e9
#define REP(i, n) for(int64 i = 0;i < (n);i++)
#define FOR(i, a, b) for(int64 i = (a);i < (b);i++)
#define all(x) x.begin(),x.end()
#define fs first
#define sc second

using int32 = int_fast32_t;
using uint32 = uint_fast32_t;
using int64 = int_fast64_t;
using uint64 = uint_fast64_t;
using PII = pair<int32, int32>;
using PLL = pair<int64, int64>;

const double eps = 1e-10;

template<typename A, typename B>inline void chmin(A &a, B b){if(a > b) a = b;}
template<typename A, typename B>inline void chmax(A &a, B b){if(a < b) a = b;}

const int64 mod = 1e9+7;

int64 N;
vector<int32> bit;

int64 sum[70][70][2], cnt[70][70][2];

PLL dfs(int32 rest, int32 dig = 0, bool tight = 1){
	if(dig == bit.size()) return PLL(0, rest == 0);
	int64 &rets = sum[rest][dig][tight], &retc = cnt[rest][dig][tight];
	if(~rets) return PLL(rets, retc);
	rets = 0; retc = 0;
	int32 lim = tight ? bit[dig] : 1;
	int64 val = (1LL << (bit.size()-dig-1))%mod;
	REP(i, lim+1){
		if(i > rest) continue;
		PLL x = dfs(rest-(i == 1), dig+1, tight && i == lim);
		retc = (retc+x.sc)%mod;
		rets = (rets+x.fs+x.sc*val*(i==1)%mod)%mod;
	}
	return PLL(rets, retc);
}

int main(void){
	cin >> N;
	memset(sum, -1, sizeof sum);
	memset(cnt, -1, sizeof cnt);
	while(N){
		bit.push_back(N&1);
		N >>= 1;
	}
	reverse(all(bit));
	int64 res = 0;
	FOR(i, 1, bit.size()+1){
		res += dfs(i).fs*i%mod;
		res %= mod;
	}
	cout << res << endl;
}
0