[백준 14002] 가장 긴 증가하는 부분 수열 4 (C#, C++) - soo:bak

문제 링크

14002번 - 가장 긴 증가하는 부분 수열 4

설명

길이 N (1 ≤ N ≤ 1,000)의 정수 수열이 주어지는 상황에서, 가장 긴 증가하는 부분 수열의 길이와 그 수열을 구하는 문제입니다.

O(N log N) LIS 알고리즘을 사용하고, 경로를 복원하기 위해 각 원소의 이전 인덱스를 추적해야 합니다.

접근법

이진 탐색 기반 LIS 알고리즘에 경로 복원 기능을 추가합니다.

먼저 세 개의 배열을 사용합니다. lisVal[len]은 길이 len인 LIS의 마지막 값을, lisPos[len]은 그 값의 원본 인덱스를, prev[i]는 인덱스 i를 마지막으로 하는 LIS의 이전 원소 인덱스를 저장합니다.

다음으로 각 원소를 순회하며 lower_bound로 lisVal 배열에서 현재 원소가 들어갈 위치를 찾습니다. 이 위치가 현재 원소의 LIS 길이가 됩니다.

이후 prev[i]를 이전 위치의 인덱스로 설정하고, lisVal과 lisPos를 갱신합니다. LIS 길이가 늘어났다면 전체 길이도 증가시킵니다.

마지막으로 최종 LIS의 마지막 인덱스에서 시작하여 prev 배열을 따라가며 역순으로 수열을 복원합니다.

시간 복잡도는 O(N log N)입니다.

---

Code

C#

using System;
using System.Collections.Generic;

namespace Solution {
  class Program {
    static void Main(string[] args) {
      var n = int.Parse(Console.ReadLine()!);
      var arr = Array.ConvertAll(Console.ReadLine()!.Split(), int.Parse);

      var lisVal = new int[n];
      var lisPos = new int[n];
      var prev = new int[n];
      var len = 0;

      for (var i = 0; i < n; i++) {
        var x = arr[i];
        var idx = Array.BinarySearch(lisVal, 0, len, x);
        if (idx < 0)
          idx = ~idx;

        prev[i] = (idx == 0) ? -1 : lisPos[idx - 1];
        lisVal[idx] = x;
        lisPos[idx] = i;
        if (idx == len)
          len++;
      }

      Console.WriteLine(len);
      var st = new Stack<int>();
      for (var cur = lisPos[len - 1]; cur != -1; cur = prev[cur])
        st.Push(arr[cur]);

      while (st.Count > 0) {
        Console.Write(st.Pop());
        Console.Write(st.Count > 0 ? " " : "\n");
      }
    }
  }
}

C++

#include <bits/stdc++.h>
using namespace std;

typedef vector<int> vi;
typedef stack<int> si;

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);

  int n; cin >> n;
  vi a(n);
  for (int i = 0; i < n; i++)
    cin >> a[i];

  vi lisVal(n), lisPos(n), prev(n);
  int len = 0;

  for (int i = 0; i < n; i++) {
    int x = a[i];
    auto it = lower_bound(lisVal.begin(), lisVal.begin() + len, x);
    int idx = it - lisVal.begin();

    prev[i] = (idx == 0) ? -1 : lisPos[idx - 1];
    lisVal[idx] = x;
    lisPos[idx] = i;
    if (idx == len)
      len++;
  }

  cout << len << "\n";
  si st;
  for (int cur = lisPos[len - 1]; cur != -1; cur = prev[cur])
    st.push(a[cur]);
  while (!st.empty()) {
    cout << st.top(); st.pop();
    if (!st.empty())
      cout << ' ';
  }
  cout << "\n";

  return 0;
}