본문 바로가기
알고리즘

최장 증가 수열 (LIS : Longest Increasing Subsequence)

by 출근길 2018. 7. 19.

1. 수열을 받는 A배열

2. LIS를 저장하는 L배열

3. 순서를 저장하는 P배열


* O(NlogN) 로직

1. LIS(L배열) 마지막 값보다 수열(A배열) 값이 크면 LIS 뒤에 추가

2. 값이 작거나 같다면

1) 같으면 continue;

2) 작으면 lower bound를 이용하여 위치교환



import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.Stack;
import java.util.StringTokenizer;

public class Main {
	/*
	 * 가장 긴 증가하는 부분 수열 5
	 * https://www.acmicpc.net/problem/14003
	 */

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st;
		
		/*
		 * 첫째 줄에 수열 A의 크기 N (1 ≤ N ≤ 1,000)이 주어진다.
		 * 둘째 줄에는 수열 A를 이루고 있는 Ai가 주어진다. (1 ≤ Ai ≤ 1,000)
		 * 
		 * ex) input
		 * 6
		 * 10 20 10 30 20 50
		 */
		
		int N = Integer.parseInt(br.readLine());
		st = new StringTokenizer(br.readLine());
		
		int[] A = new int[N+1];	//수열
		int[] L = new int[N+1];	//LIS 저장
		int[] P = new int[N+1];	//LIS 순번 저장
		
		for (int i = 1; i <= N; i++) {
			A[i] = Integer.parseInt(st.nextToken());
		}
		
		int size = 0;
		for (int i = 1; i <= N; i++) {
			if (L[size] < A[i]) {
				size++;
				L[size] = A[i];
				P[i] = size;
			} else if (L[size] == A[i]) {
				continue;			
			} else {
				int idx = lowerBound(L, 1, size, A[i]);
				L[idx] = A[i];
				P[i] = idx;
			}
		}
		
		bw.write(size+"\n");
		
		
		Stack stk = new Stack();
		for (int i = N; i >= 1; i--) {
			if(P[i] != size) continue;
			if (stk.isEmpty()) {
				stk.add(A[i]);
				size--;
			} else {
				if (stk.peek() > A[i]) {
					stk.add(A[i]);
					size--;
				}
			}
		}
		
		while(!stk.isEmpty()) {
			bw.write(stk.pop()+" ");
		}
		bw.write("\n");		
		
		br.close();
		bw.flush();
		bw.close();
	}
	
	static int lowerBound (int[] L, int l, int r, int val) {
		
		int mid = 0;
		int ans = 0;
		while(l <= r) {
			mid = (l+r)/2;
			if (L[mid] < val) {
				l = mid + 1;
			} else {
				ans = mid;
				r = mid - 1;
			} 
		}
		return ans;
	}
}



'알고리즘' 카테고리의 다른 글

벨만포드  (0) 2018.09.11
에라토스 테네스의 체  (0) 2018.07.18
최소 공통 조상(LCA : Lowest Common Ancestor)  (0) 2018.07.18
페르마의 소정리  (0) 2018.07.17
Union-Find  (0) 2018.07.17
위상정렬 (Topological Sort)  (0) 2018.07.17
SPFA (Shortest Path Faster Algorithm)  (0) 2018.07.17
인덱스 트리 (Indexed Tree)  (0) 2018.07.17