1번 문제(탑)
#include <vector>
#include <iostream>
using namespace std;
vector solution(vector<int> heights) {
vector<int> answer;
vector<int>::iterator it;
for(int i=heights.size() -1; i >= 0 ;i--) {
int ans = 0;
for (int j=i-1;j >= 0;j--) {
if (heights[i] < heights[j]) {
ans = j+1;
break;
}
}
it = answer.begin();
it = answer.insert(it, ans);
}
return answer;
}
int main() {
vector<int> heights = {6,9,5,7,4};
vector<int> answer = solution(heights);
for(int i:answer) {
cout << i << " ";
}
}
2번문제(다리를 지나는 트럭)
#include <iostream>
#include <string>
#include <vector>
using namespace std;
int calc_bridge_sum(vector<int> bridge) {
int sum = 0;
for(int i=0;i<bridge.size();i++) {
sum += bridge[i];
}
return sum;
}
void print_bridge(vector<int> bridge) {
int sum = 0;
for(int i=0;i<bridge.size();i++) {
cout << bridge[i] << " ";
sum += bridge[i];
}
cout << "[" << sum << "]" << endl;
}
int solution(int bridge_length, int weight, vector<int> truck_weights) {
int answer = 0;
vector<int> bridge;
vector<int>::iterator it;
int bridge_sum = 0;
int pos = 0;
while (true) {
print_bridge(bridge);
if (bridge.size() >= bridge_length) {
bridge.pop_back();
bridge_sum = calc_bridge_sum(bridge);
}
it = bridge.begin();
if (pos >= truck_weights.size()) {
it = bridge.insert(it, 0);
} else if ((bridge_sum + truck_weights[pos]) <= weight) {
it = bridge.insert(it, truck_weights[pos]);
pos ++;
bridge_sum = calc_bridge_sum(bridge);
} else {
it = bridge.insert(it, 0);
}
answer ++;
if (pos >= truck_weights.size() && bridge_sum == 0) break;
}
return answer;
}
int main() {
int bridge_length = 2;
int weight = 10;
vector<int> truck_weights = {7,4,5,6};
int answer = solution(bridge_length, weight, truck_weights);
cout << "answer : " << answer << endl;
}
3번째 문제(기능개발)
문제를 풀고나서 다른 사람의 답을 확인했는데, 큐를 이용해서 풀었다. 원래 이번 파트가 큐를 이용한 것인데, 난 아직도 벡터를 이용중이다. 다음 문제는 큐를 이용해서 풀어야 겠다.
#include <string>
#include <vector>
#include <iostream>
using namespace std;
vector solution(vector<int> progresses, vector<int> speeds) {
vector<int> answer;
int success = 0;
int pos = 0;
while (true) {
for (int i=0;i<progresses.size();i++) {
progresses[i] += speeds[i];
}
success = 0;
for(int j=pos;j<progresses.size();j++) {
if (progresses[j] >= 100) {
success ++;
pos ++;
} else break;
}
if (success > 0) answer.push_back(success);
if (pos >= progresses.size()) break;
}
return answer;
}
void print_vector(vector<int> v) {
for (int i=0;i<v.size();i++) {
cout << v[i] << " ";
}
}
int main() {
vector<int> progresses = {93,30,55};
vector<int> speeds = {1,30,5};
vector<int> answer;
answer = solution(progresses, speeds);
print_vector(answer);
return 0;
}
4번째(프린터)
#include <iostream>
#include <string>
#include <vector>
#include <map>
#include <queue>
using namespace std;
void print_map(map<int, int> m) {
map<int, int>::iterator i;
for(i = m.begin(); i != m.end(); i++){
cout << "[" << i->first << "]" << i->second << endl ;
}
}
int solution(vector<int> priorities, int location) {
int answer = 0;
queue<int> que1, que2, que3;
map<int, int> map;
int q1, q2, q3;
int seq = 1;
for (int i=0;i<priorities.size();i++) {
que1.push(i); // 순서
que2.push(priorities[i]); // 우선순위
}
while (!que2.empty()) {
q2 = que2.front();
que2.pop();
que3 = que2;
bool bRet = false;
while (!que3.empty()) { // 큰 값이 있으면 마지막 큐로 이동(순서 큐도 동일하게 이동)
q3 = que3.front();
que3.pop();
if (q2 < q3) {
bRet = true;
break;
}
}
if (bRet) {
que2.push(q2);
q1 = que1.front();
que1.pop();
que1.push(q1);
} else {
q1 = que1.front();
que1.pop();
map[q1] = seq++;
}
}
//print_map(map);
answer = map[location];
return answer;
}
int main(void) {
int answer = 0;
vector<int> priorities = {1, 1, 9, 1, 1, 1};
int location = 0;
answer = solution(priorities, location);
cout << "answer = " << answer << endl;
return 0;
}
이번의 경우에도 프린터 출력순서를 큐에 넣는 방법으로 간단하게 해결한 사람이 있다. 코딩하기 전에 알고리즘을 좀 더 생각해야 겠다.
int solution(vector<int> priorities, int location) {
queue<int> printer; //queue에 index 삽입.
vector<int> sorted; //정렬된 결과 저장용
for(int i=0; i<priorities.size(); i++) {
printer.push(i);
}
while(!printer.empty()) {
int now_index = printer.front();
printer.pop();
if(priorities[now_index] != *max_element(priorities.begin(),priorities.end())) {
//아닌경우 push
printer.push(now_index);
} else {
//맞는경우
sorted.push_back(now_index);
priorities[now_index] = 0;
}
}
for(int i=0; i<sorted.size(); i++) {
if(sorted[i] == location) return i+1;
}
return 0;
}
5번째 문제(쇠막대기)
#include <iostream> #include <string> #include <vector> #include <deque>
다른 사람의 풀이를 보면 스택을 사용했다. 스택은 큐의 반대 개념이다. 나중에 넣은 것을 먼저 꺼내는 방식이다. 내가 푼 방식에서 스택으로 고치지 않아도 굳이 벡터에 값을 넣지 않고 바로 결과값에 더하는 방식으로도 풀이가 가능하다.
int solution(string arrangement) {
int answer = 0;
deque deq;
int n = 0;
for(int i=1;i<arrangement.length();i++) {
if (arrangement[i-1] == '(' ) {
if (arrangement[i] == '(') {
deq.push_back(n++);
answer ++;
} else {
answer += deq.size();
}
} else if (arrangement[i-1] == ')') {
if (arrangement[i-2] == ')') {
deq.pop_back();
}
}
}
return answer;
}
6번째 문제(주식가격)
#include <string>
#include <vector>
#include <iostream>
#include <deque>
using namespace std;
void print_vector(vector<int> v) {
for (int i=0;i<v.size();i++) {
cout << v[i] << " ";
}
cout << endl;
}
vector solution(vector<int> prices) {
vector<int> answer(prices.size(), 0);
vector<int> values;
vector<int>::iterator iter;
values.push_back(0);
for (int i=1;i<prices.size();i++) {
for (iter=values.begin();iter != values.end();) {
answer[*iter] += 1;
if (prices[*iter] <= prices[i]) {
++iter;
} else {
values.erase(iter);
}
}
values.push_back(i);
}
return answer;
}
int main() {
vector out;
vector in = {1, 2, 3, 2, 3};
print_vector(in);
out = solution(in);
print_vector(out);
}
위의 solution함수를 조금 수정하면 아래와 같다. iterator를 사용하지 않게 수정한 것이지만 이해하기 어렵다. 순번을 저장하다 보니, values의 성격이 헷갈린다. 소스를 조금 복잡해도 이해하기 쉽게 위의 코드가 더 나을 거 같다.
vector solution(vector prices) {
vector answer(prices.size(), 0);
vector values;
values.push_back(0);
for (int i=1;i<prices.size();i++) {
for (int j=0;j<values.size();) {
answer[values[j]] += 1;
if (prices[values[j]] > prices[i]) {
values.erase(values.begin()+j);
} else {
j++;
}
}
values.push_back(i);
}
return answer;
}