취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 2508번 문제인 사탕 박사 고창영이다.
문제는 아래 링크를 확인하자.

https://www.acmicpc.net/problem/2508 

주어지는 행렬에서 "행 방향으로 인접한 세 문자가 '>' 'o' '<'인 위치의 개수" 및 "열 방향으로 인접한 세 문자가 'v' 'o' '^'인 위치의 개수"를 직접 세어 문제를 해결할 수 있다.

살펴볼 인접한 위치의 개수는 $O(RC)$개이고, 각 위치를 살펴볼 때 한번에 3개의 원소($O(1)$)만을 살펴보면 되므로 이와 같은 해법의 시간복잡도는 $O(RC)$와 같다.

아래는 제출한 소스코드이다.

#include <iostream>
using namespace std;

int T;
int R, C;
string board[400];

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	cin >> T;
	while (T--) {
		int ans = 0;
		cin >> R >> C;
		for (int r = 0; r < R; r++) cin >> board[r];

		for (int r = 0; r < R; r++) {
			for (int c = 0; c + 2 < C; c++) {
				if (board[r][c] == '>' && board[r][c + 1] == 'o' && board[r][c + 2] == '<') ans++;
			}
		}

		for (int c = 0; c < C; c++) {
			for (int r = 0; r + 2 < R; r++) {
				if (board[r][c] == 'v' && board[r + 1][c] == 'o' && board[r + 2][c] == '^') ans++;
			}
		}

		cout << ans << '\n';
	}
}