結果

問題 No.3118 Increment or Multiply
コンテスト
ユーザー 入野拳樹
提出日時 2025-04-20 13:25:00
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 51 ms / 2,000 ms
コード長 999 bytes
コンパイル時間 802 ms
コンパイル使用メモリ 70,072 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2025-04-20 13:25:04
合計ジャッジ時間 2,952 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#define rep(i,n) for(ll i=0;i<(n);i++)

using namespace std;
using ll = long long;

//////////////////////////////////////////////////////////////////
ll N,A;
ll mod = 998244353;

int main(){
	ll T;ll ans = 0;
	cin >> T;
	rep(i,T){

		ans = 0;
		
		cin >> N >> A;

		ll v = N; ll dist = 0;

		if(A == 1){
			if(v % 2 == 0){
				ans = ((v/2)%mod) * ((v-1)%mod)%mod;
			}else{
				ans = (v%mod) * (((v-1)/2)%mod)%mod;
			}
		}else{
			while(true){

				if(v == 0){
					break;
				}
				if(v == 1){
					ans += dist;
					ans %= mod;
					break;
				}
			
				ll nv = v/A;
				ll ndist = dist + v - v/A*A + 1;
				ll k = v - nv;
				ll sum;
				if(k % 2 == 0){
					sum = (((k/2)%mod)*((k-1)%mod))%mod + ((dist%mod) * (k%mod))%mod;
				}else{
					sum = ((k%mod)*(((k-1)/2)%mod))%mod + ((dist%mod) * (k%mod))%mod;
				}
				
				sum %= mod;

				ans += sum;
				ans %= mod;

				v = nv;dist = ndist;
			
			}
		}
		ans %= mod;
		std::cout << ans << endl;

	}
}
0