結果

問題 No.1853 Many Operations
コンテスト
ユーザー 蜜蜂
提出日時 2021-12-28 22:45:46
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,501 bytes
記録
コンパイル時間 3,919 ms
コンパイル使用メモリ 234,656 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-17 10:47:17
合計ジャッジ時間 5,062 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

//g++ 1.cpp -std=c++14 -O2 -I .
#include <bits/stdc++.h>
using namespace std;

#include <atcoder/all>
using namespace atcoder;

using ll = long long;
using ld = long double;

using vi = vector<int>;
using vvi = vector<vi>;
using vll = vector<ll>;
using vvll = vector<vll>;
using vld = vector<ld>;
using vvld = vector<vld>;
using vst = vector<string>;
using vvst = vector<vst>;

#define fi first
#define se second
#define pb push_back
#define pq_big(T) priority_queue<T,vector<T>,less<T>>
#define pq_small(T) priority_queue<T,vector<T>,greater<T>>
#define all(a) a.begin(),a.end()
#define rep(i,start,end) for(ll i=start;i<(ll)(end);i++)
#define per(i,start,end) for(ll i=start;i>=(ll)(end);i--)
#define uniq(a) sort(all(a));a.erase(unique(all(a)),a.end())

constexpr ll mod = 998244353;

map<ll,ll> value;

// sum g[0] + ... + g[x]
ll h(ll x){
  if(x<=1){
    return x;
  }
  if(value[x]!=0){
    return value[x];
  }
  ll res=0;
  ll n;

  if(x>=2){
    //even g[2]+g[4]+...+g[2n] = g[1]+g[2]+...+g[n]+n
    n=x/2;
    res+=h(n)+n;
  }

  if(x>=1){
    //1 mod 4 g[1]+g[5]+...+g[4n+1] = g[0]+g[4]+...+g[4n]+n+1 = g[4]+...+g[4n]+n+1 = g[1]+...+g[n]+3n+1
    n=(x-1)/4;
    res+=h(n)+3*n+1;
  }

  if(x>=3){
    //3 mod 4 g[3]+g[7]+...+g[4n+3] = g[4]+g[8]+...+g[4n+4]+n+1 = g[1]+...+g[n+1]+3n+3
    n=(x-3)/4;
    res+=h(n+1)+3*n+3;
    res--;
  }

  res%=mod;
  value[x]=res;

  return res;
}

int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);

  ll n;
  cin>>n;
  cout<<h(n)%mod<<endl;
}
0