99클럽 코테 스터디 1일차 TIL + 배열

오늘의 문제 - 2차원 배열

문제 설명

정수 n, left, right가 주어집니다. 다음 과정을 거쳐서 1차원 배열을 만들고자 합니다.

n행 n열 크기의 비어있는 2차원 배열을 만듭니다.
i = 1, 2, 3, ..., n에 대해서, 다음 과정을 반복합니다.
1행 1열부터 i행 i열까지의 영역 내의 모든 빈 칸을 숫자 i로 채웁니다.
1행, 2행, ..., n행을 잘라내어 모두 이어붙인 새로운 1차원 배열을 만듭니다.
새로운 1차원 배열을 arr이라 할 때, arr[left], arr[left+1], ..., arr[right]만 남기고 나머지는 지웁니다.
정수 n, left, right가 매개변수로 주어집니다. 주어진 과정대로 만들어진 1차원 배열을 return 하도록 solution 함수를 완성해주세요.

내 풀이 및 접근 방식

def solution(n, left, right):
    answer = []

    arr = [[ i+1 if j <= i else j+1 for j in range(n)] for i in range(n)]   

    for i in range(int(left//n), int(right//n)+1):
        for j in range(n):
            if i == int(left//n) and j < left%n:
                continue
            elif i == int(right//n) and j > right%n:
                break
            answer.append(arr[i][j])

    return answer

1. arr 배열에 이중 for문을 이용해서 숫자를 채워 넣는다.
2. 시작과 끝의 행과 열을 구한다.
   ex) 시작행: left//n, 시작열: left%n
3. 이중 for문을 돌며 범위에 해당하는 행과 열만 answer에 추가한다.

left, right 가지고 행과 열을 구하는 것까지는 이해했는데, 이중 for문 2번 써서 시간 초과 문제 발생.. 총점 30점...

다른 사람 코드

1차원 배열과 max 이용하는 방식 발견! 천재만재!!

행과 열 인덱스 중 더 큰 값에 1을 더한 값이 배열에 저장되는 규칙 찾는 것이 중요했던 것 같다.
ex) 행 인덱스 >= 열 인덱스: 행 인덱스 + 1, 행 인덱스 < 열 인덱스: 열 인덱스 + 1

def solution(n, left, right):
    answer = [0] * (right - left + 1) # answer 초기화

    for i in range(left, right + 1):
        temp = max(i // n, i % n)    # i//n은 행, i%n은 열
        answer[i - left] = temp + 1  # answer 변경

    return answer

회고

gif 방식대로 이중 for문 생성하고 값 추가한 후, 범위에 맞게 자르려고 하니 시간 초과가 발생했다. 문제에 함정이 있을수도,, 배열에 숨겨진 규칙 찾아보고 효율적으로 풀자

 

비기너로 바꿔야할지..