結果

問題 No.2752 文字列の数え上げ mod 998244353
ユーザー ks2mks2m
提出日時 2024-05-10 22:37:49
言語 Java21
(openjdk 21)
結果
TLE  
実行時間 -
コード長 1,973 bytes
コンパイル時間 2,699 ms
コンパイル使用メモリ 79,660 KB
実行使用メモリ 71,992 KB
最終ジャッジ日時 2024-05-10 22:38:05
合計ジャッジ時間 13,178 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 48 ms
50,404 KB
testcase_01 AC 49 ms
50,528 KB
testcase_02 AC 55 ms
50,572 KB
testcase_03 AC 49 ms
50,136 KB
testcase_04 AC 48 ms
50,472 KB
testcase_05 AC 47 ms
50,536 KB
testcase_06 AC 49 ms
50,516 KB
testcase_07 AC 50 ms
50,500 KB
testcase_08 AC 50 ms
50,292 KB
testcase_09 AC 49 ms
50,588 KB
testcase_10 AC 49 ms
50,196 KB
testcase_11 AC 49 ms
50,536 KB
testcase_12 AC 49 ms
50,572 KB
testcase_13 AC 50 ms
50,532 KB
testcase_14 AC 48 ms
50,456 KB
testcase_15 AC 50 ms
50,372 KB
testcase_16 AC 50 ms
50,484 KB
testcase_17 AC 50 ms
50,516 KB
testcase_18 AC 49 ms
50,268 KB
testcase_19 AC 49 ms
50,500 KB
testcase_20 AC 1,526 ms
63,712 KB
testcase_21 TLE -
testcase_22 TLE -
testcase_23 TLE -
testcase_24 AC 404 ms
71,992 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int t = Integer.parseInt(br.readLine());
		int mod = 998244353;
		long[][] a = new long[2][2];
		a[0][0] = 26;
		a[0][1] = 26;
		a[1][0] = 0;
		a[1][1] = 1;
		long[] c = new long[] {26, 1};

		long[] l = new long[t];
		long[] l2 = new long[t + 1];
		for (int i = 0; i < t; i++) {
			l[i] = Long.parseLong(br.readLine()) - 1;
			l2[i] = l[i];
		}
		br.close();

		Arrays.sort(l2);
		Map<Long, Long> map = new HashMap<>();
		map.put(0L, 26L);
		for (int i = 1; i <= t; i++) {
			if (l2[i - 1] < l2[i]) {
				long[][] b = matrixPow(a, l2[i] - l2[i - 1], mod);
				c = matrixMul1(c, b, mod);
				long ans = c[1] + 25;
				if (ans >= mod) ans -= mod;
				map.put(l2[i], ans);
			}
		}

		PrintWriter pw = new PrintWriter(System.out);
		for (int i = 0; i < t; i++) {
			pw.println(map.get(l[i]));
		}
		pw.flush();
	}

	static long[][] matrixPow(long[][] a, long k, int m) {
		if (k == 1) {
			return a;
		}
		long[][] ret = matrixPow(a, k / 2, m);
		ret = matrixMul(ret, ret, m);
		if (k % 2 == 1) {
			ret = matrixMul(ret, a, m);
		}
		return ret;
	}

	static long[][] matrixMul(long[][] a, long[][] b, int m) {
		int h = a.length;
		int w = b[0].length;
		int n = a[0].length;
		long[][] c = new long[h][w];
		for (int i = 0; i < h; i++) {
			for (int j = 0; j < w; j++) {
				for (int x = 0; x < n; x++) {
					c[i][j] += a[i][x] * b[x][j];
					c[i][j] %= m;
				}
			}
		}
		return c;
	}

	static long[] matrixMul1(long[] a, long[][] b, int m) {
		int w = b[0].length;
		int n = a.length;
		long[] c = new long[w];
		for (int j = 0; j < w; j++) {
			for (int x = 0; x < n; x++) {
				c[j] += a[x] * b[x][j];
				c[j] %= m;
			}
		}
		return c;
	}
}
0