By 정우석 in Problem Solving —

싸움 땅 [2022-하반기 오전 1번 문제]

문제 링크

삼성 코딩테스트 기출 문제: 싸움땅 - 코드트리
주요 IT 기업의 코딩테스트 기출 문제를 연습하며 실전 감각을 키우세요. 취업 합격률을 높이는 필수 연습 과정입니다.
코드트리

문제 설명

n*n 크기의 격자에 무기들이 있을 수 있음. 초기에 무기들이 없는 빈 격자에 플레이어들이 위치하고, 플레이어들은 초기 능력치를 가짐. 각 플레이어의 초기 능력치는 모두 다름.

  1. 첫 번째 플레이어부터 순차적으로 본인이 향하는 방향대로 한 칸 이동. 격자를 벗어나면 정반대 방향으로 방향을 바꾸어서 1만큼 이동
  2. 이동한 방향에 플레이어가 없다면 총이 있는지 확인. 본인이 가지고 있는 총과 총들 중 가장 쎈 총을 획득하고 나머지는 격자에 놔둠
  3. 이동한 방향에 플레이어가 있다면 플레이어가 싸움. 초기능력치+가지고 있는 총이 큰 사람이 승자. 같다면 초기 능력치가 높은 사람이 승자. 수치가 같은 경우 초기 능력치가 높은 플레이어가 승자. 각 플레이어의 초기 능력치와 가지고 있는 총 공격력의 합의 차이만큼 포인트 획득
  4. 진 플레이어는 가지고 있는 총을 격자에 내려놓고 원래 방향으로 한 칸 이동. 해당 위치에 사람이 있거나 격자 범위 밖이면 90도 회전해서 이동. 해당 위치에 총이 있다면 가장 쎈 총을 획득.
  5. 이긴 플레이어는 가장 쎈 총을 획득하고 나머지는 내려놓음.

해당 과정을 n번해서 각 플레이어들이 얻는 포인트 획득.

해결 방법

패배자가 총을 버리고, 움직이기. 승자는 총을 줍기

void lose_player_move(int p) {
	int nx, ny;
	for (int i = 0;i < 4;i++) {
		int d = player[p].d;
		nx = player[p].x + dx[d];
		ny = player[p].y + dy[d];

		if (nx < 0 || ny < 0 || nx >= n || ny >= n) {
			player[p].d = (player[p].d + 1) % 4;
			continue;
		}

		bool chk_player = false;
		for (int j = 0;j < m;j++) {
			if (nx == player[j].x && ny == player[j].y) {
				chk_player = true;
			}
		}
		if (chk_player) {
			player[p].d = (player[p].d + 1) % 4;
			continue;
		}	
		break;
	}

	player[p].x = nx, player[p].y = ny;
	root_gun(p);
}

void lose_player_drop_gun(int p) {
	int x, y;
	x = player[p].x, y = player[p].y;
	bool dropped = false;

	if (player[p].gun == 0) return;

	for (int i = 0;i < arr[x][y].size();i++) {
		if (arr[x][y][i] == 0) {
			swap(arr[x][y][i], player[p].gun);
			dropped = true;
			break;
		}
	}
	if (!dropped) {
		int gun = player[p].gun;
		arr[x][y].push_back(gun);
		player[p].gun = 0;
	}

	return;
}

있다면 결투

void player_fight(int a, int b) {
	int gun_a, gun_b, s_a, s_b;
	gun_a = player[a].gun;
	gun_b = player[b].gun;
	s_a = player[a].s;
	s_b = player[b].s;

	if (gun_a + s_a > gun_b + s_b) {
		// b가 패배
		player[a].point += gun_a + s_a - (gun_b + s_b);
		lose_player_drop_gun(b);
		lose_player_move(b);
		root_gun(a);
	}
	else if (gun_a + s_a < gun_b + s_b) {
		// a가 패배
		player[b].point += gun_b + s_b - (gun_a + s_a);
		lose_player_drop_gun(a);
		lose_player_move(a);
		root_gun(b);
	}
	else {
		if (s_a > s_b) {
			// b가 패배
			player[a].point += gun_a + s_a - (gun_b + s_b);
			lose_player_drop_gun(b);
			lose_player_move(b);
			root_gun(a);
		}
		else {
			// a가 패배
			player[b].point += gun_b + s_b - (gun_a + s_a);
			lose_player_drop_gun(a);
			lose_player_move(a);
			root_gun(b);
		}
	}
}

없다면 해당 위치의 총을 줍기

void root_gun(int p) {
	int x = player[p].x;
	int y = player[p].y;
	int gun = player[p].gun;

	int target = -1;

	for (int i = 0;i < arr[x][y].size();i++) {
		if (arr[x][y][i] > gun) {
			target = i;
			gun = arr[x][y][i];
		}	
	}

	if (target == -1) {
		return;
	}

	swap(player[p].gun, arr[x][y][target]);
	return;
}

이동한 위치에 플레이어가 있는 지 확인

int check_player(int p) {
	int x = player[p].x;
	int y = player[p].y;

	for (int i = 0;i < m;i++) {
		if (i == p) continue;
		if (player[i].x == x && player[i].y == y) {
			return i;
		}
	}

	return -1;
}

플레이어 이동

void player_move(int p) {
	int nx, ny;
	int d = player[p].d;
	nx = player[p].x + dx[d];
	ny = player[p].y + dy[d];
	if (nx < 0 || ny < 0 || nx >= n || ny >= n) {
		d = (d + 2) % 4;
		nx = player[p].x + dx[d];
		ny = player[p].y + dy[d];
		player[p].d = d;
	}

	player[p].x = nx, player[p].y = ny;
}

전체 코드

#include <iostream>
#include <cstring>
#include <queue>
#include <tuple>
#include <climits>
using namespace std;

typedef struct {
	int x, y, d, s;
	int point;
	int gun;
}_player;

int n, m, k;
vector<int> arr[21][21];
_player player[31];

int dx[4] = { -1,0,1,0 };
int dy[4] = { 0,1,0,-1 };

void input() {
	cin >> n >> m >> k;

	int tmp;
	for (int i = 0;i < n;i++) {
		for (int j = 0;j < n;j++) {
			cin >> tmp;
			arr[i][j].push_back(tmp);
		}
	}
	
	for (int i = 0;i < m;i++) {
		cin >> player[i].x >> player[i].y >> player[i].d >> player[i].s;
		player[i].x--, player[i].y--;
		player[i].point = 0;
		player[i].gun = 0;
	}
}

void player_move(int p) {
	int nx, ny;
	int d = player[p].d;
	nx = player[p].x + dx[d];
	ny = player[p].y + dy[d];
	if (nx < 0 || ny < 0 || nx >= n || ny >= n) {
		d = (d + 2) % 4;
		nx = player[p].x + dx[d];
		ny = player[p].y + dy[d];
		player[p].d = d;
	}

	player[p].x = nx, player[p].y = ny;
}

int check_player(int p) {
	int x = player[p].x;
	int y = player[p].y;

	for (int i = 0;i < m;i++) {
		if (i == p) continue;
		if (player[i].x == x && player[i].y == y) {
			return i;
		}
	}

	return -1;
}

void root_gun(int p) {
	int x = player[p].x;
	int y = player[p].y;
	int gun = player[p].gun;

	int target = -1;

	for (int i = 0;i < arr[x][y].size();i++) {
		if (arr[x][y][i] > gun) {
			target = i;
			gun = arr[x][y][i];
		}	
	}

	if (target == -1) {
		return;
	}

	swap(player[p].gun, arr[x][y][target]);
	return;
}

void lose_player_move(int p) {
	int nx, ny;
	for (int i = 0;i < 4;i++) {
		int d = player[p].d;
		nx = player[p].x + dx[d];
		ny = player[p].y + dy[d];

		if (nx < 0 || ny < 0 || nx >= n || ny >= n) {
			player[p].d = (player[p].d + 1) % 4;
			continue;
		}

		bool chk_player = false;
		for (int j = 0;j < m;j++) {
			if (nx == player[j].x && ny == player[j].y) {
				chk_player = true;
			}
		}
		if (chk_player) {
			player[p].d = (player[p].d + 1) % 4;
			continue;
		}	
		break;
	}

	player[p].x = nx, player[p].y = ny;
	root_gun(p);
}

void lose_player_drop_gun(int p) {
	int x, y;
	x = player[p].x, y = player[p].y;
	bool dropped = false;

	if (player[p].gun == 0) return;

	for (int i = 0;i < arr[x][y].size();i++) {
		if (arr[x][y][i] == 0) {
			swap(arr[x][y][i], player[p].gun);
			dropped = true;
			break;
		}
	}
	if (!dropped) {
		int gun = player[p].gun;
		arr[x][y].push_back(gun);
		player[p].gun = 0;
	}

	return;
}

void player_fight(int a, int b) {
	int gun_a, gun_b, s_a, s_b;
	gun_a = player[a].gun;
	gun_b = player[b].gun;
	s_a = player[a].s;
	s_b = player[b].s;

	if (gun_a + s_a > gun_b + s_b) {
		// b가 패배
		player[a].point += gun_a + s_a - (gun_b + s_b);
		lose_player_drop_gun(b);
		lose_player_move(b);
		root_gun(a);
	}
	else if (gun_a + s_a < gun_b + s_b) {
		// a가 패배
		player[b].point += gun_b + s_b - (gun_a + s_a);
		lose_player_drop_gun(a);
		lose_player_move(a);
		root_gun(b);
	}
	else {
		if (s_a > s_b) {
			// b가 패배
			player[a].point += gun_a + s_a - (gun_b + s_b);
			lose_player_drop_gun(b);
			lose_player_move(b);
			root_gun(a);
		}
		else {
			// a가 패배
			player[b].point += gun_b + s_b - (gun_a + s_a);
			lose_player_drop_gun(a);
			lose_player_move(a);
			root_gun(b);
		}
	}
}

void solve(int t) {
	for (int i = 0;i < m;i++) {
		player_move(i);
		// 플레이어가 있는지 확인
		int enemy_player = check_player(i);
		if (enemy_player == -1) {
			root_gun(i);
		}
		else {
			player_fight(i, enemy_player);
		}
	}
}

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

//	freopen("sample_input.txt", "r", stdin);

	input();
	for (int i = 0;i < k;i++) {
		solve(i);
	}

	for (int i = 0;i < m;i++) {
		cout << player[i].point << " ";
	}
}
Previous

코드트리 빵 [2022-하반기 오후 1번 문제]

 Next

나무 박멸 [2022-상반기 오후 2번 문제]

You might also like...