BOJ 17143. 낚시왕

🙇‍♀️[Gold I] 낚시왕 - 17143

문제 링크

성능 요약

메모리: 2352 KB, 시간: 152 ms

분류

구현, 시뮬레이션

제출 일자

2024년 2월 17일 04:06:56

문제 설명

낚시왕이 상어 낚시를 하는 곳은 크기가 R×C인 격자판으로 나타낼 수 있다. 격자판의 각 칸은 (r, c)로 나타낼 수 있다. r은 행, c는 열이고, (R, C)는 아래 그림에서 가장 오른쪽 아래에 있는 칸이다. 칸에는 상어가 최대 한 마리 들어있을 수 있다. 상어는 크기와 속도를 가지고 있다.

낚시왕은 처음에 1번 열의 한 칸 왼쪽에 있다. 다음은 1초 동안 일어나는 일이며, 아래 적힌 순서대로 일어난다. 낚시왕은 가장 오른쪽 열의 오른쪽 칸에 이동하면 이동을 멈춘다.

1. 낚시왕이 오른쪽으로 한 칸 이동한다.
2. 낚시왕이 있는 열에 있는 상어 중에서 땅과 제일 가까운 상어를 잡는다. 상어를 잡으면 격자판에서 잡은 상어가 사라진다.
3. 상어가 이동한다.

상어는 입력으로 주어진 속도로 이동하고, 속도의 단위는 칸/초이다. 상어가 이동하려고 하는 칸이 격자판의 경계를 넘는 경우에는 방향을 반대로 바꿔서 속력을 유지한채로 이동한다.

왼쪽 그림의 상태에서 1초가 지나면 오른쪽 상태가 된다. 상어가 보고 있는 방향이 속도의 방향, 왼쪽 아래에 적힌 정수는 속력이다. 왼쪽 위에 상어를 구분하기 위해 문자를 적었다.

상어가 이동을 마친 후에 한 칸에 상어가 두 마리 이상 있을 수 있다. 이때는 크기가 가장 큰 상어가 나머지 상어를 모두 잡아먹는다.

낚시왕이 상어 낚시를 하는 격자판의 상태가 주어졌을 때, 낚시왕이 잡은 상어 크기의 합을 구해보자.

입력

첫째 줄에 격자판의 크기 R, C와 상어의 수 M이 주어진다. (2 ≤ R, C ≤ 100, 0 ≤ M ≤ R×C)

둘째 줄부터 M개의 줄에 상어의 정보가 주어진다. 상어의 정보는 다섯 정수 r, c, s, d, z (1 ≤ r ≤ R, 1 ≤ c ≤ C, 0 ≤ s ≤ 1000, 1 ≤ d ≤ 4, 1 ≤ z ≤ 10000) 로 이루어져 있다. (r, c)는 상어의 위치, s는 속력, d는 이동 방향, z는 크기이다. d가 1인 경우는 위, 2인 경우는 아래, 3인 경우는 오른쪽, 4인 경우는 왼쪽을 의미한다.

두 상어가 같은 크기를 갖는 경우는 없고, 하나의 칸에 둘 이상의 상어가 있는 경우는 없다.

출력

낚시왕이 잡은 상어 크기의 합을 출력한다.

🚀풀이

문제의 포인트는 시간초과가 안나게 풀어야한다는 것이다.

상어가 이동하는게 최소한으로 움직이게 만들어야한다.

상어의 속도가 매우 높아서 맵을 반복해서 움직인다면 중복되는 값을 없앨수있다.

R, C 보드에서 상어가 반복되는걸 없애려면
s = s % ((R - 1) * 2) 로 방향이 R라인일 때 이렇게 고칠수 있고 마찬가지로 C 방향이라면
s = s % ((C - 1) * 2) 로 고칠 수 있다.

그리고 상어가 다 움직이고 상어를 잡아먹는다면 시간초과가 난다.

M은 최대 10000이라서 만번 * 만번을 돌면 시간이 너무 오래걸린다.

그래서 상어가 이동하자마자 먹는 것을 해야하는데 만난 상어가 이동한 후 상어인지 이동 전 상어인지 알아야한다.

// 이동을 완료한 상어인지 아닌지 판별해야함
if (board[y][x] == 0)
{
    board[y][x] = i;
    sharks[i].r = y;
    sharks[i].c = x;
}
else if (board[y][x] < i)
{
    if (sharks[board[y][x]].z > sharks[i].z)
    {
        removeShark(i);
    }
    else
    {
        removeShark(board[y][x]);
        board[y][x] = i;
        sharks[i].r = y;
        sharks[i].c = x;
    }
}

그래서 각 상어마다 인덱스가 있으므로 그것을 통해서 판별했다.

상어가 움직이는 코드는 아래와 같다.

void moveShark()
{
	for (int i = 0; i < sharks.size(); ++i)
	{
		if (sharks[i].r == 0)
			continue;

		int y = sharks[i].r;
		int x = sharks[i].c;
		//board[y][x] = 0; // 자리를 떠났다.

		for (int j = 0; j < sharks[i].s; ++j)
		{
			int ny = y + dy[sharks[i].d];
			int nx = x + dx[sharks[i].d];

			if (ny < 1 || nx < 1 || ny > R || nx > C)
			{
				if (sharks[i].d == 1 || sharks[i].d == 3)
				{
					sharks[i].d++;
					ny = y + dy[sharks[i].d];
					nx = x + dx[sharks[i].d];
				}
				else if (sharks[i].d == 2 || sharks[i].d == 4)
				{
					sharks[i].d--;
					ny = y + dy[sharks[i].d];
					nx = x + dx[sharks[i].d];
				}
			}

			y = ny;
			x = nx;
		}

		// 이동을 완료한 상어인지 아닌지 판별해야함
		if (board[y][x] == 0)
		{
			board[y][x] = i;
			sharks[i].r = y;
			sharks[i].c = x;
		}
		else if (board[y][x] < i)
		{
			if (sharks[board[y][x]].z > sharks[i].z)
			{
				removeShark(i);
			}
			else
			{
				removeShark(board[y][x]);
				board[y][x] = i;
				sharks[i].r = y;
				sharks[i].c = x;
			}
		}
	}

	//printBoard();

	for (int i = 0; i < sharks.size(); ++i)
	{
		board[sharks[i].r][sharks[i].c] = 0;
	}
}

마지막에 상어의 인덱스를 다 지워서 흔적을 남기지 않게 만들었다.

상어를 없애는 건 상어의 정보를 모두 0으로 만들었다.

왜냐하면 vector로 상어를 담고 있는데 정말로 지우게 된다면 인덱스 문제도 있을 뿐만아니라 복사 과정이 있으므로 시간이 오래 걸리기때문이다.

🚀전체 코드

#define _CRT_SECURE_NO_WARNINGS

#include <stdio.h>
#include <string.h>
#include <ctype.h>
#include<iostream>
#include <fstream>
#include <vector>
#include <algorithm>
#include <math.h>

using namespace std;

struct Shark
{
	int r, c, s, d, z;

	bool operator<(const Shark other)
	{
		if (r < other.r)
			return true;
		return false;
	}
};

void setting();
void solve();
void moveShark();
void removeShark(int idx);
void printBoard();

// 1. 낚시왕이 오른쪽으로 한 칸 이동
// 2. 낚시왕이 있는 열에 있는 상어 중에서 땅과 제일 가까운 상어를 잡는다. 상어를 잡으면 격자판에서 잡은 상어가 사라진다.
// 3. 상어가 이동한다.

int R, C, M, r, c, s, d, z, res = 0;
int dy[] = { 0, -1, 1, 0, 0 };
int dx[] = { 0, 0, 0, 1, -1 };
vector<vector<int>> board;
vector<Shark> sharks;

void setting()
{
	cin >> R >> C >> M;
	board = vector<vector<int>>(R + 1, vector<int>(C + 1));
	sharks = vector<Shark>(M + 1);

	sharks[0].r = 0;
	sharks[0].c = 0;
	sharks[0].s = 0;
	sharks[0].d = 0;
	sharks[0].z = 0;

	for (int i = 1; i <= M; ++i)
	{
		cin >> r >> c >> s >> d >> z;
		sharks[i].r = r;
		sharks[i].c = c;
		sharks[i].s = s;
		sharks[i].d = d;
		sharks[i].z = z;

		if (d <= 2)
		{
			sharks[i].s = s % ((R - 1) * 2);
		}
		else if (d >= 3)
		{
			sharks[i].s = s % ((C - 1) * 2);
		}
	}
}

void moveShark()
{
	for (int i = 0; i < sharks.size(); ++i)
	{
		if (sharks[i].r == 0)
			continue;

		int y = sharks[i].r;
		int x = sharks[i].c;
		//board[y][x] = 0; // 자리를 떠났다.

		for (int j = 0; j < sharks[i].s; ++j)
		{
			int ny = y + dy[sharks[i].d];
			int nx = x + dx[sharks[i].d];

			if (ny < 1 || nx < 1 || ny > R || nx > C)
			{
				if (sharks[i].d == 1 || sharks[i].d == 3)
				{
					sharks[i].d++;
					ny = y + dy[sharks[i].d];
					nx = x + dx[sharks[i].d];
				}
				else if (sharks[i].d == 2 || sharks[i].d == 4)
				{
					sharks[i].d--;
					ny = y + dy[sharks[i].d];
					nx = x + dx[sharks[i].d];
				}
			}

			y = ny;
			x = nx;
		}

		// 이동을 완료한 상어인지 아닌지 판별해야함
		if (board[y][x] == 0)
		{
			board[y][x] = i;
			sharks[i].r = y;
			sharks[i].c = x;
		}
		else if (board[y][x] < i)
		{
			if (sharks[board[y][x]].z > sharks[i].z)
			{
				removeShark(i);
			}
			else
			{
				removeShark(board[y][x]);
				board[y][x] = i;
				sharks[i].r = y;
				sharks[i].c = x;
			}
		}
	}

	//printBoard();

	for (int i = 0; i < sharks.size(); ++i)
	{
		board[sharks[i].r][sharks[i].c] = 0;
	}
}

void removeShark(int idx)
{
	sharks[idx].r = 0;
	sharks[idx].c = 0;
	sharks[idx].d = 0;
	sharks[idx].s = 0;
	sharks[idx].z = 0;
}

void printBoard()
{
	for (int i = 1; i <= R; ++i)
	{
		for (int j = 1; j <= C; ++j)
		{
			cout << board[i][j] << " ";
		}

		cout << endl;
	}

	cout << endl;
}

void solve()
{
	setting();
	// 주인공은 1부터 C까지 이동
	for (int pos = 1; pos <= C; ++pos)
	{
		// 상어 잡기
		pair<int, int> temp = { 1000, 1000 }; // 인덱스, r값
		for (int k = 0; k < sharks.size(); ++k)
		{
			if (sharks[k].c == pos)
			{
				if (sharks[k].r < temp.second)
				{
					temp.first = k;
					temp.second = sharks[k].r;
				}
			}
		}

		if (temp.first != 1000)
		{
			res += sharks[temp.first].z;
			removeShark(temp.first);
		}

		moveShark();
	}

	cout << res;
}

int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);
	//freopen("input.txt", "rt", stdin);

	solve();

	return 0;
}