[알고리즘] 백준 2816 - 디지털 티비 (C++)

[문제 링크] : https://www.acmicpc.net/problem/2816

#include <iostream>
#include <string>
using namespace std;

int N;
string Arr[101];

int main(){
    cin >> N;
    int idx1 = 0, idx2 = 0;
    for(int i=1; i<=N; i++){
        cin >> Arr[i];
        if(Arr[i] == "KBS1"){
            idx1 = i;
        }
        if(Arr[i] == "KBS2"){
            idx2 = i;
        }
    }

    int x = idx1 - 1;

    for(int i=1; i<=x; i++){
        cout << 1;
    }

    for(int i=1; i<=x; i++){
        cout << 4;
    }

    if(idx1 > idx2){
        idx2 += 1;
    }
    
    x = idx2 - 1;
    
    for(int i=1; i<=x; i++){
        cout << 1;
    }

    for(int i=1; i<=x-1; i++){
        cout << 4;
    }

    return 0;
}