17143번 낚시왕

https://www.acmicpc.net/problem/17143

핵심은 시작부터

물고기가 끝 점(R or C) 또는 시작점(1)에 있을 수 있다.

예를 들어, 1에 있고 방향이 위로 향해있으면 아래로 고쳐줘야한다.

물론 1에 있지만 방향이 아래면 굳이 고칠 필요가 없다.

또한, 굳이 정렬을 사용할 필요가 없었다.

#include <bits/stdc++.h>
 
using namespace std;
 
typedef struct info
{
    int s = 0, d = 0, z = 0; // 미리 0으로 초기화
}info;
 
int R, C, M; // 행, 열
 
int dy[5] = { 0, -1, 1, 0, 0 }; // 1상, 2하, 3우, 4좌
int dx[5] = { 0, 0, 0, 1, -1 };
 
info fish[101][101];
 
int simulation()
{
    int ret = 0;
 
    for (int t = 1; t <= C; t++) 
    {
        info temp[101][101];
 
        // 낚시
        for (int r = 1; r <= R; r++)
            if (fish[r][t].z)
            {
                ret += fish[r][t].z;
                fish[r][t].z = 0;
                break;
            }    
 
        // 물고기 이동
        for (int i = 1; i <= R; i++)
            for (int j = 1; j <= C; j++)            
                if (fish[i][j].z)
                {
                    int moving, y = i, x = j;
 
                    if (fish[i][j].d <= 2) // 상, 하
                    {
                        moving = fish[i][j].s % (2 * (R - 1));
 
                        while (moving--)
                        {        
                            if (y == 1 && fish[i][j].d == 1) fish[i][j].d = 2;
                            if (y == R && fish[i][j].d == 2) fish[i][j].d = 1;                    
 
                            y += dy[fish[i][j].d];                    
                        }                    
                    }
                    else // 좌, 우
                    {
                        moving = fish[i][j].s % (2 * (C - 1));
 
                        while (moving--)
                        {
                            if (x == C && fish[i][j].d == 3) fish[i][j].d = 4;
                            if (x == 1 && fish[i][j].d == 4) fish[i][j].d = 3;                    
 
                            x += dx[fish[i][j].d];                
                        }
                    }
 
                    if (temp[y][x].z < fish[i][j].z) // 만약 더 큰 물고기가 있었다면
                        temp[y][x] = fish[i][j];
 
                    fish[i][j] = { 0, 0, 0 };
                }
        
        for (int i = 1; i <= R; i++)
            for (int j = 1; j <= C; j++)
                fish[i][j] = temp[i][j];
    }
 
    return ret;
}
 
int main(void)
{
    int r, c, s, d, z;
 
    scanf("%d %d %d", &R, &C, &M);
 
    for (int i = 0; i < M; i++) 
    {
        scanf("%d %d %d %d %d", &r, &c, &s, &d, &z);
        fish[r][c] = { s, d, z };
    }
 
    printf("%d\n", simulation());
 
    return 0;
}