#include #include #include #define U 1 #define D 2 #define R 3 #define L 4 using namespace std; int N, M, SC; int FishKing_pos = 0; int ans = 0; int cur_time = 0; int dx[] = { 0, -1, 1, 0, 0 }; // 0, U, D, R, L int dy[] = { 0, 0, 0, 1, -1 }; struct Shark { int x, y, speed, dir, size, time; bool operator==(const Shark& a) const { return x == a.x && y == a.y && speed == a.speed && dir == a.dir && size == a.size && time == a.time; } }; Shark map[100][105][105]; // [time][x][y] queue Q; bool inRange(int x, int y) { return 1 <= x && x <= N && 1 <= y && y <= M; } void input() { cin >> N >> M >> SC; for (int i = 0; i < SC; i++) { Shark s; cin >> s.x >> s.y >> s.speed >> s.dir >> s.size; s.time = 0; map[0][s.x][s.y] = s; Q.push(s); } } void catch_shark() { // 1. 현재 열(FishKing_pos)에서 위에서부터 가장 먼저 만나는 상어 탐색 for (int i = 1; i <= N; i++) { Shark& s = map[cur_time][i][FishKing_pos]; if (s.size > 0) { ans += s.size; // 2. 잡힌 상어의 좌표 기억 int tx = s.x, ty = s.y, ttime = s.time; // 3. map에서 제거 s = {0, 0, 0, 0, 0, 0}; // 4. Q에서도 제거 int sz = Q.size(); for (int k = 0; k < sz; k++) { Shark cur = Q.front(); Q.pop(); // 위치, 시간까지 완전 일치하는 상어 제거 if (cur.x == tx && cur.y == ty && cur.time == ttime) continue; Q.push(cur); } break; } } } void dir_calib(int& x, int& y, int& d) { int over; int limit = 10; // 무한 루프 방지 while (!inRange(x, y) && --limit) { if (x < 1) { over = 1 - x; x = 1 + over; d = D; } else if (x > N) { over = x - N; x = N - over; d = U; } else if (y < 1) { over = 1 - y; y = 1 + over; d = R; } else if (y > M) { over = y - M; y = M - over; d = L; } } } void move_shark() { // 다음 시간만 초기화 for (int i = 1; i <= N; i++) { for (int j = 1; j <= M; j++) { map[cur_time + 1][i][j] = {0, 0, 0, 0, 0, 0}; } } int nx, ny, nt; int Size = Q.size(); for (int i = 0; i < Size; i++) { Shark cur = Q.front(); Q.pop(); if (cur.dir < 1 || cur.dir > 4) continue; nt = cur.time + 1; // 속도 최적화: 한 cycle 안에서 반복되는 경우 제거 int effective_speed = cur.speed; if (cur.dir == U || cur.dir == D) effective_speed %= (2 * (N - 1)); else effective_speed %= (2 * (M - 1)); nx = cur.x + dx[cur.dir] * effective_speed; ny = cur.y + dy[cur.dir] * effective_speed; dir_calib(nx, ny, cur.dir); if (map[nt][nx][ny].size > cur.size) continue; Shark moved = {nx, ny, cur.speed, cur.dir, cur.size, nt}; map[nt][nx][ny] = moved; Q.push(moved); } } void solution() { for (FishKing_pos = 1; FishKing_pos <= M; FishKing_pos++) { catch_shark(); move_shark(); cur_time++; } } int main() { ios::sync_with_stdio(false); cin.tie(nullptr); input(); solution(); cout << ans << "\n"; return 0; }

제출 #86879

제출 정보

제출 결과

제출 코드

찾은 반례

입력:

오답 출력:

정답 출력:

입력:

오답 출력:

정답 출력:

입력:

오답 출력:

정답 출력: