글 읽기 - 반복되는 while문을 축약해서 쓸 수 있는 방법?

franj 4년 전 0

여러번 반복되는 while문을 typedef나 define으로 한줄로 쓰는 방법이 있을까요?

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

#define FUN while (1) {
	y += dy[dir]; x += dx[dir];
	if (y < 0 || y >= N || x < 0 || x >= M) break;
	if (map[y][x] == 6) break;
	cpy_map[y][x] = 9;
}
int N, M;
int map[10][10];
int cpy_map[10][10];
struct pos {
	int y, x, cctv;
};
vector <pos> V;
int cctv[10];
int dx[4] = { 1, 0 , -1, 0};
int dy[4] = { 0, 1, 0, -1 };
int ans = 1e9;

void Looking() {
	for (int i = 0; i < V.size(); i++)
	{
		int y = V[i].y; int x = V[i].x;
		int dir = cctv[i];
		switch (V[i].cctv) {
		case(1):
			while (1) {
				y += dy[dir]; x += dx[dir];
				if (y < 0 || y >= N || x < 0 || x >= M) break;
				if (map[y][x] == 6) break;
				cpy_map[y][x] = 9;
			}			
			break;
		case(2):
			while (1) {
				y += dy[dir]; x += dx[dir];
				if (y < 0 || y >= N || x < 0 || x >= M) break;
				if (map[y][x] == 6) break;
				cpy_map[y][x] = 9;
			}
			y = V[i].y; x = V[i].x;
			dir = (dir + 2) % 4;
			while (1) {
				y += dy[dir]; x += dx[dir];
				if (y < 0 || y >= N || x < 0 || x >= M) break;
				if (map[y][x] == 6) break;
				cpy_map[y][x] = 9;
			}
			break;
		case(3):
			while (1) {
				y += dy[dir]; x += dx[dir];
				if (y < 0 || y >= N || x < 0 || x >= M) break;
				if (map[y][x] == 6) break;
				cpy_map[y][x] = 9;
			}
			dir = (dir + 1) % 4;
			y = V[i].y; x = V[i].x;
			while (1) {
				y += dy[dir]; x += dx[dir];
				if (y < 0 || y >= N || x < 0 || x >= M) break;
				if (map[y][x] == 6) break;
				cpy_map[y][x] = 9;
			}
			break;
		case(4):
			while (1) {
				y += dy[dir]; x += dx[dir];
				if (y < 0 || y >= N || x < 0 || x >= M) break;
				if (map[y][x] == 6) break;
				cpy_map[y][x] = 9;
			}
			dir = (dir + 1) % 4;
			y = V[i].y; x = V[i].x;
			while (1) {
				y += dy[dir]; x += dx[dir];
				if (y < 0 || y >= N || x < 0 || x >= M) break;
				if (map[y][x] == 6) break;
				cpy_map[y][x] = 9;
			}
			dir = (dir + 1) % 4;
			y = V[i].y; x = V[i].x;
			while (1) {
				y += dy[dir]; x += dx[dir];
				if (y < 0 || y >= N || x < 0 || x >= M) break;
				if (map[y][x] == 6) break;
				cpy_map[y][x] = 9;
			}
			break;
		case(5):
			while (1) {
				y += dy[dir]; x += dx[dir];
				if (y < 0 || y >= N || x < 0 || x >= M) break;
				if (map[y][x] == 6) break;
				cpy_map[y][x] = 9;
			}
			dir = (dir + 1) % 4;
			y = V[i].y; x = V[i].x;
			while (1) {
				y += dy[dir]; x += dx[dir];
				if (y < 0 || y >= N || x < 0 || x >= M) break;
				if (map[y][x] == 6) break;
				cpy_map[y][x] = 9;
			}
			dir = (dir + 1) % 4;
			y = V[i].y; x = V[i].x;
			while (1) {
				y += dy[dir]; x += dx[dir];
				if (y < 0 || y >= N || x < 0 || x >= M) break;
				if (map[y][x] == 6) break;
				cpy_map[y][x] = 9;
			}
			dir = (dir + 1) % 4;
			y = V[i].y; x = V[i].x;
			while (1) {
				y += dy[dir]; x += dx[dir];
				if (y < 0 || y >= N || x < 0 || x >= M) break;
				if (map[y][x] == 6) break;
				cpy_map[y][x] = 9;
			}
			break;
		}
	}
}

void DFS(int cur) {
	if (cur == V.size()) {
		for (int i = 0; i < N; i++)
			for (int j = 0; j < M; j++)
				cpy_map[i][j] = map[i][j];
		Looking();
		int cnt = 0;
		for (int i = 0; i < N; i++){
			for (int j = 0; j < M; j++) {
				if (cpy_map[i][j] == 0) cnt++;
			}
		}
		ans = min(ans, cnt);
		return;
	}


	for (int i = 0; i < 4; i++)
	{
		cctv[cur] = i;
		DFS(cur + 1);
		//cctv[cur]--;
	}

	
}

int main() {
	cin >> N >> M;
	for (int i = 0; i < N; i++)
		for (int j = 0; j < M; j++) {
			cin >> map[i][j];
			if (map[i][j] != 6 && map[i][j] != 0) {
				V.push_back({ i, j, map[i][j] });
			}
		}
	/*for (int i = 0; i < V.size(); i++)
	{
		if (V[i].cctv == 1 || V[i].cctv == 3 || V[i].cctv == 4) {
			cctv[i] = 4;
		}
		if (V[i].cctv == 2) {
			cctv[i] = 2;
		}
		if (V[i].cctv == 5) {
			cctv[i] = 5;
		}
	}*/

	fill_n(cctv, 10, 0);
	DFS(0);

	cout << ans;
}

franj 4년 전 0

#define FUN while(1){\
y += dy[dir]; x += dx[dir];\
if (y < 0 || y >= N || x < 0 || x >= M) break;\
if (map[y][x] == 6) break;\
cpy_map[y][x] = 9;\
}

으로 해결되네요! ㅇㅅㅇ님 감사합니다

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

문제

문제

출처

ICPC

반복되는 while문을 축약해서 쓸 수 있는 방법?

franj 4년 전 0

franj 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말