結果

問題 No.3310 mod998
コンテスト
ユーザー ks2m
提出日時 2025-10-24 22:33:49
言語 Java
(openjdk 23)
結果
WA  
実行時間 -
コード長 2,005 bytes
コンパイル時間 3,124 ms
コンパイル使用メモリ 81,160 KB
実行使用メモリ 71,024 KB
最終ジャッジ日時 2025-10-24 22:34:12
合計ジャッジ時間 22,386 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other WA * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

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 = 998;
		boolean[] b = new boolean[mod];
		PrintWriter pw = new PrintWriter(System.out);
		for (int i = 0; i < t; i++) {
			String[] sa = br.readLine().split(" ");
			int n = Integer.parseInt(sa[0]);
			int m = Integer.parseInt(sa[1]);
			Arrays.fill(b, false);
			List<Integer> list = new ArrayList<>();
			int v = 1;
			b[v] = true;
			list.add(v);
			int ls = 0;
			while (true) {
				v *= n;
				v %= mod;
				if (b[v]) {
					for (int j = 0; j < list.size(); j++) {
						if (list.get(j) == v) {
							ls = j;
							break;
						}
					}
					break;
				}
				b[v] = true;
				list.add(v);
				if (v == 0) {
					break;
				}
			}
			int size = list.size();
			int size1 = size - ls;
			int sum = 0;
			for (int j = ls; j < size; j++) {
				sum += list.get(j);
			}
			sum %= mod;

			for (int j = 0; j < m; j++) {
				long ans = 0;
				String s = br.readLine();
				if (b[0]) {
					if (s.length() > 5) {
						pw.println(sum);
					} else {
						int end = Integer.parseInt(s);
						if (end >= size1) {
							pw.println(sum);
						} else {
							for (int k = 0; k <= end; k++) {
								ans += list.get(k);
							}
							ans %= mod;
							pw.println(ans);
						}
					}
				} else {
					long v2 = 0;
					for (int k = 0; k < s.length(); k++) {
						v2 *= 10;
						int c = s.charAt(k) - '0';
						v2 += c;
						if (v2 >= size) {
							v2 -= ls;
							ans += v2 / size1 * sum;
							ans %= mod;
							v2 %= size1;
							v2 += ls;
						}
					}
					for (int k = 0; k <= v2; k++) {
						ans += list.get(k);
					}
					ans %= mod;
					pw.println(ans);
				}
			}
		}
		pw.flush();
		br.close();
	}
}
0