인프런 커뮤니티 질문&답변

안녕하세요 bunny님ㅎㅎ

이거 확인해봤는데 그렇게 불필요한 로직은 없고 걸리는 건 2가지 인것 같은데 1. eat_shark 2. pick_shark 부분입니다.

이거 주석달았으니 참고 부탁드리구요. eat_shark 답변 부탁드려요

나머지 부분은 잘 짜셨어요 ㅎㅎ

shark[visited[shark[i][0]][shark[i][1]]][4]

이부분은 좀 그렇긴 하지만요. 근데 이거 for(int i : 이렇게 하려고 해도 밑에 shark 인덱스를 기반으로 수정하려고 어쩔 수 없이 한 것같은데... 좀 고민이 필요한 코드인 것 같긴해요.

#include <iostream>
#include <vector>
#include <memory.h>

using namespace std;

int R, C, m;
int r, c, s, d, z;
vector<vector<int>> shark;
// 그 곳에 있는 상어 중 사이즈가 가장 큰 상어의 인덱스
int visited[104][104];
int deep[104];
int ans;
int now_pos;

// 상어 움직임
void move_shark()
{
	for (int i = 0; i < shark.size(); i++)
	{
		// 죽은 상어 패스
		if (shark[i][5] == 1) continue;

		// 상어 움직임 // 스피드
		for (int j = 0; j < shark[i][2]; j++)
		{
			//이동방향
			if (shark[i][3] == 1)
			{
				if (shark[i][0] != 1) shark[i][0] -= 1;
				else
				{
					shark[i][0] += 1;
					shark[i][3] = 2;
				}
			}
			else if (shark[i][3] == 2)
			{
				if (shark[i][0] != R) shark[i][0] += 1;
				else 
				{
					shark[i][0] -= 1;
					shark[i][3] = 1;
				}

			}
			else if (shark[i][3] == 3)
			{
				if (shark[i][1] != C) shark[i][1] += 1;
				else
				{
					shark[i][1] -= 1;
					shark[i][3] = 4;
				}
			}
			else
			{
				if (shark[i][1] != 1) shark[i][1] -= 1;
				else
				{
					shark[i][1] += 1;
					shark[i][3] = 3;
				}
			}
		}


	}

	return;
}

// 상어 잡아먹음
void eat_shark()
{
	// 이동하기전 샤크와 이동한 후의 샤크를 어떻게 구분?
	// 이동하기 전 >> visited 로 걸고
	// 이동 한후 >> visited와 shark를 비교해야 하는 거 아님? 
	memset(visited, -1, sizeof(visited)); 
	for (int i = 0; i < shark.size(); i++)
	{
		// 죽은 상어 패스
		if (shark[i][5] == 1) continue;

		// 아무 상어도 없어
		if (visited[shark[i][0]][shark[i][1]] == -1)
		{
			visited[shark[i][0]][shark[i][1]] = i;
		}
		// 어떤 상어가 있어
		else
		{
			// 원래가 있던 상어가 커
			if (shark[visited[shark[i][0]][shark[i][1]]][4] > shark[i][4])
			{
				// 들어온 상어 죽음
				shark[i][5] = 1;
			}
			// 지금이 커
			else
			{
				// 원래 있던 상어 죽음
				shark[visited[shark[i][0]][shark[i][1]]][5] = 1;

				visited[shark[i][0]][shark[i][1]] = i;
			}
		}
	}

	return;

}

// 상어 줍기
void pick_shark(int now_pos)
{
	//조금 더 단순화 시켜보자. 제 코드 참고해주세요.
	memset(deep, -1, sizeof(deep)); 
	// 그 곳에 있는 상어의 인덱스
	for (int i = 0; i < shark.size(); i++)
	{
		// 죽은 상어 패스
		if (shark[i][5] == 1) continue;

		// i로 만들어놓고. 
		//deep 에 r을 담는다. 두개 이상의 상어가 있는 경우는 x
		if (shark[i][1] == now_pos)
		{
			deep[shark[i][0]] = i;
		}
	} 
	//담은 것중에서 + 죽은 상어 break  
	for (int i = 1; i <= R; i++)
	{
		if (deep[i] > -1)
		{
			ans += shark[deep[i]][4];

			// 주운 상어
			shark[deep[i]][5] = 1;
			break;
		}
	}

	return;
}


int main()
{
	cin >> R >> C >> m;
	//good
	for (int i = 0; i < m; i++)
	{
		cin >> r >> c >> s >> d >> z;
		vector<int> v = { r,c,s,d,z,0 };

		shark.push_back(v);

	}

	for (int i = 1; i <= C; i++)
	{
		now_pos = i;
		pick_shark(now_pos);
		move_shark();
		eat_shark();

	}

	cout << ans << "\n";

	return 0;
}