結果

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

ソースコード

diff #
raw source code 

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

struct Comb
{
  std::vector<long long> fac, finv, inv;
  const int mod;
  Comb(const int max, const int m) : mod(m), fac(max + 5), finv(max + 5), inv(max + 5)
  {
    fac[0] = fac[1] = finv[0] = finv[1] = inv[1] = 1;
    for (int i = 2; i < max + 5; ++i)
    {
      fac[i] = fac[i - 1] * i % mod;
      inv[i] = mod - inv[mod % i] * (mod / i) % mod;
      finv[i] = finv[i - 1] * inv[i] % mod;
    }
  }
  long long c(int n, int k)
  {
    if (n < k || n < 0 || k < 0) return 0;
    return fac[n] * (finv[k] * finv[n - k] % mod) % mod;
  }
  long long p(int n, int k)
  {
    if (n < k || n < 0 || k < 0) return 0;
    return fac[n] * finv[n - k] % mod;
  }
};

int main()
{
  const int mod = 998244353, b = 60;
  long long n;
  cin >> n;
  Comb cb(2 * b, mod);
  vector<int> bit;
  {
    long long tmp = n;
    while (tmp > 0)
    {
      bit.push_back(tmp % 2);
      tmp /= 2;
    }
  }
  int nowc = 0, siz = bit.size();
  vector<vector<long long>> cnt(siz, vector<long long> (siz + 1));
  {
    int sum = 0;
    for (int i = 0; i < siz; ++i) sum += bit[i];
    for (int i = 0; i < siz; ++i) if (bit[i]) cnt[i][sum] = 1;
  }
  for (int i = siz - 1; i >= 0; --i)
  {
    if (bit[i] == 1)
    {
      for (int j = 0; j <= i; ++j)
      {
        for (int k = i + 1; k < siz; ++k)
        {
          if (bit[k])
          {
            cnt[k][nowc + j] += cb.c(i, j);
            cnt[k][nowc + j] %= mod;
          }
        }
        for (int k = 0; k < i; ++k)
        {
          cnt[k][nowc + j] += cb.c(i - 1, j - 1);
          cnt[k][nowc + j] %= mod;
        }
      }
      ++nowc;
    }
  }
  // for (int i = 0; i < siz; ++i) cout << bit[i] << " \n"[i == siz - 1];
  // for (int i = 0; i < siz; ++i)
  // {
  //   for (int j = 0; j <= siz; ++j) cout << cnt[i][j] << " \n"[j == siz];
  // }
  long long ans = n % mod * ((n + 1) % mod) % mod * 499122177LL % mod;
  ans = -ans % mod + mod;
  for (int i = 0; i < siz; ++i)
  {
    for (int j = 0; j <= siz; ++j)
    {
      ans = (ans + (1LL << i) % mod * cnt[i][j] % mod * cnt[i][j] % mod) % mod;
    }
  }
  ans = ans * 499122177 % mod;
  ans += n % mod * ((n + 1) % mod) % mod * 499122177 % mod;
  ans %= mod;
  cout << ans << endl;
}
0