結果

問題 No.3118 Increment or Multiply
コンテスト
ユーザー 入野拳樹
提出日時 2025-04-20 12:59:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 971 bytes
コンパイル時間 892 ms
コンパイル使用メモリ 70,688 KB
実行使用メモリ 16,064 KB
最終ジャッジ日時 2025-04-20 12:59:48
合計ジャッジ時間 5,005 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 1
other AC * 5 TLE * 1 -- * 29
権限があれば一括ダウンロードができます

ソースコード

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){
		
		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;
				}
			
				ll nv = ((v-1)/A)*A;
				ll ndist = dist + 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