BOJ - 4485 녹색 옷 입은 애가 젤다지?

4485번 녹색 옷 입은 애가 젤다지?

동굴이 주어지고 각 칸마다 도둑맞는 루피가 정해져 있다.

도둑맞는 루피를 최소화 하여 목적지에 도달하게 하는 것이 문제이다.

다익스트라를 활용하여 문제를 해결할 수 있다.

시작 점은 0,0이고 끝점은 N-1,N-1이다.

우선 특정 점 까지 도둑맞은 최소 루피를 저장하는 배열을 굉장히 큰 수로 초기화 해 준다.

그리고 나서 우선순위 큐에 해당 점의 상하좌우에 대해 도둑맞은 루피를 갱신하며 저장해 주면 된다.

코드

#include <iostream>
#include <queue>

using namespace std;

int xpos[4] = {-1,1,0,0};
int ypos[4] = {0,0,-1,1};
int sol(int n)
{
    int arr[126][126];
    int rupee[126][126];
    priority_queue<pair<int,pair<int,int>>> q;

    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            cin >> arr[i][j];
            rupee[i][j] = 125*125*10;
        }
    }

    q.push(make_pair(-arr[0][0],make_pair(0,0)));
    int sx,sy;
    int r;
    while(!q.empty())
    {
        sx = q.top().second.first;
        sy = q.top().second.second;
        r = -q.top().first;
        q.pop();

        if(sx == n-1 && sy == n-1)
            break;

        if(rupee[sx][sy] > r)
            rupee[sx][sy] = r;

        for(int k=0;k<4;k++)
        {
            int nx = sx + xpos[k];
            int ny = sy + ypos[k];
            if(nx<0 || nx >=n || ny<0 || ny >=n)
                continue;
            if(rupee[sx][sy] + arr[nx][ny] < rupee[nx][ny])
            {
                rupee[nx][ny] = rupee[sx][sy] + arr[nx][ny];
                q.push(make_pair(-rupee[nx][ny],make_pair(nx,ny)));
            }
        }
    }

    return rupee[n-1][n-1];
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int n;
    for(int i=1;;i++)
    {
        cin >> n;
        if(n == 0)
            break;

        int kkk = sol(n);
        cout << "Problem " << i << ": " << kkk << "\n";
    }
    return 0;
}

출처 : https://www.acmicpc.net/problem/4485

좋은 웹페이지 즐겨찾기