CodingTest/백준

[c++] 백준 - 2816: 디지털 티비

Daybreak21 2023. 5. 20. 20:02

2816번: 디지털 티비 (acmicpc.net)

 

2816번: 디지털 티비

2012년 12월 31일 새벽 4시부터 지상파 아날로그 TV방송이 종료되었다. TV를 자주보는 할머니를 위해서, 상근이네 집도 디지털 수신기를 구입했다. 원래 상근이네 집에는 KBS1과 KBS2만 나왔다. 할머니

www.acmicpc.net

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

int main() {
	int N, index = 0; cin >> N;
	string* channal = new string[N];
	for (int i = 0; i < N; i++) { cin >> channal[i]; }
	
    //KBS1
	while (!(channal[index] == "KBS1")) { index++; }
	if (index == 0) { cout << '1'; }
	else if (index == 1) { cout << '3';	}
	else {
		for (int i = 0; i < index - 1; i++) cout << '3';
		cout << '1';
		for (int i = 0; i < index; i++) cout << '4';
		cout << '1';
	}
	channal[index] = channal[0]; channal[0] = "KBS1";

	//KBS2
	index = 1;
	while (!(channal[index] == "KBS2")) { index++; }
	if (index == 1) { }
	else if (index == 2) { cout << '3'; }
	else {
		for (int i = 0; i < index - 2; i++) cout << '3';
		cout << '1';
		for (int i = 0; i < index - 1; i++) cout << '4';
	}
	channal[index] = channal[0]; channal[0] = "KBS1";
    
	return 0;
}