백준(BOJ) 16956번 늑대와 양

문제 : https://www.acmicpc.net/problem/16956

16956번: 늑대와 양

풀이 :

 

1. S의 상하좌우의 W가 있다면 울타리를 만들 수 없으니 0을 출력할 수밖에 없다.

 

2. S의 상하좌우의 W인 곳이 없다면 모두 울타리를 만들 수 있으니 1을 출력하고 임의대로 울타리를

 

만들어주자. 많은 답이 있겠지만 나의 경우는 상하좌우에 울타리를 둘러주었다.

 

코드( C++ )

	#include <iostream>
	#include <string>
	using namespace std;
	int r, c;
	string a[500];
	const int dy[4] = { 1,-1,0,0 };
	const int dx[4] = { 0,0,1,-1 };
	int main()
	{
	cin >> r >> c;
	for (int i = 0; i < r; ++i)
	cin >> a[i];
	bool isSheep = false;
	for (int i = 0; i < r; ++i)
	for (int j = 0; j < c; ++j)
	{
	if (a[i][j] == 'S') {
	// S 상하좌우에 양이 있다면 0을 출력해야 한다.
	for (int k = 0; k < 4; ++k) {
	int ny = i + dy[k];
	int nx = j + dx[k];
	if (!(0 <= ny && ny < r && 0 <= nx && nx < c)) continue;
	if (a[ny][nx] == 'W')
	isSheep = true;
	}
	}
	}
	if (isSheep)
	cout << 0 << "\n";
	else {
	cout << 1 << "\n";
	for (int i = 0; i < r; ++i)
	for (int j = 0; j < c; ++j) {
	if (a[i][j] == 'S') {
	for (int k = 0; k < 4; ++k) {
	int ny = i + dy[k];
	int nx = j + dx[k];
	if (!(0 <= ny && ny < r && 0 <= nx && nx < c)) continue;
	if (a[ny][nx] == '.')
	a[ny][nx] = 'D';
	}
	}
	}
	for (int i = 0; i < r; ++i)
	cout << a[i] << "\n";
	}
	return 0;
	}

view raw .cpp hosted with ❤ by GitHub