2816 디지털 티비

#include <stdio.h>
#include <string>
#include <iostream>
#include <vector>
 
using namespace std;
 
vector<int> arr;
 
int main() {
    int n;
    scanf("%d", &n);
 
    while (n--) {
        string str;
        cin >> str;
 
        if (str == "KBS1") {
            arr.push_back(1);
        }
        else if (str == "KBS2") {
            arr.push_back(2);
        }
        else {
            arr.push_back(0);
        }
    }
 
    int zero = 0, flag = 0, tmp = 0;
    for (int i = 0; i < arr.size(); i++) {
        if (flag == 2) {
            if (tmp == 1)
                printf("4");
 
            break;
        }
 
        if (arr[i] == 1) {
            for (int j = 0; j < zero; j++) {
                printf("4");
            }
            
            arr[i] = 0;
 
            zero = 0;
            i = flag;
            flag++;
            tmp = 1;
 
            if (flag != 2)
                printf("1");
        }
        else if (arr[i] == 2) {
            for (int j = 0; j < zero; j++) {
                printf("4");
            }
 
            arr[i] = 0;
 
            zero = 0;
            i = flag;
            flag++;
            tmp = 2;
 
            if (flag != 2)
                printf("1");
        }
        else {
            printf("1");
            zero++;
        }
    }
}