글 읽기 - 어느 부분에서 틀린 것인지 잘 모르겠습니다...

fantasy7772 2년 전 0

N과 M(가로, 세로)크기도 순서에 맞게 입력 받았는데 어디서 틀렸는지 잘 모르겠습니다ㅠㅠ 도움 주시면 감사하겠습니다!!

#include <iostream>
#include <vector>
#include <algorithm>
#include <stack>
#include <cmath>
#include <queue>
#include <cstring>
#include <string>
#include <deque>
#include <tuple>
#include <set>
using namespace std;
const int INF = 0x3f3f3f3f;
char** arr;
bool visited[101][101];
queue<pair<int, int>> q;
int N, K, W_result, B_result;
int cnt = 1;
int ym[4] = { -1, 1, 0, 0 };
int xm[4] = { 0, 0, -1, 1 };

bool isin(int y, int x) {
	if (y < 0 || y >= N || x < 0 || x >= K) {
		return false;
	}
	return true;
}

void BFS(char color) {
	while (!q.empty()) {
		int cur_y = q.front().first;
		int cur_x = q.front().second;
		
		q.pop();
		for (int i = 0; i < 4; i++) {
			int nexty = cur_y + ym[i];
			int nextx = cur_x + xm[i];
			if (isin(nexty, nextx) == false)continue;
			if (visited[nexty][nextx] == true)continue;
			if (arr[nexty][nextx] != color)continue;
			visited[nexty][nextx] = true;
			cnt++;
			q.push(make_pair(nexty, nextx));
		}
	}
	if (color == 'W') W_result += cnt * cnt;
	else B_result += cnt * cnt;
}
int main() {
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);
	cin >> N >> K;
	arr = new char* [K];
	memset(visited, false, sizeof(visited));
	for (int i = 0; i < K; i++) {
		arr[i] = new char[N];
	}
	for (int i = 0; i < K; i++) {
		for (int j = 0; j < N; j++) {
			cin >> arr[i][j];
		}
	}
	for (int i = 0; i < K; i++) {
		for (int j = 0; j < N; j++) {
			if (visited[i][j] == false) {
				cnt = 1;
				visited[i][j] = true;
				q.push(make_pair(i, j));
				BFS(arr[i][j]);
			}
		}
	}
	cout << W_result << " " << B_result << endl;

}

xingxing2001 2년 전 0

3 1

하면 입력도 안되고 프로그램이 종료됩니다. 그래서 정적으로 char arr[101][101]로 두고 했으나

위처럼 입력하면 출력이 0 3 으로 나옵니다. (정답: 0 9)

1 3

WWW 도 마찬가지로 틀리게 나옵니다.

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

어느 부분에서 틀린 것인지 잘 모르겠습니다...

fantasy7772 2년 전 0

xingxing2001 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말