結果

問題 No.1804 Intersection of LIS
ユーザー ks2m
提出日時 2022-01-07 23:30:42
言語 Java
(openjdk 23)
結果
AC  
実行時間 695 ms / 2,000 ms
コード長 2,011 bytes
コンパイル時間 2,732 ms
コンパイル使用メモリ 80,004 KB
実行使用メモリ 102,676 KB
最終ジャッジ日時 2024-11-14 09:17:01
合計ジャッジ時間 16,516 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

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[] dp = new int[n + 1];
		List<List<Integer>> list = new ArrayList<>(n + 1);
		list.add(new ArrayList<>());
		for (int i = 1; i <= n; i++) {
			dp[i] = Integer.MAX_VALUE;
			list.add(new ArrayList<>());
		}

		for (int i = 0; i < n; i++) {
			int idx = Arrays.binarySearch(dp, a[i]);
			if (idx < 0) idx = ~idx;
			dp[idx] = a[i];
			list.get(idx).add(i);
		}

		int lis = Arrays.binarySearch(dp, Integer.MAX_VALUE - 1);
		if (lis < 0) {
			lis = ~lis;
			lis--;
		}

		List<Integer> list2 = list.get(lis);
		int l = 0;
		int r = list2.size() - 1;
		List<Integer> ans = new ArrayList<>();
		int cnt = 0;
		if (r == 0) {
			ans.add(a[list2.get(0)]);
			cnt++;
		}

		for (int i = lis - 1; i > 0; i--) {
			List<Integer> cur = list.get(i + 1);
			List<Integer> next = list.get(i);
			int max = a[cur.get(l)];
			int min = a[cur.get(r)];
			int nr = -1;
			int nl = -1;
			for (int j = next.size() - 1; j >= 0; j--) {
				if (nr == -1) {
					if (next.get(j) < cur.get(r) && a[next.get(j)] < min) {
						nr = j;
						nl = nr;
					}
				} else {
					if (a[next.get(j)] < max) {
						nl = j;
					} else {
						break;
					}
				}
			}
			if (nl == nr) {
				ans.add(a[next.get(nl)]);
				cnt++;
			}
			l = nl;
			r = nr;
		}
		StringBuilder sb = new StringBuilder();
		if (cnt > 0) {
			Collections.reverse(ans);
			for (int i : ans) {
				sb.append(i).append(' ');
			}
			sb.deleteCharAt(sb.length() - 1);
		}
		System.out.println(cnt);
		System.out.println(sb.toString());
	}
}
0