結果

問題 No.3462 Buttons
コンテスト
ユーザー tau1235
提出日時 2026-02-28 15:14:47
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 768 ms / 2,000 ms
コード長 682 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,916 ms
コンパイル使用メモリ 338,376 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2026-02-28 15:15:08
合計ジャッジ時間 16,893 ms
ジャッジサーバーID
(参考情報)
judge7 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

#include<bits/stdc++.h>
#include<atcoder/modint>
using namespace std;

void solve(){
	using mint=atcoder::modint998244353;
	int a,b,k;
	cin>>a>>b>>k;
	mint ans=0;
	auto f=[&](int x){
		return mint(a)*x*mint(b).pow(k-x);
	};
	if (a==0) ans=0;
	else if (b==0) ans=mint(max(0,a))*k;
	else if (b==1) ans=mint(max(0,a))*k;
	else if (b==-1){
		if (a>0) ans=f(k);
		if (a<0) ans=f(k-1);
	}
	else{
		if (a>0&&b>0) ans=f(1);
		if (a>0&&b<0){
			if ((k-1)%2==0) ans=f(1);
			else if (k-2>=0) ans=f(2);
		}
		if (a<0&&b>0) ans=0;
		if (a<0&&b<0){
			if ((k-1)%2==1) ans=f(1);
			else if (k-2>=0) ans=f(2);
		}
	}
	cout<<ans.val()<<endl;
}

int main(){
	int t;
	cin>>t;
	while (t--) solve();
}
0