結果

問題 No.2966 Simple Plus Minus Problem
ユーザー ks2mks2m
提出日時 2024-11-16 17:40:55
言語 Java21
(openjdk 21)
結果
WA  
実行時間 -
コード長 1,993 bytes
コンパイル時間 3,392 ms
コンパイル使用メモリ 77,064 KB
実行使用メモリ 67,344 KB
最終ジャッジ日時 2024-11-16 17:42:56
合計ジャッジ時間 18,226 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 46 ms
36,776 KB
testcase_01 AC 45 ms
36,740 KB
testcase_02 AC 44 ms
36,708 KB
testcase_03 AC 391 ms
57,832 KB
testcase_04 AC 397 ms
61,836 KB
testcase_05 AC 45 ms
36,716 KB
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 AC 44 ms
36,744 KB
testcase_33 AC 44 ms
36,832 KB
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
testcase_40 WA -
testcase_41 WA -
testcase_42 WA -
testcase_43 WA -
testcase_44 WA -
testcase_45 WA -
testcase_46 WA -
testcase_47 WA -
testcase_48 WA -
testcase_49 WA -
testcase_50 WA -
testcase_51 WA -
testcase_52 AC 384 ms
60,672 KB
testcase_53 WA -
testcase_54 WA -
testcase_55 WA -
testcase_56 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.BufferedReader;
import java.io.InputStreamReader;

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String[] sa = br.readLine().split(" ");
		int n = Integer.parseInt(sa[0]);
		int k = Integer.parseInt(sa[1]);
		sa = br.readLine().split(" ");
		long[] a = new long[n];
		for (int i = 0; i < n; i++) {
			a[i] = Integer.parseInt(sa[i]);
		}
		br.close();

		int mod = 998244353;
		if (k <= 6) {
			for (int i = 0; i < k; i++) {
				long[] b = new long[n + 1];
				for (int j = 0; j < n; j++) {
					if (j % 2 == 0) {
						b[j + 1] = b[j] + a[j];
					} else {
						b[j + 1] = b[j] - a[j];
					}
				}
				System.arraycopy(b, 1, a, 0, n);
			}
			StringBuilder sb = new StringBuilder();
			for (int i = 0; i < n; i++) {
				long ans = a[i] % mod;
				if (ans < 0) {
					ans += mod;
				}
				sb.append(ans).append(' ');
			}
			sb.deleteCharAt(sb.length() - 1);
			System.out.println(sb.toString());

		} else {
			long[][] b = new long[6][n + 1];
			for (int j = 0; j < n; j++) {
				b[0][j + 1] = a[j];
			}
			for (int i = 0; i < 5; i++) {
				for (int j = 0; j < n; j++) {
					if (j % 2 == 0) {
						b[i + 1][j + 1] = b[i + 1][j] + b[i][j + 1];
					} else {
						b[i + 1][j + 1] = b[i + 1][j] - b[i][j + 1];
					}
				}
			}

			int m2 = 998244354 / 2;
			StringBuilder sb = new StringBuilder();
			for (int j = 1; j <= n; j++) {
				long b0 = 0;
				long d1 = 0;
				long d2 = 0;
				if (k % 2 == 0) {
					b0 = b[0][j];
					d1 = b[2][j] - b[0][j];
					d2 = b[4][j] - b[2][j];
				} else {
					b0 = b[1][j];
					d1 = b[3][j] - b[1][j];
					d2 = b[5][j] - b[3][j];
				}
				long e = (d2 - d1) % mod;
				int k2 = k / 2;
				long add = d1 % mod * k2 % mod + e * (k2 - 1) % mod * k2 % mod * m2 % mod;
				long ans = (b0 + add) % mod;
				sb.append(ans).append(' ');
			}
			sb.deleteCharAt(sb.length() - 1);
			System.out.println(sb.toString());
		}
	}
}
0