글 읽기 - 반례가 궁금합니다

lee2003121 2년 전 0

예제는 맞는데 틀릴 이유를 모르겠습니다

#include <bits/stdc++.h>

using namespace std;

int n,m,h, w[1002][1002][4],tank[1002][1002];
priority_queue <pair<int,pair<int,int> > > pq;


pair<int,int> pos(int x,int y,int dir)
{
    if(dir==0)
        return {x-1,y};
    else if(dir==1)
        return {x,y-1};
    else if(dir==2)
        return {x+1,y};
    else
        return {x,y+1};
}
int main()
{
    scanf("%d %d %d",&n,&m,&h);
    for(int i=1; i<=n+1; i++)
    {
        for(int j=1; j<=m; j++)
        {
            int a;
            scanf("%d",&a);
            w[i][j][0]=a;
            w[i-1][j][2]=a;
        }
    }
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=m+1; j++)
        {
            int a;
            scanf("%d",&a);
            w[i][j][1]=a;
            w[i][j-1][3]=a;
        }
    }
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=m; j++)
        {
            tank[i][j]=h;
        }
    }
    for(int i=0; i<=n+1; i++)
    {
        for(int j=0; j<=m+1; j++)
        {
            for(int k=0; k<4; k++)
            {
                if(w[i][j][k]>0)
                {
                    if(i==0||i==n+1||j==0||j==m+1)
                    pq.push({-w[i][j][k],pos(i,j,k)});
                }
            }
        }
    }


    while(!pq.empty())
    {
        int height=-pq.top().first;
        int x=pq.top().second.first;
        int y=pq.top().second.second;
        pq.pop();
        if(height<tank[x][y])
            tank[x][y]=height;
        else
            continue;
        for(int i=0; i<4; i++)
        {
            if(w[x][y][i]>0)
            {
                if(w[x][y][i]>tank[x][y])
                {
                    pq.push({-w[x][y][i],pos(x,y,i)});
                }
                else
                {
                    pq.push({-height,pos(x,y,i)});
                }
            }
        }

    }
    int total=0;

    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=m; j++)
        {
            total+=tank[i][j];
        }
    }
    printf("%d",total);
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

반례가 궁금합니다

lee2003121 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말