結果

問題 No.3118 Increment or Multiply
コンテスト
ユーザー okkuu
提出日時 2025-04-20 13:06:57
言語 C++23
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 72 ms / 2,000 ms
コード長 703 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 8,008 ms
コンパイル使用メモリ 356,780 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2025-04-20 13:07:08
合計ジャッジ時間 10,338 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include<bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using namespace atcoder; 
using ll=long long;
using mint=modint998244353;
const ll inf=(1e9+7)*(1e9+7);
int dx[4]={1,0,0,-1},dy[4]={0,1,-1,0};
const int mod=998244353;
using P=pair<ll,int>;
int main(){
	int t;
	ll n,a;
	cin>>t;
	while(cin>>n>>a){
		if(a==1){
			mint m=n;
			cout<<(m*(m-1)/2).val()<<endl;
		}else{
			vector<ll>v={n};
			while(v.back()!=0)
				v.push_back(v.back()/a);
			mint ans=0,tmp=0;
			for(int i=0;i+1<v.size();i++){
				ans+=(v[i]+tmp)*(v[i]-v[i+1])-(v[i]+v[i+1]+mint(1))*(v[i]-v[i+1])/2;
				tmp+=1+v[i]%a;
			}
			cout<<ans.val()<<endl;
		}
	}
}
0