프로그래머스 - 수열과 구간 쿼리 2 - C++

https://school.programmers.co.kr/learn/courses/30/lessons/181923

프로그래머스

 

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr, vector<vector<int>> queries) {
    vector<int> answer;
    
    for(int i=0;i<queries.size();i++)
    {
        int min = 1000001;
        for(int j=queries[i][0];j<=queries[i][1];j++)
        {
            if(arr[j]>queries[i][2]&&min>arr[j])min = arr[j];
        }
        if(min==1000001)answer.push_back(-1);
        else answer.push_back(min);
    }
    
    return answer;
}

 

설명

arr 배열의 s 부터 e 까지의 인덱스를 보며 k보다 큰 arr[i] 중 가장 작은 수를 찾아 answer에 삽입하면 되는 문제이다. 

 

따라서 queries 만큼 반복문을 돌며 각 [s,e, k] 마다 j 포문을 돌며 s 부터 e 인덱스 까지의 arr[j] 중 k 보다 크고 min 보다 작은 수를 찾으면  된다. 

 

그리고 min 이 초기값과 동일하다면 -1 을 삽입한다.