2차원으로 만들기
2024. 8. 2. 18:00ㆍ코딩테스트 입문
https://school.programmers.co.kr/learn/courses/30/lessons/120842
문제 설명
정수 배열 num_list와 정수 n이 매개변수로 주어집니다. num_list를 다음 설명과 같이 2차원 배열로 바꿔 return하도록 solution 함수를 완성해주세요.
num_list가 [1, 2, 3, 4, 5, 6, 7, 8] 로 길이가 8이고 n이 2이므로 num_list를 2 * 4 배열로 다음과 같이 변경합니다. 2차원으로 바꿀 때에는 num_list의 원소들을 앞에서부터 n개씩 나눠 2차원 배열로 변경합니다.
num_list | n | result |
[1, 2, 3, 4, 5, 6, 7, 8] | 2 | [[1, 2], [3, 4], [5, 6], [7, 8]] |
제한사항
- num_list의 길이는 n의 배수 개입니다.
- 0 ≤ num_list의 길이 ≤ 150
- 2 ≤ n < num_list의 길이
입출력 예
num_list | n | result |
[1, 2, 3, 4, 5, 6, 7, 8] | 2 | [[1, 2], [3, 4], [5, 6], [7, 8]] |
[100, 95, 2, 4, 5, 6, 18, 33, 948] | 3 | [[100, 95, 2], [4, 5, 6], [18, 33, 948]] |
입출력 예 설명
입출력 예 #1
- num_list가 [1, 2, 3, 4, 5, 6, 7, 8] 로 길이가 8이고 n이 2이므로 2 * 4 배열로 변경한 [[1, 2], [3, 4], [5, 6], [7, 8]] 을 return합니다.
입출력 예 #2
- num_list가 [100, 95, 2, 4, 5, 6, 18, 33, 948] 로 길이가 9이고 n이 3이므로 3 * 3 배열로 변경한 [[100, 95, 2], [4, 5, 6], [18, 33, 948]] 을 return합니다.
k * 1 배열을 m * n 형태의 배열로 바꾸는 문제입니다. k는 m * n이고요.
이 문제의 처음에 answer = [[]]으로 되있다고 해서 굳이 중첩 리스트 방식으로 풀 필요는 없습니다.
그래도 굳이 푼다면 이렇게 풀 수 있기는 합니다.
def solution(num_list, n):
answer = [[]]
now = []
for num in num_list:
now.append(num)
if len(now) % n == 0:
answer.append(now)
now = []
return answer[1:]
다만 이 방법은 불필요한 리스트 사용과 추가 작업들이 들어가기 때문에, 간단히 풀기 위해서는 슬라이싱을 사용하면 됩니다.
슬라이싱을 사용해 문제를 풀어보겠습니다.
def solution(num_list, n):
answer = []
# 0부터 num_list의 길이까지 n 단위로 증가
for i in range(0, len(num_list), n):
# num_list를 슬라이싱
answer.append(num_list[i:i+n])
return answer
반복문의 step과 슬라이싱을 같이 사용하여 문제를 풀 수 있습니다.