본문 바로가기

알고리즘 문제 풀이/Programmers

탑 - 2단계 (C++)

반응형
SMALL

주식가격과 비슷한 문제입니다. 대신 이 문제는 역으로 왼쪽으로 가군요 !

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> heights) {
	vector<int> answer;
    for (int i = 0; i < heights.size(); i++) {
    	if (i == 0){
        	answer.push_back(0);
        }
        else {
        	for (int j = i; j >= -1; j--) {
            	if (j == -1) {
                	answer.push_back(0);
                }
                if (heights[j] > heights[i]) {
                	answer.push_back(j + 1);
                    break;
            	}
            }
        }
    }
	return answer;
}    
반응형
LIST