結果

問題 No.2374 ASKT Subsequences
コンテスト
ユーザー ks2m
提出日時 2023-07-07 21:55:03
言語 Java
(openjdk 25.0.2)
コンパイル:
javac -encoding UTF8 _filename_
実行:
java -ea -Xmx700m -Xss256M -DONLINE_JUDGE=true _class_
結果
AC  
実行時間 881 ms / 2,000 ms
コード長 951 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,424 ms
コンパイル使用メモリ 81,440 KB
実行使用メモリ 253,204 KB
最終ジャッジ日時 2026-04-03 05:30:25
合計ジャッジ時間 19,963 ms
ジャッジサーバーID
(参考情報)
judge2_0 / judge3_1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

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));
		int n = Integer.parseInt(br.readLine());
		String[] sa = br.readLine().split(" ");
		int[] a = new int[n];
		for (int i = 0; i < n; i++) {
			a[i] = Integer.parseInt(sa[i]);
		}
		br.close();

		long[][][] dp = new long[1990][2001][4];
		for (int i = 0; i < n; i++) {
			for (int k = 1; k < 1990; k++) {
				dp[k][a[i]][0]++;
				if (a[i] - k - 10 > 0) {
					dp[k][a[i]][1] += dp[k][a[i] - k - 10][0];
				}
				if (a[i] + k <= 2000) {
					dp[k][a[i]][2] += dp[k][a[i] + k][1];
				}
				if (a[i] - k - 1 > 0) {
					dp[k][a[i]][3] += dp[k][a[i] - k - 1][2];
				}
			}
		}
		long ans = 0;
		for (int i = 0; i < 1990; i++) {
			for (int j = 0; j < 2001; j++) {
				ans += dp[i][j][3];
			}
		}
		System.out.println(ans);
	}
}
0