[BOJ] 2816. 디지털 티비

#include <iostream>
#include <vector>
#include <cstring>
using namespace std;

int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    
    int n;
    cin >> n;
    
    vector<string> v;
    int idx_1, idx_2;
    
    for(int i=0; i<n; i++){
        string s;
        cin >> s;
        if(s == "KBS1")
            idx_1 = i;
        else if( s == "KBS2")
            idx_2 = i;
    }
    
    if( idx_2 < idx_1 )
        idx_2 ++;

    for(int i=0; i<idx_1; i++) cout << "1";
    for(int i=0; i<idx_1; i++) cout << "4";
    
    for(int i=0; i<idx_2; i++) cout << "1";
    for(int i=0; i<idx_2-1; i++) cout << "4";
    
    return 0;
}