#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; int ans; int cur_time; int dx[] = { 0,-1,1,0,0 }; int dy[] = { 0,0,0,1,-1 }; struct Shark { int x; int y; int speed; int dir; int size; int 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]; queue Q; bool inRange(int x, int y) { return 0 < x && 0 < y && x <= N && 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.x, s.y, s.speed, s.dir, s.size, 0}; Q.push(s); } } void printMap() { for (int i = 1; i <= N; i++) { for (int j = 1; j <= M; j++) { cout << map[cur_time][i][j].size << " "; } cout << " "; for (int j = 1; j <= M; j++) { cout << map[cur_time][i][j].speed << " "; } cout << "\n"; } } void catch_shark() { int Size; //상어잡기 for (int i = 1; i <= N; i++) { if (map[cur_time][i][FishKing_pos].size) { auto top = map[cur_time][i][FishKing_pos]; Size = Q.size(); for (int s = 0; s < Size; s++) { auto cur = Q.front(); Q.pop(); if (cur == map[cur.time][i][FishKing_pos]) { //상어 잡기 완료 ans += cur.size; return; } Q.push(cur); } } } } void dir_calib(int& x, int& y, int& d) { int over; while (!inRange(x,y)) { if (x < 1) { over = 2-x; x = over; if (d == U) d = D; else if (d == D) d = U; } if (x > N) { over = x - N; x = N - over; if (d == U) d = D; else if (d == D) d = U; } if (y < 1) { over = 2-y; y = over; if (d == L) d = R; else if (d == R) d = L; } if (y > M) { over = y - M; y = M - over; if (d == L) d = R; else if (d == R) d = L; } } } void move_shark() { int nx, ny, nt; int Size = Q.size(); for (int i = 0; i < Size; i++) { auto cur = Q.front(); Q.pop(); nt = cur.time + 1; nx = cur.x + dx[cur.dir] * cur.speed; ny = cur.y + dy[cur.dir] * cur.speed; dir_calib(nx, ny, cur.dir); if (map[nt][nx][ny].size > cur.size) continue; Q.push({ nx,ny,cur.speed,cur.dir, cur.size, nt }); map[nt][nx][ny] = { nx,ny,cur.speed,cur.dir, cur.size, nt }; } } void solution() { //낚시왕이 한 칸 이동한다 //낚시왕이 있는 열에 있는 상어중 땅과 가장 가까운 상어를 잡는다, 상어 사라짐 //상어가 이동한다. //낚시왕이 맨 오른쪽에 갈 때 종료 -> M번 while (FishKing_pos <= M) { FishKing_pos++; catch_shark(); move_shark(); cur_time++; } } int main() { input(); solution(); cout << ans; return 0; }

제출 #86904

제출 정보

제출 결과

제출 코드

찾은 반례

입력:

오답 출력:

정답 출력:

입력:

오답 출력:

정답 출력:

입력:

오답 출력:

정답 출력: