🐜 Study: 탑 (#41)

Author: ing-eoking

ing-eoking/탑.cc

@@ -0,0 +1,16 @@
+#include <iostream>
+#include <vector>
+
+using namespace std;
+
+int main (int argc, char **argv) {
+    int N, tower;
+    vector<pair<int, int>> stack;
+    scanf("%d", &N);
+    for (int i = 1; i <= N; i++) {
+        scanf("%d", &tower);
+        while (!stack.empty() && stack.back().first < tower) stack.pop_back();
+        printf("%d ",stack.empty() ? 0 : stack.back().second);
+        stack.push_back({tower, i});  
+    }
+}