[BaekJoon] 삼성 sw 역량 테스트(이차원 배열과 연산)

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

문제

 

예시

 

풀이

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

int vec[101][101]={0};
int r,c,k;
int n=3;
int m=3;

struct node{
    int n;
    int cnt;
};

// node의 배열을 정렬하기 위한 비교함수
// 수가 많아지는 순으로, 같으면 수가 커지는 순으로
bool cmp(node a,node b){
    if(a.cnt==b.cnt){
        return a.n<b.n;
    }
    return a.cnt<b.cnt;
}

int main(){
    int answer=0;
    cin>>r>>c>>k;
    
    for(int i=1;i<=3;i++){
        for(int j=1;j<=3;j++){
            cin>>vec[i][j];
        }
    }
    
    // answer이 100이상이거나 vec[r][c]==k가 되면 종료
    while(vec[r][c]!=k && answer<=100){
        answer++;
        if(n>=m){       // 행 정렬
            for(int i=1;i<=n;i++){
                int copy[101]={0};
                // 해당 숫자의 갯수를 알기 위한 작업
                for(int j=1;j<=m;j++){
                    copy[vec[i][j]]++;
                }
                
                // node 배열에 할당
                node nd[101];
                int idx=1;
                for(int j=1;j<=100;j++){
                    if(copy[j]!=0){
                        node temp;
                        temp.n=j;
                        temp.cnt=copy[j];
                        nd[idx++]=temp;
                    }
                }
                sort(nd+1,nd+idx,cmp);  // 정렬
                
                // 추가적인 수정된 열
                int rm=0;
                for(int j=1;j<idx;j++){
                    vec[i][++rm]=nd[j].n;
                    vec[i][++rm]=nd[j].cnt;
                }
                for(int j=rm+1;j<=100;j++){
                    vec[i][j]=0;
                }
                if(rm>=m){m=rm;}
            }
        }
        else{       // 열 정렬 (똑같이 구현)
            for(int i=1;i<=m;i++){
                int copy[101]={0};
                for(int j=1;j<=n;j++){
                    copy[vec[j][i]]++;
                }
                
                node nd[101];
                int idx=1;
                for(int j=1;j<=100;j++){
                    if(copy[j]!=0){
                        node temp;
                        temp.n=j;
                        temp.cnt=copy[j];
                        nd[idx++]=temp;
                    }
                }
                sort(nd+1,nd+idx,cmp);
                
                int rn=0;
                for(int j=1;j<idx;j++){
                    vec[++rn][i]=nd[j].n;
                    vec[++rn][i]=nd[j].cnt;
                }
                for(int j=rn+1;j<=100;j++){
                    vec[j][i]=0;
                }
                
                if(rn>=n){n=rn;}
            }
        }
    }
    

    answer=answer>100?-1:answer;
    cout<<answer;
}

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