[BaekJoon] 삼성 sw 역량 테스트(경사로)

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

문제

 

예시

 

풀이

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

// 경사로 되는지 확인하는 함수
bool find_load(int data[],int n,int l){
    int s=data[0];      // 가장 앞 원소
    int cnt=1;      // 가장 앞 원소랑 같은 수 체크하기 위한 count, 처음 시작은 1
    for(int i=1;i<n;i++){
        if(s==data[i]){     // 같으면  +1
            cnt++;
        }
        else if(abs(s-data[i])>=2){     // 경사 차이가 +2, -2이상이면 종료
            return false;
        }
        else if(s-data[i]==1){      // 경사가 낮아질 경우 : l만큼의 경사길이가 존재해야 한다
            bool find=true;
            for(int j=i;j<i+l;j++){
                if(data[i]!=data[j]){       // 경사가 l만큼 확인하는 동안 달라질 경우 종료
                    return false;
                }
            }
            if(find){       // l만큼 길이 모두 같은 경우
                i+=l-1;     // 인덱스는 마지막에 해당하는 숫자.  => ex)  l=2일경우    3 2 1 이면 1에 해당하는 인덱스
                if(i<n){            // 마지막 인덱스에 해당하는 원소
                    s=data[i];
                    cnt=0;      // cnt=0이여야 하는데 이유는 3 3 2 2 3 3 이럴경우 2 2를 모두 한 경사로가 들어갔기 때문에 다시는 못들어감
                }
            }
        }
        else{       // 경사가 높을 경우 : cnt로 그 전 경사의 길이 체크하고 있었음
            if(cnt>=l){     // l보다 짧은 경우라면 종료, 그렇다면 현재 높아진 경사 데이터 획득
                s=data[i];
                cnt=1;
            }
            else{
                return false;
            }
        }
    }
    return true;
}

int main(){
    int answer=0;
    int n,l;
    cin>>n>>l;

    int arr[100][100];
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            cin>>arr[i][j];
        }
    }
    
    // 가로 경사로 확인
    for(int i=0;i<n;i++){
        int data[100];
        for(int j=0;j<n;j++){
            data[j]=arr[i][j];
        }
        bool check=find_load(data,n,l);
        if(check){
            answer++;
        }
    }
    
    // 세로 경사로 확인
    for(int j=0;j<n;j++){
        int data[100];
        for(int i=0;i<n;i++){
            data[i]=arr[i][j];
        }
        bool check=find_load(data,n,l);
        if(check){
            answer++;
        }
    }
    cout<<answer;
}

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