C++/코딩 기초 트레이닝
[C++/프로그래머스] 배열 만들기 3
서니션
2023. 10. 25. 13:56
배열 만들기 3
문제설명
정수 배열 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 합니다.
작성코드
#include <string>
#include <vector>
using namespace std;
vector<int> solution(vector<int> arr, vector<vector<int>> intervals) {
vector<int> answer;
int a1 = intervals[0][0];
int b1 = intervals[0][1];
int a2 = intervals[1][0];
int b2 = intervals[1][1];
for(int i=a1; i<=b1; i++)
{
answer.push_back(arr[i]);
}
for(int i=a2; i<=b2; i++)
{
answer.push_back(arr[i]);
}
return answer;
}
문제 URL
https://school.programmers.co.kr/learn/courses/30/lessons/181895
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr