結果

問題 No.3432 popcount & sum (Hard)
コンテスト
ユーザー tassei903
提出日時 2026-01-31 13:46:51
言語 PyPy3
(7.3.17)
結果
AC  
実行時間 512 ms / 2,000 ms
コード長 1,237 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 357 ms
コンパイル使用メモリ 82,368 KB
実行使用メモリ 172,148 KB
最終ジャッジ日時 2026-01-31 13:46:59
合計ジャッジ時間 6,633 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
input = lambda :sys.stdin.readline()[:-1]
ni = lambda :int(input())
na = lambda :list(map(int,input().split()))
yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES")
no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO")
#######################################################################

"""

\sum_k 2^k * p(k, a, b)

p(k, j, n) = 2^k を含むようなp(a) = p(b) = j の数 a, b <= n

"""
mod = 998244353

from functools import lru_cache

@lru_cache(maxsize=None)
def f(k, j, n):
    if n < 0:
        return 0
    if j < 0:
        return 0
    if k == 0:
        return g(j-1, (n-1)//2)
    return (f(k-1, j-1, (n-1)//2) + f(k-1, j, n//2)) % mod
@lru_cache(maxsize=None)
def g(j, n):
    if n < 0:
        return 0
    if j < 0:
        return 0
    if j == 0:
        return 1
    if n == 0:
        if j > 0:
            return 0
        return 0
    return (g(j-1, (n-1)//2) + g(j, n//2)) % mod

N = ni()
n = 64
ans = 0
for k in range(n):
    for j in range(n + 1):
        # print(k, j, N, f(k, j, N))
        ans += f(k, j, N) * f(k, j, N) % mod * pow(2, k, mod) % mod
        ans %= mod

ans += (N + 1) * N // 2
ans %= mod

print(ans * pow(2, mod-2, mod) % mod)
0