문제 링크: https://www.acmicpc.net/problem/17298
문제는 다음과 같습니다.
올해 첫 백준입니다.. 그리고 이 문제는 사실 12월에 풀다가 계속 틀렸습니다가 떠서 두고두고 남겨뒀었던 문제입니다. 며칠만에 다시 푸네요
백준 .. 진짜 오늘부터 무조건 하루에 한문제 이상씩 열심히 풀어야겠습니다! 😊
보이시나요.. 오기가 여기까지 느껴지네용 ㅋㅋ
저는 스택을 이용해서 풀었구요! 제 풀이는 다음과 같습니다.
#include <iostream>
#include <stack>
#include <vector>
#include <algorithm>
using namespace std;
int main(){
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
vector<int> v, res; // res: 결과벡터
stack<int> s;
int n;
cin>>n;
for(int i=0; i<n; i++){
int tmp;
cin>>tmp;
v.push_back(tmp);
}
reverse(v.begin(), v.end());
res.push_back(-1);
s.push(v[0]);
for(int i=1; i<v.size(); i++){
if(v[i] < s.top()){
res.push_back(s.top());
s.push(v[i]);
}
else{ // v[i] >= s.top()
while(!s.empty() && v[i]>=s.top()){
s.pop();
}
if(s.empty()){
res.push_back(-1);
s.push(v[i]);
}
else{ // v[i] < s.top()
res.push_back(s.top());
s.push(v[i]);
}
}
}
reverse(res.begin(), res.end());
for(int i=0; i<res.size(); i++) cout<<res[i]<<" ";
return 0;
}
for문이 가장 중요한 부분이구요
그동안 왜 계속 틀렸었냐면 바로 같은 수들에 대한 예외처리가 제대로 안되어 있었더라구요
그동안 알고리즘 문제들을 풀며 고민이 많았습니다.
많은 문제들을 풀고 맞는게 중요한가.. 한 문제를 풀고 이렇게 푼 과정이나 로직을 정리하는게 맞는가..
아무래도 저는 후자가 더 맞는 것 같습니다 ㅎㅎ
다른풀이 하나도 가져왔습니다!
참고하세용
로직은 같고, 반복되는 부분이나 겹치는 부분을 다 제거한 풀이입니다
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <math.h>
#include <stack>
using namespace std;
stack<int> st;
int arr[1000001];
vector<int> ans;
int main() {
ios_base::sync_with_stdio(false), cin.tie(NULL);
int n;
cin >> n;
for (int i = 1; i <= n; i++) {
cin >> arr[i];
}
for (int i = n; i > 0; i--) {
while (!st.empty() && st.top() <= arr[i]) {
st.pop();
}
if (st.empty()) ans.push_back(-1);
else ans.push_back(st.top());
st.push(arr[i]);
}
for (int i = ans.size() - 1; i >= 0; i--) {
cout << ans[i] << " ";
}
}