File tree Expand file tree Collapse file tree 1 file changed +49
-0
lines changed
Expand file tree Collapse file tree 1 file changed +49
-0
lines changed Original file line number Diff line number Diff line change 1+ ```
2+ import java.io.*;
3+ import java.util.*;
4+
5+ public class Main {
6+ private static final BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
7+ private static final BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
8+ private static int[] arr, answer;
9+ private static Stack<Integer> stack;
10+ private static int N;
11+
12+ public static void main(String[] args) throws IOException {
13+ init();
14+ solve();
15+
16+ for (int i = 0; i < N; i++) {
17+ if (answer[i] != -1) bw.write(answer[i] + "\n");
18+ else bw.write("infinity" + "\n");
19+ }
20+ bw.flush();
21+ bw.close();
22+ br.close();
23+ }
24+
25+ private static void init() throws IOException {
26+ N = Integer.parseInt(br.readLine());
27+
28+ arr = new int[N];
29+ answer = new int[N];
30+ stack = new Stack<>();
31+ Arrays.fill(answer, -1);
32+
33+ StringTokenizer st = new StringTokenizer(br.readLine());
34+ for (int i = 0; i < N; i++) {
35+ arr[i] = Integer.parseInt(st.nextToken());
36+ }
37+ }
38+
39+ private static void solve() {
40+ for (int i = 0; i < N; i++) {
41+ while (!stack.isEmpty() && arr[stack.peek()] >= arr[i]) {
42+ int idx = stack.pop();
43+ answer[idx] = i - idx;
44+ }
45+ stack.push(i);
46+ }
47+ }
48+ }
49+ ```
You can’t perform that action at this time.
0 commit comments