baekjoon. 14890 경사로 [Gold III][python]

https://www.acmicpc.net/problem/14890

14890번: 경사로

 

여기서, 알고리즘 구현할때 참고하면 좋은 부분은 다음과 같다. 

1. 높은곳 -> 낮은곳으로 경사로를 세울때, 

- 경사로의 높이는 낮은곳으로 동일해야 하며 

- 낮은곳 바로 다음 블럭의 높이는 낮은곳과 동일해야 한다

2. 낮은곳 -> 높은곳으로 경사로를 세울때, 

- 경사로의 높이는 낮은곳으로 동일하야 하며,

- 낮은곳 바로 전의 블럭은 낮은곳과 동일해야 한다. 

 

baekjoon. 14501

 

# 첫째 줄에 N (2 ≤ N ≤ 100)과 L (1 ≤ L ≤ N)이 주어진다.
# L : 경사로 길이 
N, L = map(int, input().split())
arr = [[None for _ in range(N)] for _ in range(N)]
for i in range(N):
    for j,v in enumerate(input().split()):
        arr[i][j] = int(v)


# 부분배열로 받아서 확인하는 부분 
def check(subArr):
    bridge = [False for _ in range(len(subArr))] # 경사로 

    cur = 1
    result = True
    tempArr = subArr.copy()

    cnt = 0 
    while (cur < N):
        if result == False: 
            break

        prev = tempArr[cur - 1]
        # 차이가 1 나면 경사로 추가함 
        if tempArr[cur] == prev:
            cur = cur + 1
            
        # 높은곳 -> 낮은곳 
        elif tempArr[cur] == (prev - 1):
            for i in range(L):
                if (cur + i) >= N or bridge[cur + i] == True or (cur + L -1)>=N:
                    result = False
                    break
                if  tempArr[cur + i] != tempArr[cur]:
                    result = False
                    break

                bridge[cur + i] = True 

            cur = cur + L
        # 낮은곳 -> 높은곳 
        elif tempArr[cur] == (prev + 1):
            for i in range(L):
                if (cur - 1 - i) < 0 or bridge[cur - 1 - i] == True:
                    result = False
                    break
                if tempArr[cur - 1 - i] != tempArr[cur - 1]:
                    result = False
                    break

            cur = cur + 1
        else:
            result = False
            break


    return result




# 시작부터, 아님 뒤에서부터 
debug = 0 
cnt = 0 
for i in range(N):
    tempRow = []
    tempCol = []

    for j in range(N):
        tempRow.append(arr[i][j])
        tempCol.append(arr[j][i])

    # 경사로 확인
    if (check(tempRow) or check(tempRow[::-1])):
        cnt = cnt + 1
    if (check(tempCol) or check(tempCol[::-1])):
        cnt = cnt + 1


print(cnt)

 

퇴사 [Silver III][python]