#include #include using namespace std; struct Shark { int y; int x; int speed; int dir; int size; Shark(int y, int x, int speed, int dir , int size) : y(y), x(x), speed(speed), dir(dir), size(size) {} bool operator < (const Shark& s) const{ if (this->size < s.size) return true; else return false; } }; int r, c, m; int board[101][101]; priority_queue sharks; int dy[4] = {-1, 1, 0, 0}; // 위 아래 오른쪽 왼쪽 int dx[4] = {0, 0, 1, -1}; int fishKing = 0; int shark_visited[101][101]; int ans = 0; void catch_shark(int turn) { int die_y = -1, die_x = -1; for (int i = 1; i <= r; i++) { if (shark_visited[i][fishKing] == turn) { die_y = i; die_x = fishKing; break; } } if (die_x == -1 && die_y == -1) return; // 잡을 게 없는 경우 priority_queue temp; while(!sharks.empty()) { if (sharks.top().y == die_y && sharks.top().x == die_x) { // 상어 제거 //cout << die_y << " " << die_x << endl; // shark_visited[die_y][die_x] = 0; ans += sharks.top().size; } else { temp.push(sharks.top()); } sharks.pop(); } sharks = temp; } pair, int > move_up(Shark s) { // 다음 위치, 방향 전환 여부 int ny; int flag = 1; if (s.dir == 1 || s.y == r) { // 위 쪽 if (s.y > s.speed) { ny = s.y - s.speed; } else { int dist = s.speed; dist = dist - s.y + 1; int turn_cnt = dist / (r-1); if (turn_cnt % 2 == 0) { ny = dist % (r - 1) + 1; flag = 2; } else { ny = r - (dist % (r - 1)); flag = 1; } } return make_pair(make_pair(ny, s.x), flag); } else if (s.dir == 2 || s.y == 1) { // 아래 쪽 if (s.y + s.speed < r) { ny = s.y + s.speed; } else { int dist = s.speed; dist = dist - (r - s.y); int turn_cnt = dist / (r-1); // if (turn_cnt % 2 == 0) { ny = r - (dist % (r-1)); flag = 1; } else { ny = 1 + (dist % (r-1)); flag = 2; } } return make_pair(make_pair(ny, s.x), flag); } } pair, int > move_right(Shark s) { // 다음 위치, 방향 전환 여부 int nx; int flag = 3; if (s.dir == 3 || s.x == 1) { // 오른 쪽 if (s.x + s.speed < c) { nx = s.x + s.speed; } else { int dist = s.speed; dist = dist - (c - s.x); int turn_cnt = dist / (c-1); if (turn_cnt % 2 == 0) { nx = c - dist % (c-1); flag = 4; } else { nx = (dist % (c-1)) + 1; // flag = 3; } } return make_pair(make_pair(s.y, nx), flag); } else if (s.dir == 4 || s.x == c) { // 왼쪽 if (s.x - s.speed > 1) { nx = s.x - s.speed; } else { int dist = s.speed; dist = dist - (s.x - 1); // int turn_cnt = dist / (c-1); // if (turn_cnt % 2 == 0) { nx = dist % (c - 1) + 1; // flag = 3; } else { // nx = c - (dist % (c - 1)); flag = 4; } } return make_pair(make_pair(s.y, nx), flag); } } void shark_move(int turn) { priority_queue temp; // 홀 수일때, while (!sharks.empty()) { // 방향이 1, 2인거와 3, 4인거를 구별해서 풀이해야 한다. Shark ns = sharks.top(); // shark_visited[ns.y][ns.x] = 0; sharks.pop(); if (ns.dir == 1 || ns.dir == 2) { pair ,int> nn = move_up(ns); if (shark_visited[nn.first.first][nn.first.second] != turn) { //cout << ns.size << " " << nn.first.first << " " << nn.first.second << endl; temp.push(Shark(nn.first.first, nn.first.second, ns.speed, nn.second, ns.size)); shark_visited[nn.first.first][nn.first.second] = turn; } } else if (ns.dir == 3 || ns.dir == 4) { pair, int> nn = move_right(ns); if (shark_visited[nn.first.first][nn.first.second] != turn) { //cout << ns.size << " " << nn.first.first << " " << nn.first.second << endl; temp.push(Shark(nn.first.first, nn.first.second, ns.speed, nn.second, ns.size)); shark_visited[nn.first.first][nn.first.second] = turn; } } } sharks = temp; } void simulation() { while (fishKing <= c) { fishKing++; catch_shark(fishKing); shark_move(fishKing+1); } } int main() { cin >> r >> c >> m; int a, b, c, d, e; for (int i = 0; i < m; i++) { cin >> a >> b >> c >> d >> e; shark_visited[a][b] = 1; sharks.push(Shark(a, b, c, d, e)); } simulation(); cout << ans << endl; return 0; }

제출 #13748

제출 정보

제출 결과

제출 코드

찾은 반례

입력:

오답 출력:

정답 출력:

입력:

오답 출력:

정답 출력:

입력:

오답 출력:

정답 출력: