[백준 c++] 2529 부등호

문제 설명

https://www.acmicpc.net/problem/2529
두 종류의 부등호 기호 ‘<’와 ‘>’가 k개 나열된 순서열 A가 있다. 우리는 이 부등호 기호 앞뒤에 서로 다른 한 자릿수 숫자를 넣어서 모든 부등호 관계를 만족시키려고 한다. 예를 들어, 제시된 부등호 순서열 A가 다음과 같다고 하자.

A ⇒ < < < > < < > < >

부등호 기호 앞뒤에 넣을 수 있는 숫자는 0부터 9까지의 정수이며 선택된 숫자는 모두 달라야 한다. 아래는 부등호 순서열 A를 만족시키는 한 예이다.

3 < 4 < 5 < 6 > 1 < 2 < 8 > 7 < 9 > 0

이 상황에서 부등호 기호를 제거한 뒤, 숫자를 모두 붙이면 하나의 수를 만들 수 있는데 이 수를 주어진 부등호 관계를 만족시키는 정수라고 한다. 그런데 주어진 부등호 관계를 만족하는 정수는 하나 이상 존재한다. 예를 들어 3456128790 뿐만 아니라 5689023174도 아래와 같이 부등호 관계 A를 만족시킨다.

5 < 6 < 8 < 9 > 0 < 2 < 3 > 1 < 7 > 4

여러분은 제시된 k개의 부등호 순서를 만족하는 (k+1)자리의 정수 중에서 최댓값과 최솟값을 찾아야 한다. 앞서 설명한 대로 각 부등호의 앞뒤에 들어가는 숫자는 { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 }중에서 선택해야 하며 선택된 숫자는 모두 달라야 한다.

• 입력
  첫 줄에 부등호 문자의 개수를 나타내는 정수 k가 주어진다. 그 다음 줄에는 k개의 부등호 기호가 하나의 공백을 두고 한 줄에 모두 제시된다. k의 범위는 2 ≤ k ≤ 9 이다.

• 출력
  여러분은 제시된 부등호 관계를 만족하는 k+1 자리의 최대, 최소 정수를 첫째 줄과 둘째 줄에 각각 출력해야 한다. 단 아래 예(1)과 같이 첫 자리가 0인 경우도 정수에 포함되어야 한다. 모든 입력에 답은 항상 존재하며 출력 정수는 하나의 문자열이 되도록 해야 한다.

아이디어

시간 복잡도는 10! 이므로 브루트포스 알고리즘으로 전체 탐색하여 풀었다.

int up[10] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
int down[10] = {9, 8, 7, 6, 5, 4, 3, 2, 1, 0};

부등호를 만족하는 최대값을 구할 때
>를 만나면 down[10]배열 안에서 -1이 아닌, 가장 앞에 있는 값을 적고
<를 만나면 연속으로 몇 번 나오는지 구해서 그 수만큼 Index를 이동하여 값을 적었다.
값이 사용되면 -1로 바꿔준다.

부등호를 만족하는 최솟값을 구하는 것도 위와 부등호만 반대고 로직은 같다.

전체 코드

#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
#include <string>
using namespace std;
int n, a;
char c;
vector<char> v;
int up[10] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
int down[10] = {9, 8, 7, 6, 5, 4, 3, 2, 1, 0};
int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cin >> n;
    for (int i = 0; i < n; i++)
    {
        cin >> c;
        v.push_back(c);
    }

    for (int i = 0; i < n; i++)
    {
        if (v[i] == '>')
        {
            for (int j = 0; j <= n; j++)
            {
                if (down[j] != -1)
                {
                    cout << down[j];
                    down[j] = -1;
                    break;
                }
            }
        }

        else if (v[i] == '<')
        {
            int cnt2 = 1;
            while (1)
            {
                if (v[i + 1] == '<')
                {
                    cnt2++; //부등호 개수
                    i++;
                }

                else
                    break;
            }
            for (int j = 0; j <= n; j++)
            {
                if (down[j] != -1)
                {
                    for (int k = j + cnt2; k > j; k--)
                    {
                        cout << down[k];
                        down[k] = -1;
                    }
                    break;
                }
            }
        }
    }

    for (int l = 0; l <= n; l++)
    {
        if (down[l] != -1)
            cout << down[l] << "\n";
    }

    for (int i = 0; i < n; i++)
    {
        if (v[i] == '<')
        {
            for (int j = 0; j <= n; j++)
            {
                if (up[j] != -1)
                {
                    cout << up[j];
                    up[j] = -1;
                    break;
                }
            }
        }

        else if (v[i] == '>')
        {
            int cnt = 1;
            while (1)
            {
                if (v[i + 1] == '>')
                {
                    cnt++; //부등호 개수
                    i++;
                }

                else
                    break;
            }
            for (int j = 0; j <= n; j++)
            {
                if (up[j] != -1)
                {
                    for (int k = j + cnt; k > j; k--)
                    {
                        cout << up[k];
                        up[k] = -1;
                    }
                    break;
                }
            }
        }
    }

    for (int l = 0; l <= n; l++)
    {
        if (up[l] != -1)
            cout << up[l] << "\n";
    }
}