結果
問題 |
No.1194 Replace
|
ユーザー |
![]() |
提出日時 | 2023-02-01 16:22:20 |
言語 | Java (openjdk 23) |
結果 |
WA
|
実行時間 | - |
コード長 | 2,599 bytes |
コンパイル時間 | 2,806 ms |
コンパイル使用メモリ | 85,868 KB |
実行使用メモリ | 147,936 KB |
最終ジャッジ日時 | 2024-07-01 11:21:17 |
合計ジャッジ時間 | 36,376 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 8 WA * 19 |
ソースコード
import java.io.*; import java.util.*; import java.util.stream.*; public class Main { static HashMap<Integer, ArrayList<Integer>> graph = new HashMap<>(); static HashMap<Integer, Integer> visited = new HashMap<>(); public static void main(String[] args) throws Exception { Scanner sc = new Scanner(); int n = sc.nextInt(); int m = sc.nextInt(); for (int i = 0; i < m; i++) { int left = sc.nextInt(); int right = sc.nextInt(); if (!graph.containsKey(left)) { graph.put(left, new ArrayList<>()); } graph.get(left).add(right); } long ans = (n + 1L) * n / 2; for (int x : graph.keySet()) { int value = getValue(x, new HashSet<>()); ans += value - x; } System.out.println(ans); } static int getValue(int x, HashSet<Integer> used) { if (used.contains(x)) { return 0; } if (!visited.containsKey(x)) { used.add(x); int value = x; if (graph.containsKey(x)) { for (int y : graph.get(x)) { value = Math.max(value, getValue(y, used)); } } visited.put(x, value); used.remove(x); } return visited.get(x); } } class Utilities { static String arrayToLineString(Object[] arr) { return Arrays.stream(arr).map(x -> x.toString()).collect(Collectors.joining("\n")); } static String arrayToLineString(int[] arr) { return String.join("\n", Arrays.stream(arr).mapToObj(String::valueOf).toArray(String[]::new)); } } class Scanner { BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); StringTokenizer st = new StringTokenizer(""); StringBuilder sb = new StringBuilder(); public Scanner() throws Exception { } public int nextInt() throws Exception { return Integer.parseInt(next()); } public long nextLong() throws Exception { return Long.parseLong(next()); } public double nextDouble() throws Exception { return Double.parseDouble(next()); } public int[] nextIntArray() throws Exception { return Stream.of(br.readLine().split(" ")).mapToInt(Integer::parseInt).toArray(); } public String next() throws Exception { while (!st.hasMoreTokens()) { st = new StringTokenizer(br.readLine()); } return st.nextToken(); } }