[BaekJoon] 삼성 sw 역량 테스트(테트로미노)

• 이 문제는 c++로 풀이를 작성하였습니다.

문제

 

예시

 

풀이

#include<iostream>
#include<algorithm>
using namespace std;


int n,m;
int map[500][500];

// 각 테트로미노가 나올수 있는 경우의 수
int cnt[5]={2,1,8,4,4};
// 방향 벡터  dir[5][2] 는 대각선을 위해 표시
int dir[6][2]={{0,0},{0,1},{1,0},{0,-1},{-1,0},{1,-1}};

// 테트로미노가 나올수 있는 경우의 수를 방향으로 표시
int vec[5][8][4]={
    {{0,1,1,1},{0,2,2,2}},
    {{0,1,2,3}},
    {{0,2,2,1},{0,3,3,2},{0,1,2,2},{0,2,3,3},{0,2,2,3},{0,2,1,1},{0,3,2,2},{0,1,1,2}},
    {{0,2,1,2},{0,1,4,1},{0,2,3,2},{0,1,2,1}},
    {{0,1,1,5},{0,5,1,2},{0,5,1,1},{0,2,1,5}}
};

bool inside(int y,int x){
    return 0<=y && y<n && 0<=x && x<m;
}


int main(){
    int answer=0;
    
    cin>>n>>m;
    
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            cin>>map[i][j];
        }
    }
    
    // 모든 테트로미노에서 검사
    for(int a=0;a<5;a++){
        
        // 모든 좌표검사
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                
                // i,j 좌표에서부터 시작, 해당하는 테트로미노의 경우의 수만큼 반복
                for(int b=0;b<cnt[a];b++){
                    
                    // 처음 좌표 표시
                    int y=i,x=j;
                    int sum=0;
                    bool flag=true;
                    
                    // 테트로미노는 4개의 정사각형
                    for(int k=0;k<4;k++){
                        int d=vec[a][b][k];
                        int dy=y+dir[d][0];
                        int dx=x+dir[d][1];
                        
                        // 사각형 안에 있을경우
                        if(inside(dy,dx)){
                            y=dy;
                            x=dx;
                            sum+=map[dy][dx];
                        }
                        // 없으면 종료
                        else{
                            flag=false;
                            break;
                        }
                    }
                    
                    if(flag){
                        answer=max(answer,sum);
                    }
                }
            }
        }
    }
    
    cout<<answer;
}

• 추가로 궁금한 점이나 수정할 부분 있으면 댓글로 남겨주세요.