結果

問題 No.3432 popcount & sum (Hard)
コンテスト
ユーザー 00 Sakuda
提出日時 2026-02-04 01:30:05
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
結果
AC  
実行時間 15 ms / 2,000 ms
コード長 1,227 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,286 ms
コンパイル使用メモリ 199,728 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2026-02-04 01:30:09
合計ジャッジ時間 4,147 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <vector>
#include <set>
#include <map>
#include <algorithm>
#include <iomanip>
#include <numeric>
#include <queue>
#include <stack>
using namespace std;
using ll = long long;
#include <atcoder/modint>
using namespace atcoder;
using mint = modint998244353;
bool isflg(int k, ll x) {
	return ((1LL << k) & x);
}
int main() {
	ll N;cin >> N;
	int K = 60;
	mint res = 0;
	for (int k = 0;k <= K;k++) {
		vector<vector<mint>> dp(K + 1, vector<mint>(2, 0));
		dp[0][0] = 1;
		for (int i = K;i >= 0;i--) {
			vector<vector<mint>> ndp(K + 1, vector<mint>(2, 0));
			for (int p = 0;p <= K;p++) {
				if (i == k) {
					if (p+1 <= K) {
						ndp[p + 1][1] += dp[p][1];
						if (isflg(k, N)) {
							ndp[p + 1][0] += dp[p][0];
						}
					}
					continue;
				}
				ndp[p][1] += dp[p][1];
				if (p+1 <= K) {
					ndp[p + 1][1] += dp[p][1];
				}

				if (!isflg(i, N)) {
					ndp[p][0] += dp[p][0];
				} else {
					ndp[p][1] += dp[p][0];
					if (p+1 <= K) {
						ndp[p + 1][0] += dp[p][0];
					}
				}
			}
			swap(dp, ndp);
		}
		for (int t = 0;t <= K;t++) {
			mint sm = dp[t][0] + dp[t][1];
			mint a = (1LL << k);
			sm = (sm * (sm +1)) / 2;
			res += a * sm;
		}
	}
	cout << res.val() << endl;
}
0