BOJ 17143

문제

낚시왕 성공

 

1 초

512 MB

38078

11153

6336

25.940%

문제

낚시왕이 상어 낚시를 하는 곳은 크기가 R×C인 격자판으로 나타낼 수 있다. 격자판의 각 칸은 (r, c)로 나타낼 수 있다. r은 행, c는 열이고, (R, C)는 아래 그림에서 가장 오른쪽 아래에 있는 칸이다. 칸에는 상어가 최대 한 마리 들어있을 수 있다. 상어는 크기와 속도를 가지고 있다.

낚시왕은 처음에 1번 열의 한 칸 왼쪽에 있다. 다음은 1초 동안 일어나는 일이며, 아래 적힌 순서대로 일어난다. 낚시왕은 가장 오른쪽 열의 오른쪽 칸에 이동하면 이동을 멈춘다.

1. 낚시왕이 오른쪽으로 한 칸 이동한다.
2. 낚시왕이 있는 열에 있는 상어 중에서 땅과 제일 가까운 상어를 잡는다. 상어를 잡으면 격자판에서 잡은 상어가 사라진다.
3. 상어가 이동한다.

상어는 입력으로 주어진 속도로 이동하고, 속도의 단위는 칸/초이다. 상어가 이동하려고 하는 칸이 격자판의 경계를 넘는 경우에는 방향을 반대로 바꿔서 속력을 유지한채로 이동한다.

왼쪽 그림의 상태에서 1초가 지나면 오른쪽 상태가 된다. 상어가 보고 있는 방향이 속도의 방향, 왼쪽 아래에 적힌 정수는 속력이다. 왼쪽 위에 상어를 구분하기 위해 문자를 적었다.

상어가 이동을 마친 후에 한 칸에 상어가 두 마리 이상 있을 수 있다. 이때는 크기가 가장 큰 상어가 나머지 상어를 모두 잡아먹는다.

낚시왕이 상어 낚시를 하는 격자판의 상태가 주어졌을 때, 낚시왕이 잡은 상어 크기의 합을 구해보자.

입력

첫째 줄에 격자판의 크기 R, C와 상어의 수 M이 주어진다. (2 ≤ R, C ≤ 100, 0 ≤ M ≤ R×C)

둘째 줄부터 M개의 줄에 상어의 정보가 주어진다. 상어의 정보는 다섯 정수 r, c, s, d, z (1 ≤ r ≤ R, 1 ≤ c ≤ C, 0 ≤ s ≤ 1000, 1 ≤ d ≤ 4, 1 ≤ z ≤ 10000) 로 이루어져 있다. (r, c)는 상어의 위치, s는 속력, d는 이동 방향, z는 크기이다. d가 1인 경우는 위, 2인 경우는 아래, 3인 경우는 오른쪽, 4인 경우는 왼쪽을 의미한다.

두 상어가 같은 크기를 갖는 경우는 없고, 하나의 칸에 둘 이상의 상어가 있는 경우는 없다.

출력

낚시왕이 잡은 상어 크기의 합을 출력한다.

예제 입력 1 복사

4 6 8
4 1 3 3 8
1 3 5 2 9
2 4 8 4 1
4 5 0 1 4
3 3 1 2 7
1 5 8 4 3
3 6 2 1 2
2 2 2 3 5

예제 출력 1 복사

22

각 칸의 왼쪽 아래에 적힌 수는 속력, 오른쪽 아래는 크기, 왼쪽 위는 상어를 구분하기 위한 문자이다. 오른쪽 위에 ❤️는 낚시왕이 잡은 물고기 표시이다.

초기 상태

1초

2초 (E번 상어는 B번에게 먹혔다)

3초

4초

5초

6초

 

풀이

 

빡구현시키면서 시간초과 유도하는 문제이다.

 

풀이자체는 간단하다. 그냥 문제에서 설명하는 대로 쭉 짜주면 된다.

다만 입력 배열의 사이즈가 100 * 100이고, 상어의 속력이 1000이라 상어를 이동시킬때 상어가 모든칸에 있다는 가정 하에

상어의 개수 * 열 * 상어의 속력으로 N*M^2*S라는 괴랄한 시간이 나오게 된다.

그러므로 상어를 이동하는 동작에서 최대한 시간을 빼주어야 한다.

 

void move_shark(){
    vector<shark>move[102][102];
    vector<pair<int,int>>si;
    int vis[102][102] = {0};
        for(auto point : sh_point){
            if(board[point.first][point.second].empty()) continue;
            shark cur = board[point.first][point.second][0];
            int curx = point.first, cury = point.second;
            board[point.first][point.second].pop_back();
            int ns = cur.s;
            //for(int k=0; k<board[point.first][point.second][0].s; k++){ 
            while(ns){
                int s;
                if(cur.d == 1)
                    s = curx-1;
                else if(cur.d == 2)
                    s = R-curx;
                else if(cur.d == 3)
                    s = C-cury;
                else if(cur.d == 4)
                    s = cury-1;
                if(ns < s)
                    s = ns;
                ns -= s;
                int nx = curx + dx[cur.d-1]*s;
                int ny = cury + dy[cur.d-1]*s;
                // if((nx == 0 || nx == R || ny == 0 || ny == C)){
                //     if(cur.d == 1)
                //         cur.d = 2;
                //     else if(cur.d == 2)
                //         cur.d = 1;
                //     if(cur.d == 3)
                //         cur.d = 4;
                //     else if(cur.d == 4)
                //         cur.d = 3;
                // }
                if((nx == 1 || nx == R) &&(cur.d == 1 || cur.d ==2)){
                    if(cur.d == 1)
                        cur.d = 2;
                    else if(cur.d == 2)
                        cur.d = 1;
                }else if((ny == 1 || ny == C) && (cur.d == 3 || cur.d == 4) ){
                    if(cur.d == 3)
                        cur.d = 4;
                    else if(cur.d == 4)
                        cur.d = 3;
                }
                curx = nx;
                cury = ny;
            }
            
             
            if(move[curx][cury].empty())
                move[curx][cury].push_back(cur);
            else{
                shark com = move[curx][cury][0];
                move[curx][cury].pop_back();
                if(com.z > cur.z)
                    move[curx][cury].push_back(com);
                else
                    move[curx][cury].push_back(cur);
            }
            if(vis[curx][cury] == 0){
                vis[curx][cury] = 1;
                si.push_back({curx,cury});
            }
        }
        for(auto e : si){
            if(move[e.first][e.second].empty()) continue;
            board[e.first][e.second].push_back(move[e.first][e.second][0]);
        }
    sh_point = si;
    return;
}

 

이 부분이 상어를 움직이고, 먹는 과정까지 한 코드이다.

여기서 중요한 부분은

 

while(ns){
                int s;
                if(cur.d == 1)
                    s = curx-1;
                else if(cur.d == 2)
                    s = R-curx;
                else if(cur.d == 3)
                    s = C-cury;
                else if(cur.d == 4)
                    s = cury-1;
                if(ns < s)
                    s = ns;
                ns -= s;
                int nx = curx + dx[cur.d-1]*s;
                int ny = cury + dy[cur.d-1]*s;
                // if((nx == 0 || nx == R || ny == 0 || ny == C)){
                //     if(cur.d == 1)
                //         cur.d = 2;
                //     else if(cur.d == 2)
                //         cur.d = 1;
                //     if(cur.d == 3)
                //         cur.d = 4;
                //     else if(cur.d == 4)
                //         cur.d = 3;
                // }
                if((nx == 1 || nx == R) &&(cur.d == 1 || cur.d ==2)){
                    if(cur.d == 1)
                        cur.d = 2;
                    else if(cur.d == 2)
                        cur.d = 1;
                }else if((ny == 1 || ny == C) && (cur.d == 3 || cur.d == 4) ){
                    if(cur.d == 3)
                        cur.d = 4;
                    else if(cur.d == 4)
                        cur.d = 3;
                }
                curx = nx;
                cury = ny;
            }

이 상어를 움직이는 부분이다.

상어가 방향을 바꾸는 구간은 0번째, R번째 행에 있을때. 0번째, C번째 열에 있는 경우이다.

그러므로 상어가 3번행에 있다 가정하고 R == 7이라 가정하자.

상어가 방향이 바뀌지 않고 위로 움직이는 경우는 최대 3(상어의 행의 위치값) 만큼 움직일 수 있다.

상어가 방향이 바뀌지 않고 아래로 움직이는 경우는 최대 4(R - 상어의 행의 위치값) 만큼 움직일 수 있다.

이를 이용해서 상어를 한번 움직일때 최대한 많이 움직여주면 된다.

 

포인트

상어가 0번이나 가장자리에 있는 경우에는, 방향이 계속 바뀌어도 된다.

물론 약간의 시간이 더 들 수는 있지만, 방향이 반대여도 어처피 1초 후의 방향과 위치는 같기 때문이다.

 

전체 코드

#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
int R,C,M,dx[4]={-1,1,0,0},dy[4]={0,0,1,-1}, ans;
struct shark{
    int s,d,z=0;
};
vector<pair<int,int>>sh_point;
vector<shark>board[102][102];
void catch_shark(int idx){
    for(int i=1; i<=R; i++){
        if(board[i][idx].empty()) continue;
        ans += board[i][idx][0].z;
        board[i][idx].pop_back();
        return;
    }
}
void move_shark(){
    vector<shark>move[102][102];
    vector<pair<int,int>>si;
    int vis[102][102] = {0};
        for(auto point : sh_point){
            if(board[point.first][point.second].empty()) continue;
            shark cur = board[point.first][point.second][0];
            int curx = point.first, cury = point.second;
            board[point.first][point.second].pop_back();
            int ns = cur.s;
            //for(int k=0; k<board[point.first][point.second][0].s; k++){ 
            while(ns){
                int s;
                if(cur.d == 1)
                    s = curx-1;
                else if(cur.d == 2)
                    s = R-curx;
                else if(cur.d == 3)
                    s = C-cury;
                else if(cur.d == 4)
                    s = cury-1;
                if(ns < s)
                    s = ns;
                ns -= s;
                int nx = curx + dx[cur.d-1]*s;
                int ny = cury + dy[cur.d-1]*s;
                // if((nx == 0 || nx == R || ny == 0 || ny == C)){
                //     if(cur.d == 1)
                //         cur.d = 2;
                //     else if(cur.d == 2)
                //         cur.d = 1;
                //     if(cur.d == 3)
                //         cur.d = 4;
                //     else if(cur.d == 4)
                //         cur.d = 3;
                // }
                if((nx == 1 || nx == R) &&(cur.d == 1 || cur.d ==2)){
                    if(cur.d == 1)
                        cur.d = 2;
                    else if(cur.d == 2)
                        cur.d = 1;
                }else if((ny == 1 || ny == C) && (cur.d == 3 || cur.d == 4) ){
                    if(cur.d == 3)
                        cur.d = 4;
                    else if(cur.d == 4)
                        cur.d = 3;
                }
                curx = nx;
                cury = ny;
            }
            
             
            if(move[curx][cury].empty())
                move[curx][cury].push_back(cur);
            else{
                shark com = move[curx][cury][0];
                move[curx][cury].pop_back();
                if(com.z > cur.z)
                    move[curx][cury].push_back(com);
                else
                    move[curx][cury].push_back(cur);
            }
            if(vis[curx][cury] == 0){
                vis[curx][cury] = 1;
                si.push_back({curx,cury});
            }
        }
        for(auto e : si){
            if(move[e.first][e.second].empty()) continue;
            board[e.first][e.second].push_back(move[e.first][e.second][0]);
        }
    sh_point = si;
    return;
}
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    cin>>R>>C>>M;
    //input
    for(int i=0; i<M; i++){
        shark ns;
        int x, y;
        cin>>x>>y>>ns.s>>ns.d>>ns.z;
        board[x][y].push_back(ns);
        sh_point.push_back({x,y});
    }
    for(int st=1; st<=C; st++){
        catch_shark(st);
        move_shark();
    }
    cout<<ans;
}