https://school.programmers.co.kr/learn/courses/30/lessons/181895
문제 설명
정수 배열 arr와 2개의 구간이 담긴 배열 intervals가 주어집니다.
intervals는 항상 [[a1, b1], [a2, b2]]의 꼴로 주어지며 각 구간은 닫힌 구간입니다. 닫힌 구간은 양 끝값과 그 사이의 값을 모두 포함하는 구간을 의미합니다.
이때 배열 arr의 첫 번째 구간에 해당하는 배열과 두 번째 구간에 해당하는 배열을 앞뒤로 붙여 새로운 배열을 만들어 return 하는 solution 함수를 완성해 주세요.
제한사항
1 ≤ arr의 길이 ≤ 100,000
1 ≤ arr의 원소 < 100
1 ≤ a1 ≤ b1 < arr의 길이
1 ≤ a2 ≤ b2 < arr의 길이
입출력 예 설명
입출력 예 #1
첫 번째 구간에 해당하는 배열은 [2, 3, 4] 입니다.
두 번째 구간에 해당하는 배열은 [1, 2, 3, 4, 5] 입니다.
따라서 이 두 배열을 앞뒤로 붙인 배열인 [2, 3, 4, 1, 2, 3, 4, 5]를 return 합니다.
풀이
def solution(arr, intervals):
answer = []
for a, b in intervals:
answer.append(arr[a:b+1])
return sum(answer, [])여기서 2차원 배열을 1차원 배열로 출력하여 답이 나오게끔 만들었다.
2차원 배열을 1차원 배열로 바꾸는 방법은 여러가지가 있다.
① for 반복문 사용
my_list = [[1, 2], [3, 4], [5, 6]]
answer = []
for element in my_list:
answer += element②sum 함수 사용
answer = sum(my_list, [])③itertools와 unpacking
import itertools
list(itertools.chain.from_iterable(my_list))④list comprehension 이용
[element for array in my_list for element in array]