結果

問題 No.2351 Butterfly in Summer
ユーザー ks2mks2m
提出日時 2023-06-16 21:28:17
言語 Java
(openjdk 23)
結果
AC  
実行時間 133 ms / 2,000 ms
コード長 900 bytes
コンパイル時間 2,043 ms
コンパイル使用メモリ 76,848 KB
実行使用メモリ 54,372 KB
最終ジャッジ日時 2024-06-24 13:01:54
合計ジャッジ時間 6,040 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.Scanner;

public class Main {
	public static void main(String[] args) throws Exception {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int k = sc.nextInt();
		sc.close();

		int mod = 998244353;
		long v1 = (long) k * (k - 1);
		long v2 = v1 * n % mod;
		long u1 = power(k, n, mod);
		long u2 = modinv(u1, mod);
		long ans = v2 * u2 % mod;
		System.out.println(ans);
	}

	static long modinv(long a, int m) {
		long b = m;
		long u = 1;
		long v = 0;
		long tmp = 0;

		while (b > 0) {
			long t = a / b;
			a -= t * b;
			tmp = a;
			a = b;
			b = tmp;

			u -= t * v;
			tmp = u;
			u = v;
			v = tmp;
		}

		u %= m;
		if (u < 0) u += m;
		return u;
	}

	static long power(long x, long n, int m) {
		if (n == 0) {
			return 1;
		}
		long val = power(x, n / 2, m);
		val = val * val % m;
		if (n % 2 == 1) {
			x %= m;
			val = val * x % m;
		}
		return val;
	}
}
0