結果
問題 |
No.2055 12x34...
|
ユーザー |
![]() |
提出日時 | 2022-08-21 13:04:02 |
言語 | Java (openjdk 23) |
結果 |
AC
|
実行時間 | 718 ms / 2,000 ms |
コード長 | 956 bytes |
コンパイル時間 | 2,312 ms |
コンパイル使用メモリ | 78,536 KB |
実行使用メモリ | 91,968 KB |
最終ジャッジ日時 | 2024-10-10 06:00:23 |
合計ジャッジ時間 | 23,096 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 41 |
ソースコード
import java.io.BufferedReader; import java.io.InputStreamReader; 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 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(); int mod = 998244353; Map<Integer, Long> dp = new HashMap<>(); Map<Integer, Long> val = new HashMap<>(); for (int i = 0; i < n; i++) { long d1 = dp.getOrDefault(a[i], 0L); long d2 = dp.getOrDefault(a[i] - 1, 0L); long v1 = val.getOrDefault(a[i], 0L); long v2 = val.getOrDefault(a[i] - 1, 0L); dp.put(a[i], (d1 + d2 + v2) % mod); val.put(a[i], v1 + 1); } long ans = 0; for (long v : dp.values()) { ans += v; } ans %= mod; System.out.println(ans); } }