Merge pull request #728 from baekhangyeol/main

[백한결] 115차 라이브 코테 제출

Author: github-actions[bot]

live11/test115/문제1/백한결.py

@@ -0,0 +1,22 @@
+import sys
+
+def main():
+    input = sys.stdin.readline
+    n = int(input())
+    A = [int(input()) for _ in range(n)]
+
+    v = [A[0]]
+    for x in A[1:]:
+        if x != v[-1]:
+            v.append(x)
+
+    H = max(v)
+
+    ans = H - v[0]
+    for i in range(1, len(v)):
+        if v[i] < v[i-1]:
+            ans += (v[i-1] - v[i])
+    print(ans)
+
+if __name__ == "__main__":
+    main()

live11/test115/문제2/백한결.py

@@ -0,0 +1,19 @@
+import sys
+
+def main():
+    input = sys.stdin.readline
+    size = int(input())
+    A = list(map(int, input().split()))
+    result = [-1] * size
+    stack = []
+
+    for i in range(size):
+        while stack and A[stack[-1]] < A[i]:
+            index = stack.pop()
+            result[index] = A[i]
+        stack.append(i)
+
+    print(*result)
+
+if __name__ == '__main__':
+    main()