탑 문제풀이(C++, 스택/큐)[프로그래머스]

안녕하세요 멍청한 토끼입니다.

이번 문제는 프로그래머스의 스택/큐 Lv2에 해당하는 탑문제 입니다.

시작하겠습니다.

https://programmers.co.kr/learn/courses/30/lessons/42588

코딩테스트 연습 - 탑 | 프로그래머스

---

목차

1. 문제 설명

2. 문제 해석

3. 소스 코드

4. 결과

---

3.1 주석 없는 코드

3.2 주석 있는 코드

3.3 테스트 코드

---

1. 문제 설명

수평 직선에 탑 N대를 세웠습니다. 모든 탑의 꼭대기에는 신호를 송/수신하는 장치를 설치했습니다. 발사한 신호는 신호를 보낸 탑보다 높은 탑에서만 수신합니다. 또한, 한 번 수신된 신호는 다른 탑으로 송신되지 않습니다. 예를 들어 높이가 6, 9, 5, 7, 4인 다섯 탑이 왼쪽으로 동시에 레이저 신호를 발사합니다. 그러면, 탑은 다음과 같이 신호를 주고받습니다. 높이가 4인 다섯 번째 탑에서 발사한 신호는 높이가 7인 네 번째 탑이 수신하고, 높이가 7인 네 번째 탑의 신호는 높이가 9인 두 번째 탑이, 높이가 5인 세 번째 탑의 신호도 높이가 9인 두 번째 탑이 수신합니다. 높이가 9인 두 번째 탑과 높이가 6인 첫 번째 탑이 보낸 레이저 신호는 어떤 탑에서도 수신할 수 없습니다.

 

송신 탑(높이)	수신 탑(높이)
5(4)	4(7)
4(7)	2(9)
3(5)	2(9)
2(9)	-
1(6)	-

맨 왼쪽부터 순서대로 탑의 높이를 담은 배열 heights가 매개변수로 주어질 때 각 탑이 쏜 신호를 어느 탑에서 받았는지 기록한 배열을 return 하도록 solution 함수를 작성해주세요.

 

제한사항

heights는 길이 2 이상 100 이하인 정수 배열입니다. 모든 탑의 높이는 1 이상 100 이하입니다. 신호를 수신하는 탑이 없으면 0으로 표시합니다.

 

heights	return
[6,9,5,7,4]	[0,0,2,2,4]
[3,9,9,3,5,7,2]	[0,0,0,3,3,3,6]
[1,5,3,6,7,6,5]	[0,0,2,0,0,5,6]

 

입출력 예 설명

입출력 예 #1 앞서 설명한 예와 같습니다. 입출력 예 #2 [1,2,3] 번째 탑이 쏜 신호는 아무도 수신하지 않습니다. [4,5,6] 번째 탑이 쏜 신호는 3번째 탑이 수신합니다. [7] 번째 탑이 쏜 신호는 6번째 탑이 수신합니다. 입출력 예 #3 [1,2,4,5] 번째 탑이 쏜 신호는 아무도 수신하지 않습니다. [3] 번째 탑이 쏜 신호는 2번째 탑이 수신합니다. [6] 번째 탑이 쏜 신호는 5번째 탑이 수신합니다. [7] 번째 탑이 쏜 신호는 6번째 탑이 수신합니다.

 

---

2. 문제풀이

모든 신호는 왼쪽으로 갑니다. 따라서 오른쪽부터 검사를 해서 자신보다 큰 타워가 나온다면 타워의 위치를 정답리스트에 추가합니다. 자신보다 큰타워가 없다면 0을 추가합니다.(제약사항 확인) 제일 처음타워는 신호를 받을 타워가 없으므로 0을 추가합니다.(제약사항 확인) 주의사항은 첫 타워는 1부터 시작합니다. 뒤에서 부터 검사를 하기 때문에 정답을 스택에넣고 마지막에 리스트에 옮기는 작업으로 마무리 했습니다.

 

---

3. 소스코드

 

3.1 주석없는 코드

 

#include <string>
#include <vector>
#include <stack>
 
using namespace std;
 
vector<int> solution(vector<int> heights) {
    vector<int> answer;
    stack<int> st;
    bool check = true;
    for (int j = heights.size() - 1; j >= 0; j--) {
        for (int i = j - 1; i >= 0; i--) {
            if (heights.at(i) > heights.at(j)){
                st.push(i + 1);
                check = false;
                break;
            }
        }
        if (check)
            st.push(0);
        check = true;
    }
    while (!st.empty()) {
        answer.push_back(st.top());
        st.pop();
    }
    return answer;
}

3.2 주석있는 코드

#include <string>
#include <vector>
#include <stack>
 
using namespace std;
 
vector<int> solution(vector<int> heights) {
    vector<int> answer;
    stack<int> st;
    bool check = true;
    //리스트를 뒤에서부터 순회하기
    for (int j = heights.size() - 1; j >= 0; j--) {
        //자기자신과 비교할필요가 없기때문에 비교하는것의 앞부터 계산
        for (int i = j - 1; i >= 0; i--) {
            //제일뒤에꺼보다 앞에타워가 큰경우
            if (heights.at(i) > heights.at(j)){
                //스택에 몇번째 타워인지 넣기
                st.push(i + 1);
                //조건문 비활성화
                check = false;
                //반복문 나가기
                break;
            }
        }
        //만약 앞터워가 큰게없다면
        //스택에 없다고 0을 넣기
        if (check)
            st.push(0);
        check = true;
    }
    //스택이 빌동안 정답에 푸쉬
    while (!st.empty()) {
        answer.push_back(st.top());
        st.pop();
    }
    return answer;
}

3.3 테스트 코드 추가

#include <string>
#include <iostream>
#include <vector>
#include <stack>
 
using namespace std;
 
vector<int> solution(vector<int> heights) {
    vector<int> answer;
    stack<int> st;
    bool check = true;
    //리스트를 뒤에서부터 순회하기
    for (int j = heights.size() - 1; j >= 0; j--) {
        //자기자신과 비교할필요가 없기때문에 비교하는것의 앞부터 계산
        for (int i = j - 1; i >= 0; i--) {
            //제일뒤에꺼보다 앞에타워가 큰경우
            if (heights.at(i) > heights.at(j)){
                //스택에 몇번째 타워인지 넣기
                st.push(i + 1);
                //조건문 비활성화
                check = false;
                //반복문 나가기
                break;
            }
        }
        //만약 앞터워가 큰게없다면
        //스택에 없다고 0을 넣기
        if (check)
            st.push(0);
        check = true;
    }
    //스택이 빌동안 정답에 푸쉬
    while (!st.empty()) {
        answer.push_back(st.top());
        st.pop();
    }
    return answer;
}
 
void print(vector<int> heights, vector<int> answer)
{
    vector<int> t = solution(heights);
    for (int i = 0; i < answer.size(); i++) {
        cout << answer[i] << " ";
        if (answer[i] != t[i]){
            cout << "틀림" << endl;
            return;
        }
    }
    cout << "정답" << endl;
}
 
int main(){
    print({ 6, 9, 5, 7, 4 }, { 0,0,2,2,4 });
    print({ 3,9,9,3,5,7,2 }, { 0,0,0,3,3,3,6 });
    print({ 1,5,3,6,7,6,5 }, { 0,0,2,0,0,5,6 });
    return 0;
}

 

4. 결과