結果

問題 No.2941 Sigma Music Game Score Problem
ユーザー ks2m
提出日時 2024-10-18 23:00:38
言語 Java
(openjdk 23)
結果
AC  
実行時間 627 ms / 2,500 ms
コード長 1,068 bytes
コンパイル時間 3,425 ms
コンパイル使用メモリ 77,252 KB
実行使用メモリ 145,892 KB
最終ジャッジ日時 2024-10-18 23:01:45
合計ジャッジ時間 13,599 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

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(" ");
		long m = Long.parseLong(sa[0]);
		int n = Integer.parseInt(sa[1]);
		sa = br.readLine().split(" ");
		long[] x = new long[n + 2];
		for (int i = 0; i < n; i++) {
			x[i + 1] = Long.parseLong(sa[i]);
		}
		br.close();
		x[n + 1] = m + 1;

		int mod = 998244353;
		long m6 = modinv(6, mod);
		long ans = 0;
		for (int i = 1; i < x.length; i++) {
			long d = x[i] - x[i - 1] - 1;
			d %= mod;
			long val = d * (d + 1) % mod * (2 * d + 1) % mod * m6 % mod;
			ans += val;
		}
		ans %= 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;
	}
}
0